1 /*
2
3 Copyright (C) 2014, The University of Texas at Austin
4
5 This file is part of libflame and is available under the 3-Clause
6 BSD license, which can be found in the LICENSE file at the top-level
7 directory, or at http://opensource.org/licenses/BSD-3-Clause
8
9 */
10
11 #include "FLAME.h"
12
FLA_Eig_gest_il_unb_var5(FLA_Obj A,FLA_Obj Y,FLA_Obj B)13 FLA_Error FLA_Eig_gest_il_unb_var5( FLA_Obj A, FLA_Obj Y, FLA_Obj B )
14 {
15 FLA_Obj ATL, ATR, A00, a01, A02,
16 ABL, ABR, a10t, alpha11, a12t,
17 A20, a21, A22;
18
19 FLA_Obj BTL, BTR, B00, b01, B02,
20 BBL, BBR, b10t, beta11, b12t,
21 B20, b21, B22;
22
23 //FLA_Obj yT, y01,
24 // yB, psi11,
25 // y21;
26
27 //FLA_Obj y21_l, y21_r;
28
29 FLA_Obj psi11, y12t,
30 y21, Y22;
31
32 FLA_Part_2x2( A, &ATL, &ATR,
33 &ABL, &ABR, 0, 0, FLA_TL );
34
35 FLA_Part_2x2( B, &BTL, &BTR,
36 &BBL, &BBR, 0, 0, FLA_TL );
37
38 //FLA_Part_2x1( Y, &yT,
39 // &yB, 0, FLA_TOP );
40
41 FLA_Part_2x2( Y, &psi11, &y12t,
42 &y21, &Y22, 1, 1, FLA_TL );
43
44 while ( FLA_Obj_length( ATL ) < FLA_Obj_length( A ) ){
45
46 FLA_Repart_2x2_to_3x3( ATL, /**/ ATR, &A00, /**/ &a01, &A02,
47 /* ************* */ /* ************************** */
48 &a10t, /**/ &alpha11, &a12t,
49 ABL, /**/ ABR, &A20, /**/ &a21, &A22,
50 1, 1, FLA_BR );
51
52 FLA_Repart_2x2_to_3x3( BTL, /**/ BTR, &B00, /**/ &b01, &B02,
53 /* ************* */ /* ************************* */
54 &b10t, /**/ &beta11, &b12t,
55 BBL, /**/ BBR, &B20, /**/ &b21, &B22,
56 1, 1, FLA_BR );
57
58 //FLA_Repart_2x1_to_3x1( yT, &y01,
59 // /* ** */ /* ***** */
60 // &psi11,
61 // yB, &y21, 1, FLA_BOTTOM );
62
63 /*------------------------------------------------------------*/
64
65 //FLA_Part_1x2( y21, &y21_l, &y21_r, 1, FLA_LEFT );
66
67 // alpha11 = inv(beta11) * alpha11 * inv(conj(beta11));
68 // = inv(beta11) * alpha11 * inv(beta11);
69 FLA_Inv_scal_external( beta11, alpha11 );
70 FLA_Inv_scal_external( beta11, alpha11 );
71
72 //// y21 = b21 * alpha11;
73 //FLA_Copy_external( b21, y21_l );
74 //FLA_Scal_external( alpha11, y21_l );
75 // psi11 = - 1/2 * alpha11;
76 FLA_Copy_external( alpha11, psi11 );
77 FLA_Scal_external( FLA_MINUS_ONE_HALF, psi11 );
78
79 // a21 = a21 * inv(conj(beta11));
80 // = a21 * inv(beta11);
81 FLA_Inv_scal_external( beta11, a21 );
82
83 //// a21 = a21 - 1/2 * y21;
84 //FLA_Axpy_external( FLA_MINUS_ONE_HALF, y21_l, a21 );
85 // a21 = a21 - 1/2 * alpha11 * b21;
86 FLA_Axpy_external( psi11, b21, a21 );
87
88 // A22 = A22 - a21 * b21' - b21 * a21';
89 FLA_Her2c_external( FLA_LOWER_TRIANGULAR, FLA_NO_CONJUGATE,
90 FLA_MINUS_ONE, a21, b21, A22 );
91
92 //// a21 = a21 - 1/2 * y21;
93 //FLA_Axpy_external( FLA_MINUS_ONE_HALF, y21_l, a21 );
94 // a21 = a21 - 1/2 * alpha11 * b21;
95 FLA_Axpy_external( psi11, b21, a21 );
96
97 // a21 = inv( tril( B22 ) ) * a21;
98 FLA_Trsv_external( FLA_LOWER_TRIANGULAR, FLA_NO_TRANSPOSE, FLA_NONUNIT_DIAG,
99 B22, a21 );
100
101 /*------------------------------------------------------------*/
102
103 FLA_Cont_with_3x3_to_2x2( &ATL, /**/ &ATR, A00, a01, /**/ A02,
104 a10t, alpha11, /**/ a12t,
105 /* ************** */ /* ************************ */
106 &ABL, /**/ &ABR, A20, a21, /**/ A22,
107 FLA_TL );
108
109 FLA_Cont_with_3x3_to_2x2( &BTL, /**/ &BTR, B00, b01, /**/ B02,
110 b10t, beta11, /**/ b12t,
111 /* ************** */ /* *********************** */
112 &BBL, /**/ &BBR, B20, b21, /**/ B22,
113 FLA_TL );
114
115 //FLA_Cont_with_3x1_to_2x1( &yT, y01,
116 // psi11,
117 // /* ** */ /* ***** */
118 // &yB, y21, FLA_TOP );
119 }
120
121 return FLA_SUCCESS;
122 }
123
124