libflame  revision_anchor
Functions
FLA_Gemm_ct_unb_var1.c File Reference

(r)

Functions

FLA_Error FLA_Gemm_ct_unb_var1 (FLA_Obj alpha, FLA_Obj A, FLA_Obj B, FLA_Obj beta, FLA_Obj C)
 

Function Documentation

◆ FLA_Gemm_ct_unb_var1()

FLA_Error FLA_Gemm_ct_unb_var1 ( FLA_Obj  alpha,
FLA_Obj  A,
FLA_Obj  B,
FLA_Obj  beta,
FLA_Obj  C 
)
16 {
17  FLA_Obj AT, A0,
18  AB, a1t,
19  A2;
20 
21  FLA_Obj CT, C0,
22  CB, c1t,
23  C2;
24 
25  FLA_Scal_external( beta, C );
26 
27  FLA_Part_2x1( A, &AT,
28  &AB, 0, FLA_TOP );
29 
30  FLA_Part_2x1( C, &CT,
31  &CB, 0, FLA_TOP );
32 
33  while ( FLA_Obj_length( AT ) < FLA_Obj_length( A ) ){
34 
35  FLA_Repart_2x1_to_3x1( AT, &A0,
36  /* ** */ /* *** */
37  &a1t,
38  AB, &A2, 1, FLA_BOTTOM );
39 
40  FLA_Repart_2x1_to_3x1( CT, &C0,
41  /* ** */ /* *** */
42  &c1t,
43  CB, &C2, 1, FLA_BOTTOM );
44 
45  /*------------------------------------------------------------*/
46 
47  /* c1t = a1t * B' + c1t */
48  /* c1t' = B * a1t' + c1t' */
49  FLA_Gemvc_external( FLA_NO_TRANSPOSE, FLA_CONJUGATE, alpha, B, a1t, FLA_ONE, c1t );
50 
51  /*------------------------------------------------------------*/
52 
53  FLA_Cont_with_3x1_to_2x1( &AT, A0,
54  a1t,
55  /* ** */ /* *** */
56  &AB, A2, FLA_TOP );
57 
58  FLA_Cont_with_3x1_to_2x1( &CT, C0,
59  c1t,
60  /* ** */ /* *** */
61  &CB, C2, FLA_TOP );
62 
63  }
64 
65  return FLA_SUCCESS;
66 }
FLA_Error FLA_Scal_external(FLA_Obj alpha, FLA_Obj A)
Definition: FLA_Scal_external.c:13
FLA_Error FLA_Gemvc_external(FLA_Trans transa, FLA_Conj conjx, FLA_Obj alpha, FLA_Obj A, FLA_Obj x, FLA_Obj beta, FLA_Obj y)
Definition: FLA_Gemvc_external.c:13
FLA_Obj FLA_ONE
Definition: FLA_Init.c:18
FLA_Error FLA_Cont_with_3x1_to_2x1(FLA_Obj *AT, FLA_Obj A0, FLA_Obj A1, FLA_Obj *AB, FLA_Obj A2, FLA_Side side)
Definition: FLA_View.c:428
FLA_Error FLA_Repart_2x1_to_3x1(FLA_Obj AT, FLA_Obj *A0, FLA_Obj *A1, FLA_Obj AB, FLA_Obj *A2, dim_t mb, FLA_Side side)
Definition: FLA_View.c:226
FLA_Error FLA_Part_2x1(FLA_Obj A, FLA_Obj *A1, FLA_Obj *A2, dim_t mb, FLA_Side side)
Definition: FLA_View.c:76
dim_t FLA_Obj_length(FLA_Obj obj)
Definition: FLA_Query.c:116
Definition: FLA_type_defs.h:159

References FLA_Cont_with_3x1_to_2x1(), FLA_Gemvc_external(), FLA_Obj_length(), FLA_ONE, FLA_Part_2x1(), FLA_Repart_2x1_to_3x1(), and FLA_Scal_external().

Referenced by FLA_Gemm_ct().