libflame revision_anchor
FLA_Trmm_external.c File Reference

(r)

Functions

FLA_Error FLA_Trmm_external (FLA_Side side, FLA_Uplo uplo, FLA_Trans trans, FLA_Diag diag, FLA_Obj alpha, FLA_Obj A, FLA_Obj B)
 

Function Documentation

◆ FLA_Trmm_external()

FLA_Error FLA_Trmm_external ( FLA_Side side,
FLA_Uplo uplo,
FLA_Trans trans,
FLA_Diag diag,
FLA_Obj alpha,
FLA_Obj A,
FLA_Obj B )
14{
15 FLA_Datatype datatype;
16 int m_B, n_B;
17 int rs_A, cs_A;
18 int rs_B, cs_B;
19 side1_t blis_side;
20 uplo1_t blis_uplo;
21 trans1_t blis_trans;
22 diag1_t blis_diag;
23
24 if ( FLA_Check_error_level() == FLA_FULL_ERROR_CHECKING )
25 FLA_Trmm_check( side, uplo, trans, diag, alpha, A, B );
26
27 if ( FLA_Obj_has_zero_dim( B ) ) return FLA_SUCCESS;
28
29 datatype = FLA_Obj_datatype( A );
30
31 rs_A = FLA_Obj_row_stride( A );
32 cs_A = FLA_Obj_col_stride( A );
33
34 m_B = FLA_Obj_length( B );
35 n_B = FLA_Obj_width( B );
36 rs_B = FLA_Obj_row_stride( B );
37 cs_B = FLA_Obj_col_stride( B );
38
39 FLA_Param_map_flame_to_blis_side( side, &blis_side );
40 FLA_Param_map_flame_to_blis_uplo( uplo, &blis_uplo );
41 FLA_Param_map_flame_to_blis_trans( trans, &blis_trans );
42 FLA_Param_map_flame_to_blis_diag( diag, &blis_diag );
43
44
45 switch( datatype ){
46
47 case FLA_FLOAT:
48 {
49 float *buff_A = ( float * ) FLA_FLOAT_PTR( A );
50 float *buff_B = ( float * ) FLA_FLOAT_PTR( B );
51 float *buff_alpha = ( float * ) FLA_FLOAT_PTR( alpha );
52
53 bl1_strmm( blis_side,
54 blis_uplo,
55 blis_trans,
56 blis_diag,
57 m_B,
58 n_B,
59 buff_alpha,
60 buff_A, rs_A, cs_A,
61 buff_B, rs_B, cs_B );
62
63 break;
64 }
65
66 case FLA_DOUBLE:
67 {
68 double *buff_A = ( double * ) FLA_DOUBLE_PTR( A );
69 double *buff_B = ( double * ) FLA_DOUBLE_PTR( B );
70 double *buff_alpha = ( double * ) FLA_DOUBLE_PTR( alpha );
71
72 bl1_dtrmm( blis_side,
73 blis_uplo,
74 blis_trans,
75 blis_diag,
76 m_B,
77 n_B,
78 buff_alpha,
79 buff_A, rs_A, cs_A,
80 buff_B, rs_B, cs_B );
81
82 break;
83 }
84
85 case FLA_COMPLEX:
86 {
87 scomplex *buff_A = ( scomplex * ) FLA_COMPLEX_PTR( A );
88 scomplex *buff_B = ( scomplex * ) FLA_COMPLEX_PTR( B );
89 scomplex *buff_alpha = ( scomplex * ) FLA_COMPLEX_PTR( alpha );
90
91 bl1_ctrmm( blis_side,
92 blis_uplo,
93 blis_trans,
94 blis_diag,
95 m_B,
96 n_B,
97 buff_alpha,
98 buff_A, rs_A, cs_A,
99 buff_B, rs_B, cs_B );
100
101 break;
102 }
103
104
105 case FLA_DOUBLE_COMPLEX:
106 {
107 dcomplex *buff_A = ( dcomplex * ) FLA_DOUBLE_COMPLEX_PTR( A );
108 dcomplex *buff_B = ( dcomplex * ) FLA_DOUBLE_COMPLEX_PTR( B );
109 dcomplex *buff_alpha = ( dcomplex * ) FLA_DOUBLE_COMPLEX_PTR( alpha );
110
111 bl1_ztrmm( blis_side,
112 blis_uplo,
113 blis_trans,
114 blis_diag,
115 m_B,
116 n_B,
117 buff_alpha,
118 buff_A, rs_A, cs_A,
119 buff_B, rs_B, cs_B );
120
121 break;
122 }
123
124 }
125
126 return FLA_SUCCESS;
127}
FLA_Error FLA_Trmm_check(FLA_Side side, FLA_Uplo uplo, FLA_Trans trans, FLA_Diag diag, FLA_Obj alpha, FLA_Obj A, FLA_Obj B)
Definition FLA_Trmm_check.c:13
void FLA_Param_map_flame_to_blis_side(FLA_Uplo side, side1_t *blis_side)
Definition FLA_Param.c:301
void FLA_Param_map_flame_to_blis_diag(FLA_Diag diag, diag1_t *blis_diag)
Definition FLA_Param.c:317
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
void FLA_Param_map_flame_to_blis_trans(FLA_Trans trans, trans1_t *blis_trans)
Definition FLA_Param.c:245
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_uplo(FLA_Uplo uplo, uplo1_t *blis_uplo)
Definition FLA_Param.c:285
FLA_Datatype FLA_Obj_datatype(FLA_Obj obj)
Definition FLA_Query.c:13
int FLA_Datatype
Definition FLA_type_defs.h:49
void bl1_strmm(side1_t side, uplo1_t uplo, trans1_t trans, diag1_t diag, int m, int n, float *alpha, float *a, int a_rs, int a_cs, float *b, int b_rs, int b_cs)
Definition bl1_trmm.c:13
void bl1_ctrmm(side1_t side, uplo1_t uplo, trans1_t trans, diag1_t diag, int m, int n, scomplex *alpha, scomplex *a, int a_rs, int a_cs, scomplex *b, int b_rs, int b_cs)
Definition bl1_trmm.c:219
void bl1_ztrmm(side1_t side, uplo1_t uplo, trans1_t trans, diag1_t diag, int m, int n, dcomplex *alpha, dcomplex *a, int a_rs, int a_cs, dcomplex *b, int b_rs, int b_cs)
Definition bl1_trmm.c:369
void bl1_dtrmm(side1_t side, uplo1_t uplo, trans1_t trans, diag1_t diag, int m, int n, double *alpha, double *a, int a_rs, int a_cs, double *b, int b_rs, int b_cs)
Definition bl1_trmm.c:116
uplo1_t
Definition blis_type_defs.h:61
diag1_t
Definition blis_type_defs.h:73
trans1_t
Definition blis_type_defs.h:53
side1_t
Definition blis_type_defs.h:67
Definition blis_type_defs.h:138
Definition blis_type_defs.h:133

References bl1_ctrmm(), bl1_dtrmm(), bl1_strmm(), bl1_ztrmm(), FLA_Check_error_level(), FLA_Obj_col_stride(), FLA_Obj_datatype(), FLA_Obj_has_zero_dim(), FLA_Obj_length(), FLA_Obj_row_stride(), FLA_Obj_width(), FLA_Param_map_flame_to_blis_diag(), FLA_Param_map_flame_to_blis_side(), FLA_Param_map_flame_to_blis_trans(), FLA_Param_map_flame_to_blis_uplo(), and FLA_Trmm_check().

Referenced by FLA_LQ_UT_blk_var2(), FLA_QR_UT_blk_var2(), FLA_Trmm(), FLA_Trmm_llc_task(), FLA_Trmm_llh_task(), FLA_Trmm_lln_task(), FLA_Trmm_llt_task(), FLA_Trmm_luc_task(), FLA_Trmm_luh_task(), FLA_Trmm_lun_task(), FLA_Trmm_lut_task(), FLA_Trmm_rlc_task(), FLA_Trmm_rlh_task(), FLA_Trmm_rln_task(), FLA_Trmm_rlt_task(), FLA_Trmm_ruc_task(), FLA_Trmm_ruh_task(), FLA_Trmm_run_task(), FLA_Trmm_rut_task(), and FLA_Trmm_task().