41 if ( uplo == FLA_LOWER_TRIANGULAR ){
47 float* buff_alpha = (
float* ) FLA_FLOAT_PTR( alpha );
48 float* buff_A_gpu = (
float* ) A_gpu;
50 for (
i = 0;
i < min( n_A, m_A );
i++ )
53 buff_A_gpu +
i * ldim_A +
i, inc_A );
60 double* buff_alpha = (
double* ) FLA_DOUBLE_PTR( alpha );
61 double* buff_A_gpu = (
double* ) A_gpu;
63 for (
i = 0;
i < min( n_A, m_A );
i++ )
66 buff_A_gpu +
i * ldim_A +
i, inc_A );
73 cuComplex* buff_alpha = ( cuComplex* ) FLA_COMPLEX_PTR( alpha );
74 cuComplex* buff_A_gpu = ( cuComplex* ) A_gpu;
76 for (
i = 0;
i < min( n_A, m_A );
i++ )
79 buff_A_gpu +
i * ldim_A +
i, inc_A );
84 case FLA_DOUBLE_COMPLEX:
86 cuDoubleComplex* buff_alpha = ( cuDoubleComplex* ) FLA_DOUBLE_COMPLEX_PTR( alpha );
87 cuDoubleComplex* buff_A_gpu = ( cuDoubleComplex* ) A_gpu;
89 for (
i = 0;
i < min( n_A, m_A );
i++ )
92 buff_A_gpu +
i * ldim_A +
i, inc_A );
101 else if ( uplo == FLA_UPPER_TRIANGULAR ){
107 float* buff_alpha = (
float* ) FLA_FLOAT_PTR( alpha );
108 float* buff_A_gpu = (
float* ) A_gpu;
110 for (
i = 0;
i < n_A;
i++ )
111 cublasSscal( min(
i + 1, m_A ),
113 buff_A_gpu +
i * ldim_A, inc_A );
120 double* buff_alpha = (
double* ) FLA_DOUBLE_PTR( alpha );
121 double* buff_A_gpu = (
double* ) A_gpu;
123 for (
i = 0;
i < n_A;
i++ )
124 cublasDscal( min(
i + 1, m_A ),
126 buff_A_gpu +
i * ldim_A, inc_A );
133 cuComplex* buff_alpha = ( cuComplex* ) FLA_COMPLEX_PTR( alpha );
134 cuComplex* buff_A_gpu = ( cuComplex* ) A_gpu;
136 for (
i = 0;
i < n_A;
i++ )
137 cublasCscal( min(
i + 1, m_A ),
139 buff_A_gpu +
i * ldim_A, inc_A );
144 case FLA_DOUBLE_COMPLEX:
146 cuDoubleComplex* buff_alpha = ( cuDoubleComplex* ) FLA_DOUBLE_COMPLEX_PTR( alpha );
147 cuDoubleComplex* buff_A_gpu = ( cuDoubleComplex* ) A_gpu;
149 for (
i = 0;
i < n_A;
i++ )
150 cublasZscal( min(
i + 1, m_A ),
152 buff_A_gpu +
i * ldim_A, inc_A );
FLA_Error FLA_Scalr_check(FLA_Uplo uplo, FLA_Obj alpha, FLA_Obj A)
Definition: FLA_Scalr_check.c:13
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_length(FLA_Obj obj)
Definition: FLA_Query.c:116
unsigned int FLA_Check_error_level(void)
Definition: FLA_Check.c:18
FLA_Bool FLA_Obj_equals(FLA_Obj A, FLA_Obj B)
Definition: FLA_Query.c:507
FLA_Datatype FLA_Obj_datatype(FLA_Obj obj)
Definition: FLA_Query.c:13
int FLA_Datatype
Definition: FLA_type_defs.h:49
int i
Definition: bl1_axmyv2.c:145