FLA_Chol_u_blk_var2.c File Reference

(r)


Functions

FLA_Error FLA_Chol_u_blk_var2 (FLA_Obj A, fla_chol_t *cntl)

Function Documentation

FLA_Error FLA_Chol_u_blk_var2 ( FLA_Obj  A,
fla_chol_t cntl 
)

References FLA_Chol_internal(), FLA_Cont_with_3x3_to_2x2(), FLA_Determine_blocksize(), FLA_Gemm_internal(), FLA_Herk_internal(), FLA_MINUS_ONE, FLA_Obj_is_real(), FLA_Obj_length(), FLA_ONE, FLA_Part_2x2(), FLA_Repart_2x2_to_3x3(), FLA_Syrk_internal(), and FLA_Trsm_internal().

Referenced by FLA_Chol_u().

00038 {
00039   FLA_Obj ATL,   ATR,      A00, A01, A02,
00040           ABL,   ABR,      A10, A11, A12,
00041                            A20, A21, A22;
00042 
00043   dim_t b;
00044 
00045   int value = 0;
00046 
00047   FLA_Part_2x2( A,    &ATL, &ATR,
00048                       &ABL, &ABR,     0, 0, FLA_TL );
00049 
00050   while ( FLA_Obj_length( ATL ) < FLA_Obj_length( A ) ){
00051 
00052     b = FLA_Determine_blocksize( ABR, FLA_BR, FLA_Cntl_blocksize( cntl ) );
00053 
00054     FLA_Repart_2x2_to_3x3( ATL, /**/ ATR,       &A00, /**/ &A01, &A02,
00055                         /* ************* */   /* ******************** */
00056                                                 &A10, /**/ &A11, &A12,
00057                            ABL, /**/ ABR,       &A20, /**/ &A21, &A22,
00058                            b, b, FLA_BR );
00059 
00060     /*------------------------------------------------------------*/
00061 
00062     if ( FLA_Obj_is_real( A ) )
00063     {
00064       /* A11 = A11 - A01' * A01; */
00065       FLA_Syrk_internal( FLA_UPPER_TRIANGULAR, FLA_TRANSPOSE,
00066                          FLA_MINUS_ONE, A01, FLA_ONE, A11,
00067                          FLA_Cntl_sub_syrk( cntl ) );
00068 
00069       /* A12 = A12 - A01' * A02; */
00070       FLA_Gemm_internal( FLA_TRANSPOSE, FLA_NO_TRANSPOSE,
00071                          FLA_MINUS_ONE, A01, A02, FLA_ONE, A12,
00072                          FLA_Cntl_sub_gemm( cntl ) );
00073 
00074       /* A11 = chol( A11 ); */
00075       value = FLA_Chol_internal( FLA_UPPER_TRIANGULAR, A11,
00076                                  FLA_Cntl_sub_chol( cntl ) );
00077 
00078       if ( value != FLA_SUCCESS )
00079         return ( FLA_Obj_length( A00 ) + value );
00080 
00081       /* A12 = inv( triu( A11 )' ) * A12; */
00082       FLA_Trsm_internal( FLA_LEFT, FLA_UPPER_TRIANGULAR,
00083                          FLA_TRANSPOSE, FLA_NONUNIT_DIAG,
00084                          FLA_ONE, A11, A12,
00085                          FLA_Cntl_sub_trsm( cntl ) );
00086     }
00087     else
00088     {
00089       /* A11 = A11 - A01' * A01; */
00090       FLA_Herk_internal( FLA_UPPER_TRIANGULAR, FLA_CONJ_TRANSPOSE,
00091                          FLA_MINUS_ONE, A01, FLA_ONE, A11,
00092                          FLA_Cntl_sub_herk( cntl ) );
00093 
00094       /* A12 = A12 - A02' * A01; */
00095       FLA_Gemm_internal( FLA_CONJ_TRANSPOSE, FLA_NO_TRANSPOSE,
00096                          FLA_MINUS_ONE, A01, A02, FLA_ONE, A12,
00097                          FLA_Cntl_sub_gemm( cntl ) );
00098 
00099       /* A11 = chol( A11 ); */
00100       value = FLA_Chol_internal( FLA_UPPER_TRIANGULAR, A11,
00101                                  FLA_Cntl_sub_chol( cntl ) );
00102 
00103       if ( value != FLA_SUCCESS )
00104         return ( FLA_Obj_length( A00 ) + value );
00105 
00106       /* A12 = inv( triu( A11 )' ) * A12; */
00107       FLA_Trsm_internal( FLA_LEFT, FLA_UPPER_TRIANGULAR,
00108                          FLA_CONJ_TRANSPOSE, FLA_NONUNIT_DIAG,
00109                          FLA_ONE, A11, A12,
00110                          FLA_Cntl_sub_trsm( cntl ) );
00111     }
00112 
00113     /*------------------------------------------------------------*/
00114 
00115     FLA_Cont_with_3x3_to_2x2( &ATL, /**/ &ATR,       A00, A01, /**/ A02,
00116                                                      A10, A11, /**/ A12,
00117                             /* ************** */  /* ****************** */
00118                               &ABL, /**/ &ABR,       A20, A21, /**/ A22,
00119                               FLA_TL );
00120   }
00121 
00122   return value;
00123 }


Generated on Mon Jul 6 05:45:56 2009 for libflame by  doxygen 1.5.9