libflame  revision_anchor
Functions
FLA_SA_LU_blk.c File Reference

(r)

Functions

FLA_Error FLA_SA_LU_blk (FLA_Obj U, FLA_Obj D, FLA_Obj p, FLA_Obj L, dim_t nb_alg)
 

Function Documentation

◆ FLA_SA_LU_blk()

FLA_Error FLA_SA_LU_blk ( FLA_Obj  U,
FLA_Obj  D,
FLA_Obj  p,
FLA_Obj  L,
dim_t  nb_alg 
)

References FLA_Cont_with_1x3_to_1x2(), FLA_Cont_with_3x1_to_2x1(), FLA_Cont_with_3x3_to_2x2(), FLA_Gemm_external(), FLA_MINUS_ONE, FLA_Obj_length(), FLA_ONE, FLA_Part_1x2(), FLA_Part_2x1(), FLA_Part_2x2(), FLA_Repart_1x2_to_1x3(), FLA_Repart_2x1_to_3x1(), FLA_Repart_2x2_to_3x3(), FLA_SA_Apply_pivots(), FLA_SA_LU_unb(), and FLA_Trsm_external().

Referenced by FLA_SA_LU_task().

15 {
16  FLA_Obj UTL, UTR, U00, U01, U02,
17  UBL, UBR, U10, U11, U12,
18  U20, U21, U22;
19 
20  FLA_Obj DL, DR, D0, D1, D2;
21 
22  FLA_Obj pT, p0,
23  pB, p1,
24  p2;
25 
26  FLA_Obj LT, L0,
27  LB, L1,
28  L2;
29 
30  FLA_Obj L1_sqr, L1_rest;
31 
32  dim_t b;
33 
34  FLA_Part_2x2( U, &UTL, &UTR,
35  &UBL, &UBR, 0, 0, FLA_TL );
36 
37  FLA_Part_1x2( D, &DL, &DR, 0, FLA_LEFT );
38 
39  FLA_Part_2x1( p, &pT,
40  &pB, 0, FLA_TOP );
41 
42  FLA_Part_2x1( L, &LT,
43  &LB, 0, FLA_TOP );
44 
45  while ( FLA_Obj_length( UTL ) < FLA_Obj_length( U ) )
46  {
47  b = min( FLA_Obj_length( UBR ), nb_alg );
48 
49  FLA_Repart_2x2_to_3x3( UTL, /**/ UTR, &U00, /**/ &U01, &U02,
50  /* ************* */ /* ******************** */
51  &U10, /**/ &U11, &U12,
52  UBL, /**/ UBR, &U20, /**/ &U21, &U22,
53  b, b, FLA_BR );
54 
55  FLA_Repart_1x2_to_1x3( DL, /**/ DR, &D0, /**/ &D1, &D2,
56  b, FLA_RIGHT );
57 
58  FLA_Repart_2x1_to_3x1( pT, &p0,
59  /* ** */ /* ** */
60  &p1,
61  pB, &p2, b, FLA_BOTTOM );
62 
63  FLA_Repart_2x1_to_3x1( LT, &L0,
64  /* ** */ /* ** */
65  &L1,
66  LB, &L2, b, FLA_BOTTOM );
67 
68  /*------------------------------------------------------------*/
69 
70  FLA_Part_1x2( L1, &L1_sqr, &L1_rest, b, FLA_LEFT );
71 
72 
73  FLA_SA_LU_unb( U11,
74  D1, p1, L1_sqr );
75 
77  D2, p1 );
78 
79  FLA_Trsm_external( FLA_LEFT, FLA_LOWER_TRIANGULAR,
80  FLA_NO_TRANSPOSE, FLA_UNIT_DIAG,
81  FLA_ONE, L1_sqr, U12 );
82 
83  FLA_Gemm_external( FLA_NO_TRANSPOSE, FLA_NO_TRANSPOSE,
84  FLA_MINUS_ONE, D1, U12, FLA_ONE, D2 );
85 
86  /*------------------------------------------------------------*/
87 
88  FLA_Cont_with_3x3_to_2x2( &UTL, /**/ &UTR, U00, U01, /**/ U02,
89  U10, U11, /**/ U12,
90  /* ************** */ /* ****************** */
91  &UBL, /**/ &UBR, U20, U21, /**/ U22,
92  FLA_TL );
93 
94  FLA_Cont_with_1x3_to_1x2( &DL, /**/ &DR, D0, D1, /**/ D2,
95  FLA_LEFT );
96 
97  FLA_Cont_with_3x1_to_2x1( &pT, p0,
98  p1,
99  /* ** */ /* ** */
100  &pB, p2, FLA_TOP );
101 
102  FLA_Cont_with_3x1_to_2x1( &LT, L0,
103  L1,
104  /* ** */ /* ** */
105  &LB, L2, FLA_TOP );
106  }
107 
108  return FLA_SUCCESS;
109 }
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_Repart_1x2_to_1x3(FLA_Obj AL, FLA_Obj AR, FLA_Obj *A0, FLA_Obj *A1, FLA_Obj *A2, dim_t nb, FLA_Side side)
Definition: FLA_View.c:267
unsigned long dim_t
Definition: FLA_type_defs.h:71
FLA_Error FLA_SA_Apply_pivots(FLA_Obj C, FLA_Obj E, FLA_Obj p)
Definition: FLA_SA_Apply_pivots.c:13
FLA_Obj FLA_MINUS_ONE
Definition: FLA_Init.c:22
FLA_Error FLA_Repart_2x2_to_3x3(FLA_Obj ATL, FLA_Obj ATR, FLA_Obj *A00, FLA_Obj *A01, FLA_Obj *A02, FLA_Obj *A10, FLA_Obj *A11, FLA_Obj *A12, FLA_Obj ABL, FLA_Obj ABR, FLA_Obj *A20, FLA_Obj *A21, FLA_Obj *A22, dim_t mb, dim_t nb, FLA_Quadrant quadrant)
Definition: FLA_View.c:142
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_Part_2x2(FLA_Obj A, FLA_Obj *A11, FLA_Obj *A12, FLA_Obj *A21, FLA_Obj *A22, dim_t mb, dim_t nb, FLA_Quadrant quadrant)
Definition: FLA_View.c:17
FLA_Obj FLA_ONE
Definition: FLA_Init.c:18
Definition: FLA_type_defs.h:158
FLA_Error FLA_Cont_with_1x3_to_1x2(FLA_Obj *AL, FLA_Obj *AR, FLA_Obj A0, FLA_Obj A1, FLA_Obj A2, FLA_Side side)
Definition: FLA_View.c:475
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
FLA_Error FLA_Cont_with_3x3_to_2x2(FLA_Obj *ATL, FLA_Obj *ATR, FLA_Obj A00, FLA_Obj A01, FLA_Obj A02, FLA_Obj A10, FLA_Obj A11, FLA_Obj A12, FLA_Obj *ABL, FLA_Obj *ABR, FLA_Obj A20, FLA_Obj A21, FLA_Obj A22, FLA_Quadrant quadrant)
Definition: FLA_View.c:304
FLA_Error FLA_Part_1x2(FLA_Obj A, FLA_Obj *A1, FLA_Obj *A2, dim_t nb, FLA_Side side)
Definition: FLA_View.c:110
FLA_Error FLA_Trsm_external(FLA_Side side, FLA_Uplo uplo, FLA_Trans trans, FLA_Diag diag, FLA_Obj alpha, FLA_Obj A, FLA_Obj B)
Definition: FLA_Trsm_external.c:13
dim_t FLA_Obj_length(FLA_Obj obj)
Definition: FLA_Query.c:116
FLA_Error FLA_Gemm_external(FLA_Trans transa, FLA_Trans transb, FLA_Obj alpha, FLA_Obj A, FLA_Obj B, FLA_Obj beta, FLA_Obj C)
Definition: FLA_Gemm_external.c:13
FLA_Error FLA_SA_LU_unb(FLA_Obj U, FLA_Obj D, FLA_Obj p, FLA_Obj L)
Definition: FLA_SA_LU_unb.c:13