libflame  revision_anchor
Functions
FLA_Apply_diag_matrix.c File Reference

(r)

Functions

FLA_Error FLA_Apply_diag_matrix (FLA_Side side, FLA_Conj conj, FLA_Obj x, FLA_Obj A)
 

Function Documentation

◆ FLA_Apply_diag_matrix()

FLA_Error FLA_Apply_diag_matrix ( FLA_Side  side,
FLA_Conj  conj,
FLA_Obj  x,
FLA_Obj  A 
)

References bl1_capdiagmv(), bl1_csapdiagmv(), bl1_dapdiagmv(), bl1_sapdiagmv(), bl1_zapdiagmv(), bl1_zdapdiagmv(), FLA_Apply_diag_matrix_check(), FLA_Check_error_level(), FLA_Obj_col_stride(), FLA_Obj_datatype(), FLA_Obj_length(), FLA_Obj_row_stride(), FLA_Obj_vector_inc(), FLA_Obj_width(), FLA_Param_map_flame_to_blis_conj(), and FLA_Param_map_flame_to_blis_side().

Referenced by FLA_Hevd_lv_unb_var1(), FLA_Hevd_lv_unb_var2(), FLA_Svd_ext_u_unb_var1(), FLA_Svd_uv_unb_var1(), and FLA_Svd_uv_unb_var2().

14 {
15  FLA_Datatype dt_x, dt_A;
16  int m_A, n_A;
17  int rs_A, cs_A;
18  int inc_x;
19  side1_t blis_side;
20  conj1_t blis_conj;
21 
22  if ( FLA_Check_error_level() >= FLA_MIN_ERROR_CHECKING )
23  FLA_Apply_diag_matrix_check( side, conj, x, A );
24 
25  dt_x = FLA_Obj_datatype( x );
26  dt_A = FLA_Obj_datatype( A );
27 
28  m_A = FLA_Obj_length( A );
29  n_A = FLA_Obj_width( A );
30 
31  rs_A = FLA_Obj_row_stride( A );
32  cs_A = FLA_Obj_col_stride( A );
33 
34  inc_x = FLA_Obj_vector_inc( x );
35 
36  FLA_Param_map_flame_to_blis_side( side, &blis_side );
37  FLA_Param_map_flame_to_blis_conj( conj, &blis_conj );
38 
39 
40  switch ( dt_A )
41  {
42  case FLA_FLOAT:
43  {
44  float* buff_x = ( float* ) FLA_FLOAT_PTR( x );
45  float* buff_A = ( float* ) FLA_FLOAT_PTR( A );
46 
47  bl1_sapdiagmv( blis_side,
48  blis_conj,
49  m_A,
50  n_A,
51  buff_x, inc_x,
52  buff_A, rs_A, cs_A );
53 
54  break;
55  }
56 
57  case FLA_DOUBLE:
58  {
59  double* buff_x = ( double* ) FLA_DOUBLE_PTR( x );
60  double* buff_A = ( double* ) FLA_DOUBLE_PTR( A );
61 
62  bl1_dapdiagmv( blis_side,
63  blis_conj,
64  m_A,
65  n_A,
66  buff_x, inc_x,
67  buff_A, rs_A, cs_A );
68 
69  break;
70  }
71 
72  case FLA_COMPLEX:
73  {
74  if ( dt_x == FLA_FLOAT )
75  {
76  float* buff_x = ( float* ) FLA_FLOAT_PTR( x );
77  scomplex* buff_A = ( scomplex* ) FLA_COMPLEX_PTR( A );
78 
79  bl1_csapdiagmv( blis_side,
80  blis_conj,
81  m_A,
82  n_A,
83  buff_x, inc_x,
84  buff_A, rs_A, cs_A );
85  }
86  else if ( dt_x == FLA_COMPLEX )
87  {
88  scomplex* buff_x = ( scomplex* ) FLA_COMPLEX_PTR( x );
89  scomplex* buff_A = ( scomplex* ) FLA_COMPLEX_PTR( A );
90 
91  bl1_capdiagmv( blis_side,
92  blis_conj,
93  m_A,
94  n_A,
95  buff_x, inc_x,
96  buff_A, rs_A, cs_A );
97  }
98 
99  break;
100  }
101 
102  case FLA_DOUBLE_COMPLEX:
103  {
104  if ( dt_x == FLA_DOUBLE )
105  {
106  double* buff_x = ( double* ) FLA_DOUBLE_PTR( x );
107  dcomplex* buff_A = ( dcomplex* ) FLA_DOUBLE_COMPLEX_PTR( A );
108 
109  bl1_zdapdiagmv( blis_side,
110  blis_conj,
111  m_A,
112  n_A,
113  buff_x, inc_x,
114  buff_A, rs_A, cs_A );
115  }
116  else if ( dt_x == FLA_DOUBLE_COMPLEX )
117  {
118  dcomplex* buff_x = ( dcomplex* ) FLA_DOUBLE_COMPLEX_PTR( x );
119  dcomplex* buff_A = ( dcomplex* ) FLA_DOUBLE_COMPLEX_PTR( A );
120 
121  bl1_zapdiagmv( blis_side,
122  blis_conj,
123  m_A,
124  n_A,
125  buff_x, inc_x,
126  buff_A, rs_A, cs_A );
127  }
128 
129  break;
130  }
131  }
132 
133  return FLA_SUCCESS;
134 }
void bl1_sapdiagmv(side1_t side, conj1_t conj, int m, int n, float *x, int incx, float *a, int a_rs, int a_cs)
Definition: bl1_apdiagmv.c:13
side1_t
Definition: blis_type_defs.h:66
void bl1_zdapdiagmv(side1_t side, conj1_t conj, int m, int n, double *x, int incx, dcomplex *a, int a_rs, int a_cs)
Definition: bl1_apdiagmv.c:233
conj1_t
Definition: blis_type_defs.h:79
dim_t FLA_Obj_row_stride(FLA_Obj obj)
Definition: FLA_Query.c:167
void bl1_capdiagmv(side1_t side, conj1_t conj, int m, int n, scomplex *x, int incx, scomplex *a, int a_rs, int a_cs)
Definition: bl1_apdiagmv.c:178
void bl1_dapdiagmv(side1_t side, conj1_t conj, int m, int n, double *x, int incx, double *a, int a_rs, int a_cs)
Definition: bl1_apdiagmv.c:68
FLA_Datatype FLA_Obj_datatype(FLA_Obj obj)
Definition: FLA_Query.c:13
void bl1_zapdiagmv(side1_t side, conj1_t conj, int m, int n, dcomplex *x, int incx, dcomplex *a, int a_rs, int a_cs)
Definition: bl1_apdiagmv.c:288
dim_t FLA_Obj_width(FLA_Obj obj)
Definition: FLA_Query.c:123
void FLA_Param_map_flame_to_blis_side(FLA_Uplo side, side1_t *blis_side)
Definition: FLA_Param.c:301
void bl1_csapdiagmv(side1_t side, conj1_t conj, int m, int n, float *x, int incx, scomplex *a, int a_rs, int a_cs)
Definition: bl1_apdiagmv.c:123
Definition: blis_type_defs.h:132
unsigned int FLA_Check_error_level(void)
Definition: FLA_Check.c:18
void FLA_Param_map_flame_to_blis_conj(FLA_Conj conj, conj1_t *blis_conj)
Definition: FLA_Param.c:269
int FLA_Datatype
Definition: FLA_type_defs.h:49
dim_t FLA_Obj_col_stride(FLA_Obj obj)
Definition: FLA_Query.c:174
dim_t FLA_Obj_vector_inc(FLA_Obj obj)
Definition: FLA_Query.c:145
FLA_Error FLA_Apply_diag_matrix_check(FLA_Side side, FLA_Conj conj, FLA_Obj x, FLA_Obj A)
Definition: FLA_Apply_diag_matrix_check.c:13
dim_t FLA_Obj_length(FLA_Obj obj)
Definition: FLA_Query.c:116
Definition: blis_type_defs.h:137