libflame revision_anchor
|
Go to the source code of this file.
Functions | |
FLA_Error | FLA_Accum_T_UT_fr_unb_var1 (FLA_Obj A, FLA_Obj t, FLA_Obj T) |
FLA_Error | FLA_Accum_T_UT_fr_blk_var2 (FLA_Obj A, FLA_Obj t, FLA_Obj T) |
FLA_Error | FLA_Accum_T_UT_fr_opt_var1 (FLA_Obj A, FLA_Obj t, FLA_Obj T) |
FLA_Error | FLA_Accum_T_UT_fr_ops_var1 (int m_A, int n_A, float *A, int rs_A, int cs_A, float *t, int inc_t, float *T, int rs_T, int cs_T) |
FLA_Error | FLA_Accum_T_UT_fr_opd_var1 (int m_A, int n_A, double *A, int rs_A, int cs_A, double *t, int inc_t, double *T, int rs_T, int cs_T) |
FLA_Error | FLA_Accum_T_UT_fr_opc_var1 (int m_A, int n_A, scomplex *A, int rs_A, int cs_A, scomplex *t, int inc_t, scomplex *T, int rs_T, int cs_T) |
FLA_Error | FLA_Accum_T_UT_fr_opz_var1 (int m_A, int n_A, dcomplex *A, int rs_A, int cs_A, dcomplex *t, int inc_t, dcomplex *T, int rs_T, int cs_T) |
References FLA_Accum_T_UT_fr_opt_var1(), FLA_Cont_with_1x3_to_1x2(), FLA_Cont_with_3x1_to_2x1(), FLA_Cont_with_3x3_to_2x2(), FLA_Merge_1x2(), FLA_Obj_length(), FLA_Obj_width(), FLA_Part_1x2(), FLA_Part_2x1(), FLA_Part_2x2(), FLA_Repart_1x2_to_1x3(), FLA_Repart_2x1_to_3x1(), and FLA_Repart_2x2_to_3x3().
Referenced by FLA_Accum_T_UT_internal().
{ FLA_Obj ATL, ATR, A00, A01, A02, ABL, ABR, A10, A11, A12, A20, A21, A22; FLA_Obj tT, t0, tB, t1, t2; FLA_Obj TL, TR, T0, T1, T2; FLA_Obj AR1; dim_t b_alg, b; b_alg = FLA_Obj_length( T ); FLA_Part_2x2( A, &ATL, &ATR, &ABL, &ABR, 0, 0, FLA_TL ); FLA_Part_2x1( t, &tT, &tB, 0, FLA_TOP ); FLA_Part_1x2( T, &TL, &TR, 0, FLA_LEFT ); while ( FLA_Obj_width( TL ) < FLA_Obj_width( T ) ){ b = min( FLA_Obj_width( TR ), b_alg ); FLA_Repart_2x2_to_3x3( ATL, /**/ ATR, &A00, /**/ &A01, &A02, /* ************* */ /* ******************** */ &A10, /**/ &A11, &A12, ABL, /**/ ABR, &A20, /**/ &A21, &A22, b, b, FLA_BR ); FLA_Repart_2x1_to_3x1( tT, &t0, /* ** */ /* ** */ &t1, tB, &t2, b, FLA_BOTTOM ); FLA_Repart_1x2_to_1x3( TL, /**/ TR, &T0, /**/ &T1, &T2, b, FLA_RIGHT ); /*------------------------------------------------------------*/ FLA_Merge_1x2( A11, A12, &AR1 ); //FLA_Accum_T_UT_fr_unb_var1( AR1, t1, T1 ); FLA_Accum_T_UT_fr_opt_var1( AR1, t1, T1 ); /*------------------------------------------------------------*/ FLA_Cont_with_3x3_to_2x2( &ATL, /**/ &ATR, A00, A01, /**/ A02, A10, A11, /**/ A12, /* ************** */ /* ****************** */ &ABL, /**/ &ABR, A20, A21, /**/ A22, FLA_TL ); FLA_Cont_with_3x1_to_2x1( &tT, t0, t1, /* ** */ /* ** */ &tB, t2, FLA_TOP ); FLA_Cont_with_1x3_to_1x2( &TL, /**/ &TR, T0, T1, /**/ T2, FLA_LEFT ); } return FLA_SUCCESS; }
FLA_Error FLA_Accum_T_UT_fr_opc_var1 | ( | int | m_A, |
int | n_A, | ||
scomplex * | A, | ||
int | rs_A, | ||
int | cs_A, | ||
scomplex * | t, | ||
int | inc_t, | ||
scomplex * | T, | ||
int | rs_T, | ||
int | cs_T | ||
) |
References bli_ccopyv(), bli_cgemv(), and FLA_ONE.
Referenced by FLA_Accum_T_UT_fr_opt_var1().
{ scomplex* buff_1 = FLA_COMPLEX_PTR( FLA_ONE ); int i; for ( i = 0; i < m_A; ++i ) { scomplex* a01 = buff_A + (i )*cs_A + (0 )*rs_A; scomplex* A02 = buff_A + (i+1)*cs_A + (0 )*rs_A; scomplex* a12t = buff_A + (i+1)*cs_A + (i )*rs_A; scomplex* tau1 = buff_t + (i )*inc_t; scomplex* tau11 = buff_T + (i )*cs_T + (i )*rs_T; scomplex* t01 = buff_T + (i )*cs_T + (0 )*rs_T; int n_ahead = n_A - i - 1; int m_behind = i; /*------------------------------------------------------------*/ // FLA_Copy_external( tau1, tau11 ); *tau11 = *tau1; // FLA_Copyt_external( FLA_CONJ_NO_TRANSPOSE, a01, t01 ); bli_ccopyv( BLIS_CONJUGATE, m_behind, a01, rs_A, t01, rs_T ); // FLA_Gemvc_external( FLA_CONJ_NO_TRANSPOSE, FLA_NO_CONJUGATE, // FLA_ONE, A02, a12t, FLA_ONE, t01 ); bli_cgemv( BLIS_CONJ_NO_TRANSPOSE, BLIS_NO_CONJUGATE, m_behind, n_ahead, buff_1, A02, rs_A, cs_A, a12t, cs_A, buff_1, t01, rs_T ); /*------------------------------------------------------------*/ } return FLA_SUCCESS; }
FLA_Error FLA_Accum_T_UT_fr_opd_var1 | ( | int | m_A, |
int | n_A, | ||
double * | A, | ||
int | rs_A, | ||
int | cs_A, | ||
double * | t, | ||
int | inc_t, | ||
double * | T, | ||
int | rs_T, | ||
int | cs_T | ||
) |
References bli_dcopyv(), bli_dgemv(), and FLA_ONE.
Referenced by FLA_Accum_T_UT_fr_opt_var1().
{ double* buff_1 = FLA_DOUBLE_PTR( FLA_ONE ); int i; for ( i = 0; i < m_A; ++i ) { double* a01 = buff_A + (i )*cs_A + (0 )*rs_A; double* A02 = buff_A + (i+1)*cs_A + (0 )*rs_A; double* a12t = buff_A + (i+1)*cs_A + (i )*rs_A; double* tau1 = buff_t + (i )*inc_t; double* tau11 = buff_T + (i )*cs_T + (i )*rs_T; double* t01 = buff_T + (i )*cs_T + (0 )*rs_T; int n_ahead = n_A - i - 1; int m_behind = i; /*------------------------------------------------------------*/ // FLA_Copy_external( tau1, tau11 ); *tau11 = *tau1; // FLA_Copyt_external( FLA_CONJ_NO_TRANSPOSE, a01, t01 ); bli_dcopyv( BLIS_CONJUGATE, m_behind, a01, rs_A, t01, rs_T ); // FLA_Gemvc_external( FLA_CONJ_NO_TRANSPOSE, FLA_NO_CONJUGATE, // FLA_ONE, A02, a12t, FLA_ONE, t01 ); bli_dgemv( BLIS_CONJ_NO_TRANSPOSE, BLIS_NO_CONJUGATE, m_behind, n_ahead, buff_1, A02, rs_A, cs_A, a12t, cs_A, buff_1, t01, rs_T ); /*------------------------------------------------------------*/ } return FLA_SUCCESS; }
FLA_Error FLA_Accum_T_UT_fr_ops_var1 | ( | int | m_A, |
int | n_A, | ||
float * | A, | ||
int | rs_A, | ||
int | cs_A, | ||
float * | t, | ||
int | inc_t, | ||
float * | T, | ||
int | rs_T, | ||
int | cs_T | ||
) |
References bli_scopyv(), bli_sgemv(), and FLA_ONE.
Referenced by FLA_Accum_T_UT_fr_opt_var1().
{ float* buff_1 = FLA_FLOAT_PTR( FLA_ONE ); int i; for ( i = 0; i < m_A; ++i ) { float* a01 = buff_A + (i )*cs_A + (0 )*rs_A; float* A02 = buff_A + (i+1)*cs_A + (0 )*rs_A; float* a12t = buff_A + (i+1)*cs_A + (i )*rs_A; float* tau1 = buff_t + (i )*inc_t; float* tau11 = buff_T + (i )*cs_T + (i )*rs_T; float* t01 = buff_T + (i )*cs_T + (0 )*rs_T; int n_ahead = n_A - i - 1; int m_behind = i; /*------------------------------------------------------------*/ // FLA_Copy_external( tau1, tau11 ); *tau11 = *tau1; // FLA_Copyt_external( FLA_CONJ_NO_TRANSPOSE, a01, t01 ); bli_scopyv( BLIS_CONJUGATE, m_behind, a01, rs_A, t01, rs_T ); // FLA_Gemvc_external( FLA_CONJ_NO_TRANSPOSE, FLA_NO_CONJUGATE, // FLA_ONE, A02, a12t, FLA_ONE, t01 ); bli_sgemv( BLIS_CONJ_NO_TRANSPOSE, BLIS_NO_CONJUGATE, m_behind, n_ahead, buff_1, A02, rs_A, cs_A, a12t, cs_A, buff_1, t01, rs_T ); /*------------------------------------------------------------*/ } return FLA_SUCCESS; }
References FLA_Accum_T_UT_fr_opc_var1(), FLA_Accum_T_UT_fr_opd_var1(), FLA_Accum_T_UT_fr_ops_var1(), FLA_Accum_T_UT_fr_opz_var1(), FLA_Obj_col_stride(), FLA_Obj_datatype(), FLA_Obj_length(), FLA_Obj_row_stride(), FLA_Obj_vector_inc(), and FLA_Obj_width().
Referenced by FLA_Accum_T_UT_fr_blk_var2().
{ FLA_Datatype datatype; int m_A, n_A; int rs_A, cs_A; int inc_t; int rs_T, cs_T; datatype = FLA_Obj_datatype( A ); m_A = FLA_Obj_length( A ); n_A = FLA_Obj_width( A ); rs_A = FLA_Obj_row_stride( A ); cs_A = FLA_Obj_col_stride( A ); inc_t = FLA_Obj_vector_inc( t ); rs_T = FLA_Obj_row_stride( T ); cs_T = FLA_Obj_col_stride( T ); switch ( datatype ) { case FLA_FLOAT: { float* buff_A = FLA_FLOAT_PTR( A ); float* buff_t = FLA_FLOAT_PTR( t ); float* buff_T = FLA_FLOAT_PTR( T ); FLA_Accum_T_UT_fr_ops_var1( m_A, n_A, buff_A, rs_A, cs_A, buff_t, inc_t, buff_T, rs_T, cs_T ); break; } case FLA_DOUBLE: { double* buff_A = FLA_DOUBLE_PTR( A ); double* buff_t = FLA_DOUBLE_PTR( t ); double* buff_T = FLA_DOUBLE_PTR( T ); FLA_Accum_T_UT_fr_opd_var1( m_A, n_A, buff_A, rs_A, cs_A, buff_t, inc_t, buff_T, rs_T, cs_T ); break; } case FLA_COMPLEX: { scomplex* buff_A = FLA_COMPLEX_PTR( A ); scomplex* buff_t = FLA_COMPLEX_PTR( t ); scomplex* buff_T = FLA_COMPLEX_PTR( T ); FLA_Accum_T_UT_fr_opc_var1( m_A, n_A, buff_A, rs_A, cs_A, buff_t, inc_t, buff_T, rs_T, cs_T ); break; } case FLA_DOUBLE_COMPLEX: { dcomplex* buff_A = FLA_DOUBLE_COMPLEX_PTR( A ); dcomplex* buff_t = FLA_DOUBLE_COMPLEX_PTR( t ); dcomplex* buff_T = FLA_DOUBLE_COMPLEX_PTR( T ); FLA_Accum_T_UT_fr_opz_var1( m_A, n_A, buff_A, rs_A, cs_A, buff_t, inc_t, buff_T, rs_T, cs_T ); break; } } return FLA_SUCCESS; }
FLA_Error FLA_Accum_T_UT_fr_opz_var1 | ( | int | m_A, |
int | n_A, | ||
dcomplex * | A, | ||
int | rs_A, | ||
int | cs_A, | ||
dcomplex * | t, | ||
int | inc_t, | ||
dcomplex * | T, | ||
int | rs_T, | ||
int | cs_T | ||
) |
References bli_zcopyv(), bli_zgemv(), and FLA_ONE.
Referenced by FLA_Accum_T_UT_fr_opt_var1().
{ dcomplex* buff_1 = FLA_DOUBLE_COMPLEX_PTR( FLA_ONE ); int i; for ( i = 0; i < m_A; ++i ) { dcomplex* a01 = buff_A + (i )*cs_A + (0 )*rs_A; dcomplex* A02 = buff_A + (i+1)*cs_A + (0 )*rs_A; dcomplex* a12t = buff_A + (i+1)*cs_A + (i )*rs_A; dcomplex* tau1 = buff_t + (i )*inc_t; dcomplex* tau11 = buff_T + (i )*cs_T + (i )*rs_T; dcomplex* t01 = buff_T + (i )*cs_T + (0 )*rs_T; int n_ahead = n_A - i - 1; int m_behind = i; /*------------------------------------------------------------*/ // FLA_Copy_external( tau1, tau11 ); *tau11 = *tau1; // FLA_Copyt_external( FLA_CONJ_NO_TRANSPOSE, a01, t01 ); bli_zcopyv( BLIS_CONJUGATE, m_behind, a01, rs_A, t01, rs_T ); // FLA_Gemvc_external( FLA_CONJ_NO_TRANSPOSE, FLA_NO_CONJUGATE, // FLA_ONE, A02, a12t, FLA_ONE, t01 ); bli_zgemv( BLIS_CONJ_NO_TRANSPOSE, BLIS_NO_CONJUGATE, m_behind, n_ahead, buff_1, A02, rs_A, cs_A, a12t, cs_A, buff_1, t01, rs_T ); /*------------------------------------------------------------*/ } return FLA_SUCCESS; }
References FLA_Cont_with_3x1_to_2x1(), FLA_Cont_with_3x3_to_2x2(), FLA_Copy_external(), FLA_Copyt_external(), FLA_Gemvc_external(), FLA_Obj_min_dim(), FLA_ONE, FLA_Part_2x1(), FLA_Part_2x2(), FLA_Repart_2x1_to_3x1(), and FLA_Repart_2x2_to_3x3().
{ FLA_Obj ATL, ATR, A00, a01, A02, ABL, ABR, a10t, alpha11, a12t, A20, a21, A22; FLA_Obj tauT, tau0, tauB, tau1, tau2; FLA_Obj TTL, TTR, T00, t01, T02, TBL, TBR, t10t, tau11, t12t, T20, t21, T22; FLA_Part_2x2( A, &ATL, &ATR, &ABL, &ABR, 0, 0, FLA_TL ); FLA_Part_2x1( tau, &tauT, &tauB, 0, FLA_TOP ); FLA_Part_2x2( T, &TTL, &TTR, &TBL, &TBR, 0, 0, FLA_TL ); while ( FLA_Obj_min_dim( ABR ) > 0 ){ FLA_Repart_2x2_to_3x3( ATL, /**/ ATR, &A00, /**/ &a01, &A02, /* ************* */ /* ************************** */ &a10t, /**/ &alpha11, &a12t, ABL, /**/ ABR, &A20, /**/ &a21, &A22, 1, 1, FLA_BR ); FLA_Repart_2x1_to_3x1( tauT, &tau0, /* ** */ /* ** */ &tau1, tauB, &tau2, 1, FLA_BOTTOM ); FLA_Repart_2x2_to_3x3( TTL, /**/ TTR, &T00, /**/ &t01, &T02, /* ************* */ /* ************************ */ &t10t, /**/ &tau11, &t12t, TBL, /**/ TBR, &T20, /**/ &t21, &T22, 1, 1, FLA_BR ); /*------------------------------------------------------------*/ // tau11 = tau1; FLA_Copy_external( tau1, tau11 ); // t01 = conj(a01) + conj(A02) * u12t^T; FLA_Copyt_external( FLA_CONJ_NO_TRANSPOSE, a01, t01 ); FLA_Gemvc_external( FLA_CONJ_NO_TRANSPOSE, FLA_NO_CONJUGATE, FLA_ONE, A02, a12t, FLA_ONE, t01 ); /*------------------------------------------------------------*/ FLA_Cont_with_3x3_to_2x2( &ATL, /**/ &ATR, A00, a01, /**/ A02, a10t, alpha11, /**/ a12t, /* ************** */ /* ************************ */ &ABL, /**/ &ABR, A20, a21, /**/ A22, FLA_TL ); FLA_Cont_with_3x1_to_2x1( &tauT, tau0, tau1, /* ** */ /* ** */ &tauB, tau2, FLA_TOP ); FLA_Cont_with_3x3_to_2x2( &TTL, /**/ &TTR, T00, t01, /**/ T02, t10t, tau11, /**/ t12t, /* ************** */ /* ********************** */ &TBL, /**/ &TBR, T20, t21, /**/ T22, FLA_TL ); } return FLA_SUCCESS; }