libflame  revision_anchor
Functions
FLA_Add_to_diag.c File Reference

(r)

Functions

FLA_Error FLA_Add_to_diag (void *diag_value, FLA_Obj A)
 

Function Documentation

◆ FLA_Add_to_diag()

FLA_Error FLA_Add_to_diag ( void *  diag_value,
FLA_Obj  A 
)
14 {
15  FLA_Datatype datatype;
16  dim_t j, min_m_n;
17  dim_t rs, cs;
18 
19  if ( FLA_Check_error_level() >= FLA_MIN_ERROR_CHECKING )
20  FLA_Add_to_diag_check( diag_value, A );
21 
22  datatype = FLA_Obj_datatype( A );
23  min_m_n = FLA_Obj_min_dim( A );
24  rs = FLA_Obj_row_stride( A );
25  cs = FLA_Obj_col_stride( A );
26 
27  switch ( datatype ){
28 
29  case FLA_FLOAT:
30  {
31  float *buff_A = ( float * ) FLA_FLOAT_PTR( A );
32  float *value_ptr = ( float * ) diag_value;
33 
34  for ( j = 0; j < min_m_n; j++ )
35  buff_A[ j*cs + j*rs ] += *value_ptr;
36 
37  break;
38  }
39 
40  case FLA_DOUBLE:
41  {
42  double *buff_A = ( double * ) FLA_DOUBLE_PTR( A );
43  double *value_ptr = ( double * ) diag_value;
44 
45  for ( j = 0; j < min_m_n; j++ )
46  buff_A[ j*cs + j*rs ] += *value_ptr;
47 
48  break;
49  }
50 
51  case FLA_COMPLEX:
52  {
53  scomplex *buff_A = ( scomplex * ) FLA_COMPLEX_PTR( A );
54  scomplex *value_ptr = ( scomplex * ) diag_value;
55 
56  for ( j = 0; j < min_m_n; j++ )
57  {
58  buff_A[ j*cs + j*rs ].real += value_ptr->real;
59  buff_A[ j*cs + j*rs ].imag += value_ptr->imag;
60  }
61 
62  break;
63  }
64 
65  case FLA_DOUBLE_COMPLEX:
66  {
67  dcomplex *buff_A = ( dcomplex * ) FLA_DOUBLE_COMPLEX_PTR( A );
68  dcomplex *value_ptr = ( dcomplex * ) diag_value;
69 
70  for ( j = 0; j < min_m_n; j++ )
71  {
72  buff_A[ j*cs + j*rs ].real += value_ptr->real;
73  buff_A[ j*cs + j*rs ].imag += value_ptr->imag;
74  }
75 
76  break;
77  }
78 
79  }
80 
81  return FLA_SUCCESS;
82 }
FLA_Error FLA_Add_to_diag_check(void *diag_value, FLA_Obj A)
Definition: FLA_Add_to_diag_check.c:13
dim_t FLA_Obj_row_stride(FLA_Obj obj)
Definition: FLA_Query.c:167
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
dim_t FLA_Obj_min_dim(FLA_Obj obj)
Definition: FLA_Query.c:153
FLA_Datatype FLA_Obj_datatype(FLA_Obj obj)
Definition: FLA_Query.c:13
int FLA_Datatype
Definition: FLA_type_defs.h:49
unsigned long dim_t
Definition: FLA_type_defs.h:71
Definition: blis_type_defs.h:138
double real
Definition: blis_type_defs.h:139
double imag
Definition: blis_type_defs.h:139
Definition: blis_type_defs.h:133
float imag
Definition: blis_type_defs.h:134
float real
Definition: blis_type_defs.h:134

References FLA_Add_to_diag_check(), FLA_Check_error_level(), FLA_Obj_col_stride(), FLA_Obj_datatype(), FLA_Obj_min_dim(), FLA_Obj_row_stride(), scomplex::imag, dcomplex::imag, scomplex::real, and dcomplex::real.