libflame  revision_anchor
Functions
FLA_Scalc_external.c File Reference

(r)

Functions

FLA_Error FLA_Scalc_external (FLA_Conj conj, FLA_Obj alpha, FLA_Obj A)
 

Function Documentation

◆ FLA_Scalc_external()

FLA_Error FLA_Scalc_external ( FLA_Conj  conj,
FLA_Obj  alpha,
FLA_Obj  A 
)
14 {
15  FLA_Datatype datatype, dt_alpha;
16  int m_A, n_A;
17  int rs_A, cs_A;
18  conj1_t blis_conj;
19 
20  if ( FLA_Check_error_level() == FLA_FULL_ERROR_CHECKING )
21  FLA_Scalc_check( conj, alpha, A );
22 
23  if ( FLA_Obj_has_zero_dim( A ) ) return FLA_SUCCESS;
24 
25  if ( FLA_Obj_equals( alpha, FLA_ONE ) )
26  {
27  return FLA_SUCCESS;
28  }
29  else if ( FLA_Obj_equals( alpha, FLA_ZERO ) )
30  {
31  FLA_Set( FLA_ZERO, A );
32  return FLA_SUCCESS;
33  }
34 
35  datatype = FLA_Obj_datatype( A );
36 
37  m_A = FLA_Obj_length( A );
38  n_A = FLA_Obj_width( A );
39  rs_A = FLA_Obj_row_stride( A );
40  cs_A = FLA_Obj_col_stride( A );
41 
42  if ( FLA_Obj_is_constant( alpha ) )
43  dt_alpha = datatype;
44  else
45  dt_alpha = FLA_Obj_datatype( alpha );
46 
47  FLA_Param_map_flame_to_blis_conj( conj, &blis_conj );
48 
49  switch ( datatype ){
50 
51  case FLA_FLOAT:
52  {
53  float *buff_A = ( float * ) FLA_FLOAT_PTR( A );
54  float *buff_alpha = ( float * ) FLA_FLOAT_PTR( alpha );
55 
56  bl1_sscalm( blis_conj,
57  m_A,
58  n_A,
59  buff_alpha,
60  buff_A, rs_A, cs_A );
61 
62  break;
63  }
64 
65  case FLA_DOUBLE:
66  {
67  double *buff_A = ( double * ) FLA_DOUBLE_PTR( A );
68  double *buff_alpha = ( double * ) FLA_DOUBLE_PTR( alpha );
69 
70  bl1_dscalm( blis_conj,
71  m_A,
72  n_A,
73  buff_alpha,
74  buff_A, rs_A, cs_A );
75 
76  break;
77  }
78 
79  case FLA_COMPLEX:
80  {
81  if ( dt_alpha == FLA_COMPLEX )
82  {
83  scomplex *buff_A = ( scomplex * ) FLA_COMPLEX_PTR( A );
84  scomplex *buff_alpha = ( scomplex * ) FLA_COMPLEX_PTR( alpha );
85 
86  bl1_cscalm( blis_conj,
87  m_A,
88  n_A,
89  buff_alpha,
90  buff_A, rs_A, cs_A );
91  }
92  else if ( dt_alpha == FLA_FLOAT )
93  {
94  scomplex *buff_A = ( scomplex * ) FLA_COMPLEX_PTR( A );
95  float *buff_alpha = ( float * ) FLA_FLOAT_PTR( alpha );
96 
97  bl1_csscalm( blis_conj,
98  m_A,
99  n_A,
100  buff_alpha,
101  buff_A, rs_A, cs_A );
102  }
103 
104  break;
105  }
106 
107  case FLA_DOUBLE_COMPLEX:
108  {
109  if ( dt_alpha == FLA_DOUBLE_COMPLEX )
110  {
111  dcomplex *buff_A = ( dcomplex * ) FLA_DOUBLE_COMPLEX_PTR( A );
112  dcomplex *buff_alpha = ( dcomplex * ) FLA_DOUBLE_COMPLEX_PTR( alpha );
113 
114  bl1_zscalm( blis_conj,
115  m_A,
116  n_A,
117  buff_alpha,
118  buff_A, rs_A, cs_A );
119  }
120  else if ( dt_alpha == FLA_DOUBLE )
121  {
122  dcomplex *buff_A = ( dcomplex * ) FLA_DOUBLE_COMPLEX_PTR( A );
123  double *buff_alpha = ( double * ) FLA_DOUBLE_PTR( alpha );
124 
125  bl1_zdscalm( blis_conj,
126  m_A,
127  n_A,
128  buff_alpha,
129  buff_A, rs_A, cs_A );
130  }
131 
132  break;
133  }
134 
135  }
136 
137  return FLA_SUCCESS;
138 }
FLA_Error FLA_Scalc_check(FLA_Conj conjalpha, FLA_Obj alpha, FLA_Obj A)
Definition: FLA_Scalc_check.c:13
FLA_Obj FLA_ZERO
Definition: FLA_Init.c:20
FLA_Obj FLA_ONE
Definition: FLA_Init.c:18
dim_t FLA_Obj_width(FLA_Obj obj)
Definition: FLA_Query.c:123
FLA_Bool FLA_Obj_has_zero_dim(FLA_Obj A)
Definition: FLA_Query.c:400
dim_t FLA_Obj_row_stride(FLA_Obj obj)
Definition: FLA_Query.c:167
dim_t FLA_Obj_length(FLA_Obj obj)
Definition: FLA_Query.c:116
dim_t FLA_Obj_col_stride(FLA_Obj obj)
Definition: FLA_Query.c:174
unsigned int FLA_Check_error_level(void)
Definition: FLA_Check.c:18
FLA_Bool FLA_Obj_is_constant(FLA_Obj A)
Definition: FLA_Query.c:290
FLA_Bool FLA_Obj_equals(FLA_Obj A, FLA_Obj B)
Definition: FLA_Query.c:507
void FLA_Param_map_flame_to_blis_conj(FLA_Conj conj, conj1_t *blis_conj)
Definition: FLA_Param.c:269
FLA_Datatype FLA_Obj_datatype(FLA_Obj obj)
Definition: FLA_Query.c:13
int FLA_Datatype
Definition: FLA_type_defs.h:49
FLA_Error FLA_Set(FLA_Obj alpha, FLA_Obj A)
Definition: FLA_Set.c:13
void bl1_zscalm(conj1_t conj, int m, int n, dcomplex *alpha, dcomplex *a, int a_rs, int a_cs)
Definition: bl1_scalm.c:273
void bl1_csscalm(conj1_t conj, int m, int n, float *alpha, scomplex *a, int a_rs, int a_cs)
Definition: bl1_scalm.c:117
void bl1_zdscalm(conj1_t conj, int m, int n, double *alpha, dcomplex *a, int a_rs, int a_cs)
Definition: bl1_scalm.c:221
void bl1_cscalm(conj1_t conj, int m, int n, scomplex *alpha, scomplex *a, int a_rs, int a_cs)
Definition: bl1_scalm.c:169
void bl1_sscalm(conj1_t conj, int m, int n, float *alpha, float *a, int a_rs, int a_cs)
Definition: bl1_scalm.c:13
void bl1_dscalm(conj1_t conj, int m, int n, double *alpha, double *a, int a_rs, int a_cs)
Definition: bl1_scalm.c:65
conj1_t
Definition: blis_type_defs.h:80
Definition: blis_type_defs.h:138
Definition: blis_type_defs.h:133

References bl1_cscalm(), bl1_csscalm(), bl1_dscalm(), bl1_sscalm(), bl1_zdscalm(), bl1_zscalm(), FLA_Check_error_level(), FLA_Obj_col_stride(), FLA_Obj_datatype(), FLA_Obj_equals(), FLA_Obj_has_zero_dim(), FLA_Obj_is_constant(), FLA_Obj_length(), FLA_Obj_row_stride(), FLA_Obj_width(), FLA_ONE, FLA_Param_map_flame_to_blis_conj(), FLA_Scalc_check(), FLA_Set(), and FLA_ZERO.

Referenced by FLA_Scalc(), FLA_Trmm_llc_unb_var1(), FLA_Trmm_llc_unb_var2(), FLA_Trmm_llh_unb_var1(), FLA_Trmm_llh_unb_var2(), FLA_Trmm_luc_unb_var1(), FLA_Trmm_luc_unb_var2(), FLA_Trmm_luh_unb_var1(), FLA_Trmm_luh_unb_var2(), FLA_Trmm_rlc_unb_var1(), FLA_Trmm_rlc_unb_var2(), FLA_Trmm_rlh_unb_var1(), FLA_Trmm_rlh_unb_var2(), FLA_Trmm_ruc_unb_var1(), FLA_Trmm_ruc_unb_var2(), FLA_Trmm_ruh_unb_var1(), and FLA_Trmm_ruh_unb_var2().