libflame  revision_anchor
Functions
FLA_Axpy_external.c File Reference

(r)

Functions

FLA_Error FLA_Axpy_external (FLA_Obj alpha, FLA_Obj A, FLA_Obj B)
 

Function Documentation

◆ FLA_Axpy_external()

FLA_Error FLA_Axpy_external ( FLA_Obj  alpha,
FLA_Obj  A,
FLA_Obj  B 
)

References bl1_caxpymt(), bl1_daxpymt(), bl1_saxpymt(), bl1_zaxpymt(), FLA_Axpy_check(), FLA_Check_error_level(), FLA_Obj_col_stride(), FLA_Obj_datatype(), FLA_Obj_has_zero_dim(), FLA_Obj_is_conformal_to(), FLA_Obj_length(), FLA_Obj_row_stride(), FLA_Obj_width(), and FLA_Param_map_flame_to_blis_trans().

Referenced by FLA_Apply_H2_UT_l_unb_var1(), FLA_Apply_H2_UT_piv_row(), FLA_Apply_H2_UT_r_unb_var1(), FLA_Apply_HUD_UT_lh_unb_var1(), FLA_Axpy(), FLA_Axpy_task(), FLA_Eig_gest_il_unb_var1(), FLA_Eig_gest_il_unb_var2(), FLA_Eig_gest_il_unb_var3(), FLA_Eig_gest_il_unb_var4(), FLA_Eig_gest_il_unb_var5(), FLA_Eig_gest_iu_unb_var1(), FLA_Eig_gest_iu_unb_var2(), FLA_Eig_gest_iu_unb_var3(), FLA_Eig_gest_iu_unb_var4(), FLA_Eig_gest_iu_unb_var5(), FLA_Eig_gest_nl_unb_var1(), FLA_Eig_gest_nl_unb_var2(), FLA_Eig_gest_nl_unb_var4(), FLA_Eig_gest_nl_unb_var5(), FLA_Eig_gest_nu_unb_var1(), FLA_Eig_gest_nu_unb_var2(), FLA_Eig_gest_nu_unb_var4(), FLA_Eig_gest_nu_unb_var5(), FLA_Trmvsx_external(), FLA_Trsvsx_external(), and FLASH_Axpy_hierarchy().

14 {
15  FLA_Datatype datatype;
16  int m_B, n_B;
17  int rs_A, cs_A;
18  int rs_B, cs_B;
19  trans1_t blis_trans;
20 
21  if ( FLA_Check_error_level() == FLA_FULL_ERROR_CHECKING )
22  FLA_Axpy_check( alpha, A, B );
23 
24  if ( FLA_Obj_has_zero_dim( A ) ) return FLA_SUCCESS;
25 
26  datatype = FLA_Obj_datatype( A );
27 
28  rs_A = FLA_Obj_row_stride( A );
29  cs_A = FLA_Obj_col_stride( A );
30 
31  m_B = FLA_Obj_length( B );
32  n_B = FLA_Obj_width( B );
33  rs_B = FLA_Obj_row_stride( B );
34  cs_B = FLA_Obj_col_stride( B );
35 
36  if ( FLA_Obj_is_conformal_to( FLA_NO_TRANSPOSE, A, B ) )
37  FLA_Param_map_flame_to_blis_trans( FLA_NO_TRANSPOSE, &blis_trans );
38  else // if ( FLA_Obj_is_conformal_to( FLA_TRANSPOSE, A, B ) )
39  FLA_Param_map_flame_to_blis_trans( FLA_TRANSPOSE, &blis_trans );
40 
41  switch ( datatype ){
42 
43  case FLA_FLOAT:
44  {
45  float *buff_alpha = ( float * ) FLA_FLOAT_PTR( alpha );
46  float *buff_A = ( float * ) FLA_FLOAT_PTR( A );
47  float *buff_B = ( float * ) FLA_FLOAT_PTR( B );
48 
49  bl1_saxpymt( blis_trans,
50  m_B,
51  n_B,
52  buff_alpha,
53  buff_A, rs_A, cs_A,
54  buff_B, rs_B, cs_B );
55 
56  break;
57  }
58 
59  case FLA_DOUBLE:
60  {
61  double *buff_alpha = ( double * ) FLA_DOUBLE_PTR( alpha );
62  double *buff_A = ( double * ) FLA_DOUBLE_PTR( A );
63  double *buff_B = ( double * ) FLA_DOUBLE_PTR( B );
64 
65  bl1_daxpymt( blis_trans,
66  m_B,
67  n_B,
68  buff_alpha,
69  buff_A, rs_A, cs_A,
70  buff_B, rs_B, cs_B );
71 
72  break;
73  }
74 
75  case FLA_COMPLEX:
76  {
77  scomplex *buff_alpha = ( scomplex * ) FLA_COMPLEX_PTR( alpha );
78  scomplex *buff_A = ( scomplex * ) FLA_COMPLEX_PTR( A );
79  scomplex *buff_B = ( scomplex * ) FLA_COMPLEX_PTR( B );
80 
81  bl1_caxpymt( blis_trans,
82  m_B,
83  n_B,
84  buff_alpha,
85  buff_A, rs_A, cs_A,
86  buff_B, rs_B, cs_B );
87 
88  break;
89  }
90 
91  case FLA_DOUBLE_COMPLEX:
92  {
93  dcomplex *buff_alpha = ( dcomplex * ) FLA_DOUBLE_COMPLEX_PTR( alpha );
94  dcomplex *buff_A = ( dcomplex * ) FLA_DOUBLE_COMPLEX_PTR( A );
95  dcomplex *buff_B = ( dcomplex * ) FLA_DOUBLE_COMPLEX_PTR( B );
96 
97  bl1_zaxpymt( blis_trans,
98  m_B,
99  n_B,
100  buff_alpha,
101  buff_A, rs_A, cs_A,
102  buff_B, rs_B, cs_B );
103 
104  break;
105  }
106 
107  }
108 
109  return FLA_SUCCESS;
110 }
void FLA_Param_map_flame_to_blis_trans(FLA_Trans trans, trans1_t *blis_trans)
Definition: FLA_Param.c:245
dim_t FLA_Obj_row_stride(FLA_Obj obj)
Definition: FLA_Query.c:167
trans1_t
Definition: blis_type_defs.h:52
FLA_Bool FLA_Obj_is_conformal_to(FLA_Trans trans, FLA_Obj A, FLA_Obj B)
Definition: FLA_Query.c:438
void bl1_caxpymt(trans1_t trans, int m, int n, scomplex *alpha, scomplex *a, int a_rs, int a_cs, scomplex *b, int b_rs, int b_cs)
Definition: bl1_axpymt.c:149
void bl1_zaxpymt(trans1_t trans, int m, int n, dcomplex *alpha, dcomplex *a, int a_rs, int a_cs, dcomplex *b, int b_rs, int b_cs)
Definition: bl1_axpymt.c:248
FLA_Datatype FLA_Obj_datatype(FLA_Obj obj)
Definition: FLA_Query.c:13
FLA_Bool FLA_Obj_has_zero_dim(FLA_Obj A)
Definition: FLA_Query.c:400
dim_t FLA_Obj_width(FLA_Obj obj)
Definition: FLA_Query.c:123
void bl1_saxpymt(trans1_t trans, int m, int n, float *alpha, float *a, int a_rs, int a_cs, float *b, int b_rs, int b_cs)
Definition: bl1_axpymt.c:13
Definition: blis_type_defs.h:132
FLA_Error FLA_Axpy_check(FLA_Obj alpha, FLA_Obj A, FLA_Obj B)
Definition: FLA_Axpy_check.c:13
unsigned int FLA_Check_error_level(void)
Definition: FLA_Check.c:18
int FLA_Datatype
Definition: FLA_type_defs.h:49
dim_t FLA_Obj_col_stride(FLA_Obj obj)
Definition: FLA_Query.c:174
void bl1_daxpymt(trans1_t trans, int m, int n, double *alpha, double *a, int a_rs, int a_cs, double *b, int b_rs, int b_cs)
Definition: bl1_axpymt.c:81
dim_t FLA_Obj_length(FLA_Obj obj)
Definition: FLA_Query.c:116
Definition: blis_type_defs.h:137