libflame revision_anchor
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 )
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}
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
void FLA_Param_map_flame_to_blis_side(FLA_Uplo side, side1_t *blis_side)
Definition FLA_Param.c:301
dim_t FLA_Obj_width(FLA_Obj obj)
Definition FLA_Query.c:123
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
void FLA_Param_map_flame_to_blis_conj(FLA_Conj conj, conj1_t *blis_conj)
Definition FLA_Param.c:269
dim_t FLA_Obj_vector_inc(FLA_Obj obj)
Definition FLA_Query.c:145
FLA_Datatype FLA_Obj_datatype(FLA_Obj obj)
Definition FLA_Query.c:13
int FLA_Datatype
Definition FLA_type_defs.h:49
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
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_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
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
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
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
conj1_t
Definition blis_type_defs.h:80
side1_t
Definition blis_type_defs.h:67
Definition blis_type_defs.h:138
Definition blis_type_defs.h:133

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().