|
libflame
revision_anchor
|
Functions | |
| FLA_Error | FLA_Sylv_nh_blk_var18 (FLA_Obj isgn, FLA_Obj A, FLA_Obj B, FLA_Obj C, FLA_Obj scale, fla_sylv_t *cntl) |
| FLA_Error FLA_Sylv_nh_blk_var18 | ( | FLA_Obj | isgn, |
| FLA_Obj | A, | ||
| FLA_Obj | B, | ||
| FLA_Obj | C, | ||
| FLA_Obj | scale, | ||
| fla_sylv_t * | cntl | ||
| ) |
References FLA_Cont_with_1x3_to_1x2(), FLA_Cont_with_3x3_to_2x2(), FLA_Determine_blocksize(), FLA_Gemm_internal(), FLA_Obj_length(), FLA_ONE, FLA_Part_1x2(), FLA_Part_2x2(), FLA_Repart_1x2_to_1x3(), FLA_Repart_2x2_to_3x3(), and FLA_Sylv_internal().
Referenced by FLA_Sylv_nh().
{
FLA_Obj BTL, BTR, B00, B01, B02,
BBL, BBR, B10, B11, B12,
B20, B21, B22;
FLA_Obj CL, CR, C0, C1, C2;
dim_t b;
FLA_Part_2x2( B, &BTL, &BTR,
&BBL, &BBR, 0, 0, FLA_BR );
FLA_Part_1x2( C, &CL, &CR, 0, FLA_RIGHT );
while ( FLA_Obj_length( BBR ) < FLA_Obj_length( B ) ){
b = FLA_Determine_blocksize( CL, FLA_LEFT, FLA_Cntl_blocksize( cntl ) );
FLA_Repart_2x2_to_3x3( BTL, /**/ BTR, &B00, &B01, /**/ &B02,
&B10, &B11, /**/ &B12,
/* ************* */ /* ******************** */
BBL, /**/ BBR, &B20, &B21, /**/ &B22,
b, b, FLA_TL );
FLA_Repart_1x2_to_1x3( CL, /**/ CR, &C0, &C1, /**/ &C2,
b, FLA_LEFT );
// Loop Invariant:
// CL =
// CR =
/*------------------------------------------------------------*/
// C1 = sylv( A, B11', C1 );
FLA_Sylv_internal( FLA_NO_TRANSPOSE, FLA_CONJ_TRANSPOSE,
isgn, A, B11, C1, scale,
FLA_Cntl_sub_sylv1( cntl ) );
// C0 = C0 -/+ C1 * B01';
FLA_Gemm_internal( FLA_NO_TRANSPOSE, FLA_CONJ_TRANSPOSE,
FLA_NEGATE( isgn ), C1, B01, FLA_ONE, C0,
FLA_Cntl_sub_gemm1( cntl ) );
/*------------------------------------------------------------*/
FLA_Cont_with_3x3_to_2x2( &BTL, /**/ &BTR, B00, /**/ B01, B02,
/* ************** */ /* ****************** */
B10, /**/ B11, B12,
&BBL, /**/ &BBR, B20, /**/ B21, B22,
FLA_BR );
FLA_Cont_with_1x3_to_1x2( &CL, /**/ &CR, C0, /**/ C1, C2,
FLA_RIGHT );
}
return FLA_SUCCESS;
}
1.7.6.1