libflame revision_anchor
|
Go to the source code of this file.
Functions | |
FLA_Error | FLA_Tridiag_UT_l_unb_var1 (FLA_Obj A, FLA_Obj T) |
FLA_Error | FLA_Tridiag_UT_l_blk_var1 (FLA_Obj A, FLA_Obj T) |
FLA_Error | FLA_Tridiag_UT_l_step_unb_var1 (FLA_Obj A, FLA_Obj T) |
FLA_Error | FLA_Tridiag_UT_l_unb_var2 (FLA_Obj A, FLA_Obj T) |
FLA_Error | FLA_Tridiag_UT_l_blk_var2 (FLA_Obj A, FLA_Obj T) |
FLA_Error | FLA_Tridiag_UT_l_step_unb_var2 (FLA_Obj A, FLA_Obj T) |
FLA_Error | FLA_Tridiag_UT_l_unb_var3 (FLA_Obj A, FLA_Obj T) |
FLA_Error | FLA_Tridiag_UT_l_blk_var3 (FLA_Obj A, FLA_Obj T) |
FLA_Error | FLA_Tridiag_UT_l_step_unb_var3 (FLA_Obj A, FLA_Obj Z, FLA_Obj T) |
FLA_Error | FLA_Tridiag_UT_l_opt_var1 (FLA_Obj A, FLA_Obj T) |
FLA_Error | FLA_Tridiag_UT_l_step_opt_var1 (FLA_Obj A, FLA_Obj T) |
FLA_Error | FLA_Tridiag_UT_l_step_ops_var1 (int m_A, int m_T, float *buff_A, int rs_A, int cs_A, float *buff_T, int rs_T, int cs_T) |
FLA_Error | FLA_Tridiag_UT_l_step_opd_var1 (int m_A, int m_T, double *buff_A, int rs_A, int cs_A, double *buff_T, int rs_T, int cs_T) |
FLA_Error | FLA_Tridiag_UT_l_step_opc_var1 (int m_A, int m_T, scomplex *buff_A, int rs_A, int cs_A, scomplex *buff_T, int rs_T, int cs_T) |
FLA_Error | FLA_Tridiag_UT_l_step_opz_var1 (int m_A, int m_T, dcomplex *buff_A, int rs_A, int cs_A, dcomplex *buff_T, int rs_T, int cs_T) |
FLA_Error | FLA_Tridiag_UT_l_opt_var2 (FLA_Obj A, FLA_Obj T) |
FLA_Error | FLA_Tridiag_UT_l_step_opt_var2 (FLA_Obj A, FLA_Obj T) |
FLA_Error | FLA_Tridiag_UT_l_step_ops_var2 (int m_A, int m_T, float *buff_A, int rs_A, int cs_A, float *buff_T, int rs_T, int cs_T) |
FLA_Error | FLA_Tridiag_UT_l_step_opd_var2 (int m_A, int m_T, double *buff_A, int rs_A, int cs_A, double *buff_T, int rs_T, int cs_T) |
FLA_Error | FLA_Tridiag_UT_l_step_opc_var2 (int m_A, int m_T, scomplex *buff_A, int rs_A, int cs_A, scomplex *buff_T, int rs_T, int cs_T) |
FLA_Error | FLA_Tridiag_UT_l_step_opz_var2 (int m_A, int m_T, dcomplex *buff_A, int rs_A, int cs_A, dcomplex *buff_T, int rs_T, int cs_T) |
FLA_Error | FLA_Tridiag_UT_l_opt_var3 (FLA_Obj A, FLA_Obj T) |
FLA_Error | FLA_Tridiag_UT_l_step_opt_var3 (FLA_Obj A, FLA_Obj Z, FLA_Obj T) |
FLA_Error | FLA_Tridiag_UT_l_step_ops_var3 (int m_A, int m_T, float *buff_A, int rs_A, int cs_A, float *buff_Z, int rs_Z, int cs_Z, float *buff_T, int rs_T, int cs_T) |
FLA_Error | FLA_Tridiag_UT_l_step_opd_var3 (int m_A, int m_T, double *buff_A, int rs_A, int cs_A, double *buff_Z, int rs_Z, int cs_Z, double *buff_T, int rs_T, int cs_T) |
FLA_Error | FLA_Tridiag_UT_l_step_opc_var3 (int m_A, int m_T, scomplex *buff_A, int rs_A, int cs_A, scomplex *buff_Z, int rs_Z, int cs_Z, scomplex *buff_T, int rs_T, int cs_T) |
FLA_Error | FLA_Tridiag_UT_l_step_opz_var3 (int m_A, int m_T, dcomplex *buff_A, int rs_A, int cs_A, dcomplex *buff_Z, int rs_Z, int cs_Z, dcomplex *buff_T, int rs_T, int cs_T) |
FLA_Error | FLA_Tridiag_UT_l_ofu_var1 (FLA_Obj A, FLA_Obj T) |
FLA_Error | FLA_Tridiag_UT_l_step_ofu_var1 (FLA_Obj A, FLA_Obj T) |
FLA_Error | FLA_Tridiag_UT_l_step_ofs_var1 (int m_A, int m_T, float *buff_A, int rs_A, int cs_A, float *buff_T, int rs_T, int cs_T) |
FLA_Error | FLA_Tridiag_UT_l_step_ofd_var1 (int m_A, int m_T, double *buff_A, int rs_A, int cs_A, double *buff_T, int rs_T, int cs_T) |
FLA_Error | FLA_Tridiag_UT_l_step_ofc_var1 (int m_A, int m_T, scomplex *buff_A, int rs_A, int cs_A, scomplex *buff_T, int rs_T, int cs_T) |
FLA_Error | FLA_Tridiag_UT_l_step_ofz_var1 (int m_A, int m_T, dcomplex *buff_A, int rs_A, int cs_A, dcomplex *buff_T, int rs_T, int cs_T) |
FLA_Error | FLA_Tridiag_UT_l_ofu_var2 (FLA_Obj A, FLA_Obj T) |
FLA_Error | FLA_Tridiag_UT_l_step_ofu_var2 (FLA_Obj A, FLA_Obj T) |
FLA_Error | FLA_Tridiag_UT_l_step_ofs_var2 (int m_A, int m_T, float *buff_A, int rs_A, int cs_A, float *buff_T, int rs_T, int cs_T) |
FLA_Error | FLA_Tridiag_UT_l_step_ofd_var2 (int m_A, int m_T, double *buff_A, int rs_A, int cs_A, double *buff_T, int rs_T, int cs_T) |
FLA_Error | FLA_Tridiag_UT_l_step_ofc_var2 (int m_A, int m_T, scomplex *buff_A, int rs_A, int cs_A, scomplex *buff_T, int rs_T, int cs_T) |
FLA_Error | FLA_Tridiag_UT_l_step_ofz_var2 (int m_A, int m_T, dcomplex *buff_A, int rs_A, int cs_A, dcomplex *buff_T, int rs_T, int cs_T) |
FLA_Error | FLA_Tridiag_UT_l_ofu_var3 (FLA_Obj A, FLA_Obj T) |
FLA_Error | FLA_Tridiag_UT_l_step_ofu_var3 (FLA_Obj A, FLA_Obj Z, FLA_Obj T) |
FLA_Error | FLA_Tridiag_UT_l_step_ofs_var3 (int m_A, int m_T, float *buff_A, int rs_A, int cs_A, float *buff_Z, int rs_Z, int cs_Z, float *buff_T, int rs_T, int cs_T) |
FLA_Error | FLA_Tridiag_UT_l_step_ofd_var3 (int m_A, int m_T, double *buff_A, int rs_A, int cs_A, double *buff_Z, int rs_Z, int cs_Z, double *buff_T, int rs_T, int cs_T) |
FLA_Error | FLA_Tridiag_UT_l_step_ofc_var3 (int m_A, int m_T, scomplex *buff_A, int rs_A, int cs_A, scomplex *buff_Z, int rs_Z, int cs_Z, scomplex *buff_T, int rs_T, int cs_T) |
FLA_Error | FLA_Tridiag_UT_l_step_ofz_var3 (int m_A, int m_T, dcomplex *buff_A, int rs_A, int cs_A, dcomplex *buff_Z, int rs_Z, int cs_Z, dcomplex *buff_T, int rs_T, int cs_T) |
FLA_Error | FLA_Fused_Her2_Ax_opt_var1 (FLA_Obj alpha, FLA_Obj u, FLA_Obj z, FLA_Obj A, FLA_Obj x, FLA_Obj w) |
FLA_Error | FLA_Fused_Her2_Ax_ops_var1 (int m_A, float *buff_alpha, float *buff_u, int inc_u, float *buff_z, int inc_z, float *buff_A, int rs_A, int cs_A, float *buff_x, int inc_x, float *buff_w, int inc_w) |
FLA_Error | FLA_Fused_Her2_Ax_opd_var1 (int m_A, double *buff_alpha, double *buff_u, int inc_u, double *buff_z, int inc_z, double *buff_A, int rs_A, int cs_A, double *buff_x, int inc_x, double *buff_w, int inc_w) |
FLA_Error | FLA_Fused_Her2_Ax_opc_var1 (int m_A, scomplex *buff_alpha, scomplex *buff_u, int inc_u, scomplex *buff_z, int inc_z, scomplex *buff_A, int rs_A, int cs_A, scomplex *buff_x, int inc_x, scomplex *buff_w, int inc_w) |
FLA_Error | FLA_Fused_Her2_Ax_opz_var1 (int m_A, dcomplex *buff_alpha, dcomplex *buff_u, int inc_u, dcomplex *buff_z, int inc_z, dcomplex *buff_A, int rs_A, int cs_A, dcomplex *buff_x, int inc_x, dcomplex *buff_w, int inc_w) |
FLA_Error | FLA_Fused_UZhu_ZUhu_opt_var1 (FLA_Obj delta, FLA_Obj U, FLA_Obj Z, FLA_Obj u, FLA_Obj w) |
FLA_Error | FLA_Fused_UZhu_ZUhu_ops_var1 (int m_U, int n_U, float *buff_delta, float *buff_U, int rs_U, int cs_U, float *buff_Z, int rs_Z, int cs_Z, float *buff_u, int inc_u, float *buff_w, int inc_w) |
FLA_Error | FLA_Fused_UZhu_ZUhu_opd_var1 (int m_U, int n_U, double *buff_delta, double *buff_U, int rs_U, int cs_U, double *buff_Z, int rs_Z, int cs_Z, double *buff_u, int inc_u, double *buff_w, int inc_w) |
FLA_Error | FLA_Fused_UZhu_ZUhu_opc_var1 (int m_U, int n_U, scomplex *buff_delta, scomplex *buff_U, int rs_U, int cs_U, scomplex *buff_Z, int rs_Z, int cs_Z, scomplex *buff_u, int inc_u, scomplex *buff_w, int inc_w) |
FLA_Error | FLA_Fused_UZhu_ZUhu_opz_var1 (int m_U, int n_U, dcomplex *buff_delta, dcomplex *buff_U, int rs_U, int cs_U, dcomplex *buff_Z, int rs_Z, int cs_Z, dcomplex *buff_u, int inc_u, dcomplex *buff_w, int inc_w) |
FLA_Error FLA_Fused_Her2_Ax_opc_var1 | ( | int | m_A, |
scomplex * | buff_alpha, | ||
scomplex * | buff_u, | ||
int | inc_u, | ||
scomplex * | buff_z, | ||
int | inc_z, | ||
scomplex * | buff_A, | ||
int | rs_A, | ||
int | cs_A, | ||
scomplex * | buff_x, | ||
int | inc_x, | ||
scomplex * | buff_w, | ||
int | inc_w | ||
) |
References bli_cdot(), bli_csetv(), F77_caxpy(), FLA_ZERO, scomplex::imag, and scomplex::real.
Referenced by FLA_Fused_Her2_Ax_opt_var1(), and FLA_Tridiag_UT_l_step_ofc_var2().
{ scomplex* buff_0 = FLA_COMPLEX_PTR( FLA_ZERO ); int i; bli_csetv( m_A, buff_0, buff_w, inc_w ); for ( i = 0; i < m_A; ++i ) { scomplex* alpha11 = buff_A + (i )*cs_A + (i )*rs_A; scomplex* a21 = buff_A + (i )*cs_A + (i+1)*rs_A; scomplex* upsilon1 = buff_u + (i )*inc_u; scomplex* u2 = buff_u + (i+1)*inc_u; scomplex* zeta1 = buff_z + (i )*inc_z; scomplex* z2 = buff_z + (i+1)*inc_z; scomplex* chi1 = buff_x + (i )*inc_x; scomplex* x2 = buff_x + (i+1)*inc_x; scomplex* omega1 = buff_w + (i )*inc_w; scomplex* w2 = buff_w + (i+1)*inc_w; // scomplex* beta = buff_beta; scomplex minus_conj_upsilon1; scomplex minus_conj_zeta1; scomplex temp; int m_ahead = m_A - i - 1; /*------------------------------------------------------------*/ // bli_ccopyconj( zeta1, &conj_zeta1 ); // bli_cmult3( beta, &conj_zeta1, &minus_conj_zeta1 ); // bli_cmult3( &minus_conj_zeta1, upsilon1, &temp ); // bli_cadd3( &temp, alpha11, alpha11 ); //bli_ccopyconj( upsilon1, &conj_upsilon1 ); //bli_cmult3( beta, &conj_upsilon1, &minus_conj_upsilon1 ); //bli_cmult3( &minus_conj_upsilon1, zeta1, &temp ); //bli_cadd3( &temp, alpha11, alpha11 ); minus_conj_zeta1.real = - zeta1->real; minus_conj_zeta1.imag = - -zeta1->imag; minus_conj_upsilon1.real = - upsilon1->real; minus_conj_upsilon1.imag = - -upsilon1->imag; alpha11->real -= zeta1->real * upsilon1->real - -zeta1->imag * upsilon1->imag + zeta1->real * upsilon1->real - zeta1->imag * -upsilon1->imag; alpha11->imag -= -zeta1->imag * upsilon1->real + zeta1->real * upsilon1->imag + zeta1->imag * upsilon1->real + zeta1->real * -upsilon1->imag; // bli_caxpyv( BLIS_NO_CONJUGATE, // m_ahead, // &minus_conj_zeta1, // u2, inc_u, // a21, rs_A ); F77_caxpy( &m_ahead, &minus_conj_zeta1, u2, &inc_u, a21, &rs_A ); // bli_caxpyv( BLIS_NO_CONJUGATE, // m_ahead, // &minus_conj_upsilon1, // z2, inc_z, // a21, rs_A ); F77_caxpy( &m_ahead, &minus_conj_upsilon1, z2, &inc_z, a21, &rs_A ); // bli_cmult3( alpha11, chi1, &temp ); // bli_cadd3( &temp, omega1, omega1 ); omega1->real += alpha11->real * chi1->real - alpha11->imag * chi1->imag; omega1->imag += alpha11->imag * chi1->real + alpha11->real * chi1->imag; bli_cdot( BLIS_CONJUGATE, m_ahead, a21, rs_A, x2, inc_x, &temp ); // bli_cadd3( &temp, omega1, omega1 ); omega1->real += temp.real; omega1->imag += temp.imag; // bli_caxpyv( BLIS_NO_CONJUGATE, // m_ahead, // chi1, // a21, rs_A, // w2, inc_w ); F77_caxpy( &m_ahead, chi1, a21, &rs_A, w2, &inc_w ); /*------------------------------------------------------------*/ } return FLA_SUCCESS; }
FLA_Error FLA_Fused_Her2_Ax_opd_var1 | ( | int | m_A, |
double * | buff_alpha, | ||
double * | buff_u, | ||
int | inc_u, | ||
double * | buff_z, | ||
int | inc_z, | ||
double * | buff_A, | ||
int | rs_A, | ||
int | cs_A, | ||
double * | buff_x, | ||
int | inc_x, | ||
double * | buff_w, | ||
int | inc_w | ||
) |
References bli_dsetv(), F77_daxpy(), F77_ddot(), and FLA_ZERO.
Referenced by FLA_Fused_Her2_Ax_opt_var1(), and FLA_Tridiag_UT_l_step_ofd_var2().
{ double* buff_0 = FLA_DOUBLE_PTR( FLA_ZERO ); int i; bli_dsetv( m_A, buff_0, buff_w, inc_w ); for ( i = 0; i < m_A; ++i ) { double* alpha11 = buff_A + (i )*cs_A + (i )*rs_A; double* a21 = buff_A + (i )*cs_A + (i+1)*rs_A; double* upsilon1 = buff_u + (i )*inc_u; double* u2 = buff_u + (i+1)*inc_u; double* zeta1 = buff_z + (i )*inc_z; double* z2 = buff_z + (i+1)*inc_z; double* chi1 = buff_x + (i )*inc_x; double* x2 = buff_x + (i+1)*inc_x; double* omega1 = buff_w + (i )*inc_w; double* w2 = buff_w + (i+1)*inc_w; // double* beta = buff_beta; double minus_conj_upsilon1; double minus_conj_zeta1; double temp; int m_ahead = m_A - i - 1; /*------------------------------------------------------------*/ // bli_dcopyconj( zeta1, &conj_zeta1 ); // bli_dmult3( beta, &conj_zeta1, &minus_conj_zeta1 ); // bli_dmult3( &minus_conj_zeta1, upsilon1, &temp ); // bli_dadd3( &temp, alpha11, alpha11 ); //bli_dcopyconj( upsilon1, &conj_upsilon1 ); //bli_dmult3( beta, &conj_upsilon1, &minus_conj_upsilon1 ); //bli_dmult3( &minus_conj_upsilon1, zeta1, &temp ); //bli_dadd3( &temp, alpha11, alpha11 ); minus_conj_zeta1 = - *zeta1; minus_conj_upsilon1 = - *upsilon1; *alpha11 -= 2.0 * *zeta1 * *upsilon1; // bli_daxpyv( BLIS_NO_CONJUGATE, // m_ahead, // &minus_conj_zeta1, // u2, inc_u, // a21, rs_A ); F77_daxpy( &m_ahead, &minus_conj_zeta1, u2, &inc_u, a21, &rs_A ); // bli_daxpyv( BLIS_NO_CONJUGATE, // m_ahead, // &minus_conj_upsilon1, // z2, inc_z, // a21, rs_A ); F77_daxpy( &m_ahead, &minus_conj_upsilon1, z2, &inc_z, a21, &rs_A ); // bli_dmult3( alpha11, chi1, &temp ); // bli_dadd3( &temp, omega1, omega1 ); *omega1 += *alpha11 * *chi1; // bli_ddot( BLIS_CONJUGATE, // m_ahead, // a21, rs_A, // x2, inc_x, // &temp ); temp = F77_ddot( &m_ahead, a21, &rs_A, x2, &inc_x ); // bli_dadd3( &temp, omega1, omega1 ); *omega1 += temp; // bli_daxpyv( BLIS_NO_CONJUGATE, // m_ahead, // chi1, // a21, rs_A, // w2, inc_w ); F77_daxpy( &m_ahead, chi1, a21, &rs_A, w2, &inc_w ); /*------------------------------------------------------------*/ } return FLA_SUCCESS; }
FLA_Error FLA_Fused_Her2_Ax_ops_var1 | ( | int | m_A, |
float * | buff_alpha, | ||
float * | buff_u, | ||
int | inc_u, | ||
float * | buff_z, | ||
int | inc_z, | ||
float * | buff_A, | ||
int | rs_A, | ||
int | cs_A, | ||
float * | buff_x, | ||
int | inc_x, | ||
float * | buff_w, | ||
int | inc_w | ||
) |
References bli_ssetv(), F77_saxpy(), F77_sdot(), and FLA_ZERO.
Referenced by FLA_Fused_Her2_Ax_opt_var1(), and FLA_Tridiag_UT_l_step_ofs_var2().
{ float* buff_0 = FLA_FLOAT_PTR( FLA_ZERO ); int i; bli_ssetv( m_A, buff_0, buff_w, inc_w ); for ( i = 0; i < m_A; ++i ) { float* alpha11 = buff_A + (i )*cs_A + (i )*rs_A; float* a21 = buff_A + (i )*cs_A + (i+1)*rs_A; float* upsilon1 = buff_u + (i )*inc_u; float* u2 = buff_u + (i+1)*inc_u; float* zeta1 = buff_z + (i )*inc_z; float* z2 = buff_z + (i+1)*inc_z; float* chi1 = buff_x + (i )*inc_x; float* x2 = buff_x + (i+1)*inc_x; float* omega1 = buff_w + (i )*inc_w; float* w2 = buff_w + (i+1)*inc_w; // float* beta = buff_beta; float minus_conj_upsilon1; float minus_conj_zeta1; float temp; int m_ahead = m_A - i - 1; /*------------------------------------------------------------*/ // bli_scopyconj( zeta1, &conj_zeta1 ); // bli_smult3( beta, &conj_zeta1, &minus_conj_zeta1 ); // bli_smult3( &minus_conj_zeta1, upsilon1, &temp ); // bli_sadd3( &temp, alpha11, alpha11 ); //bli_scopyconj( upsilon1, &conj_upsilon1 ); //bli_smult3( beta, &conj_upsilon1, &minus_conj_upsilon1 ); //bli_smult3( &minus_conj_upsilon1, zeta1, &temp ); //bli_sadd3( &temp, alpha11, alpha11 ); minus_conj_zeta1 = - *zeta1; minus_conj_upsilon1 = - *upsilon1; *alpha11 -= 2.0F * *zeta1 * *upsilon1; // bli_saxpyv( BLIS_NO_CONJUGATE, // m_ahead, // &minus_conj_zeta1, // u2, inc_u, // a21, rs_A ); F77_saxpy( &m_ahead, &minus_conj_zeta1, u2, &inc_u, a21, &rs_A ); // bli_saxpyv( BLIS_NO_CONJUGATE, // m_ahead, // &minus_conj_upsilon1, // z2, inc_z, // a21, rs_A ); F77_saxpy( &m_ahead, &minus_conj_upsilon1, z2, &inc_z, a21, &rs_A ); // bli_smult3( alpha11, chi1, &temp ); // bli_sadd3( &temp, omega1, omega1 ); *omega1 += *alpha11 * *chi1; // bli_sdot( BLIS_CONJUGATE, // m_ahead, // a21, rs_A, // x2, inc_x, // &temp ); temp = F77_sdot( &m_ahead, a21, &rs_A, x2, &inc_x ); // bli_sadd3( &temp, omega1, omega1 ); *omega1 += temp; // bli_saxpyv( BLIS_NO_CONJUGATE, // m_ahead, // chi1, // a21, rs_A, // w2, inc_w ); F77_saxpy( &m_ahead, chi1, a21, &rs_A, w2, &inc_w ); /*------------------------------------------------------------*/ } return FLA_SUCCESS; }
FLA_Error FLA_Fused_Her2_Ax_opt_var1 | ( | FLA_Obj | alpha, |
FLA_Obj | u, | ||
FLA_Obj | z, | ||
FLA_Obj | A, | ||
FLA_Obj | x, | ||
FLA_Obj | w | ||
) |
References FLA_Fused_Her2_Ax_opc_var1(), FLA_Fused_Her2_Ax_opd_var1(), FLA_Fused_Her2_Ax_ops_var1(), FLA_Fused_Her2_Ax_opz_var1(), FLA_Obj_col_stride(), FLA_Obj_datatype(), FLA_Obj_length(), FLA_Obj_row_stride(), and FLA_Obj_vector_inc().
{ /* Effective computation: A = A + beta * ( u * z' + z * u' ); w = A * x; */ FLA_Datatype datatype; int m_A; int rs_A, cs_A; int inc_u, inc_z, inc_x, inc_w; datatype = FLA_Obj_datatype( A ); m_A = FLA_Obj_length( A ); rs_A = FLA_Obj_row_stride( A ); cs_A = FLA_Obj_col_stride( A ); inc_u = FLA_Obj_vector_inc( u ); inc_z = FLA_Obj_vector_inc( z ); inc_x = FLA_Obj_vector_inc( x ); inc_w = FLA_Obj_vector_inc( w ); switch ( datatype ) { case FLA_FLOAT: { float* buff_A = FLA_FLOAT_PTR( A ); float* buff_u = FLA_FLOAT_PTR( u ); float* buff_z = FLA_FLOAT_PTR( z ); float* buff_x = FLA_FLOAT_PTR( x ); float* buff_w = FLA_FLOAT_PTR( w ); float* buff_beta = FLA_FLOAT_PTR( beta ); FLA_Fused_Her2_Ax_ops_var1( m_A, buff_beta, buff_u, inc_u, buff_z, inc_z, buff_A, rs_A, cs_A, buff_x, inc_x, buff_w, inc_w ); break; } case FLA_DOUBLE: { double* buff_A = FLA_DOUBLE_PTR( A ); double* buff_u = FLA_DOUBLE_PTR( u ); double* buff_z = FLA_DOUBLE_PTR( z ); double* buff_x = FLA_DOUBLE_PTR( x ); double* buff_w = FLA_DOUBLE_PTR( w ); double* buff_beta = FLA_DOUBLE_PTR( beta ); FLA_Fused_Her2_Ax_opd_var1( m_A, buff_beta, buff_u, inc_u, buff_z, inc_z, buff_A, rs_A, cs_A, buff_x, inc_x, buff_w, inc_w ); break; } case FLA_COMPLEX: { scomplex* buff_A = FLA_COMPLEX_PTR( A ); scomplex* buff_u = FLA_COMPLEX_PTR( u ); scomplex* buff_z = FLA_COMPLEX_PTR( z ); scomplex* buff_x = FLA_COMPLEX_PTR( x ); scomplex* buff_w = FLA_COMPLEX_PTR( w ); scomplex* buff_beta = FLA_COMPLEX_PTR( beta ); FLA_Fused_Her2_Ax_opc_var1( m_A, buff_beta, buff_u, inc_u, buff_z, inc_z, buff_A, rs_A, cs_A, buff_x, inc_x, buff_w, inc_w ); break; } case FLA_DOUBLE_COMPLEX: { dcomplex* buff_A = FLA_DOUBLE_COMPLEX_PTR( A ); dcomplex* buff_u = FLA_DOUBLE_COMPLEX_PTR( u ); dcomplex* buff_z = FLA_DOUBLE_COMPLEX_PTR( z ); dcomplex* buff_x = FLA_DOUBLE_COMPLEX_PTR( x ); dcomplex* buff_w = FLA_DOUBLE_COMPLEX_PTR( w ); dcomplex* buff_beta = FLA_DOUBLE_COMPLEX_PTR( beta ); FLA_Fused_Her2_Ax_opz_var1( m_A, buff_beta, buff_u, inc_u, buff_z, inc_z, buff_A, rs_A, cs_A, buff_x, inc_x, buff_w, inc_w ); break; } } return FLA_SUCCESS; }
FLA_Error FLA_Fused_Her2_Ax_opz_var1 | ( | int | m_A, |
dcomplex * | buff_alpha, | ||
dcomplex * | buff_u, | ||
int | inc_u, | ||
dcomplex * | buff_z, | ||
int | inc_z, | ||
dcomplex * | buff_A, | ||
int | rs_A, | ||
int | cs_A, | ||
dcomplex * | buff_x, | ||
int | inc_x, | ||
dcomplex * | buff_w, | ||
int | inc_w | ||
) |
References bli_zdot(), bli_zsetv(), F77_zaxpy(), FLA_ZERO, dcomplex::imag, and dcomplex::real.
Referenced by FLA_Fused_Her2_Ax_opt_var1(), and FLA_Tridiag_UT_l_step_ofz_var2().
{ dcomplex* buff_0 = FLA_DOUBLE_COMPLEX_PTR( FLA_ZERO ); int i; bli_zsetv( m_A, buff_0, buff_w, inc_w ); for ( i = 0; i < m_A; ++i ) { dcomplex* alpha11 = buff_A + (i )*cs_A + (i )*rs_A; dcomplex* a21 = buff_A + (i )*cs_A + (i+1)*rs_A; dcomplex* upsilon1 = buff_u + (i )*inc_u; dcomplex* u2 = buff_u + (i+1)*inc_u; dcomplex* zeta1 = buff_z + (i )*inc_z; dcomplex* z2 = buff_z + (i+1)*inc_z; dcomplex* chi1 = buff_x + (i )*inc_x; dcomplex* x2 = buff_x + (i+1)*inc_x; dcomplex* omega1 = buff_w + (i )*inc_w; dcomplex* w2 = buff_w + (i+1)*inc_w; // dcomplex* beta = buff_beta; dcomplex minus_conj_upsilon1; dcomplex minus_conj_zeta1; dcomplex temp; int m_ahead = m_A - i - 1; /*------------------------------------------------------------*/ // bli_zcopyconj( zeta1, &conj_zeta1 ); // bli_zmult3( beta, &conj_zeta1, &minus_conj_zeta1 ); // bli_zmult3( &minus_conj_zeta1, upsilon1, &temp ); // bli_zadd3( &temp, alpha11, alpha11 ); //bli_zcopyconj( upsilon1, &conj_upsilon1 ); //bli_zmult3( beta, &conj_upsilon1, &minus_conj_upsilon1 ); //bli_zmult3( &minus_conj_upsilon1, zeta1, &temp ); //bli_zadd3( &temp, alpha11, alpha11 ); minus_conj_zeta1.real = - zeta1->real; minus_conj_zeta1.imag = - -zeta1->imag; minus_conj_upsilon1.real = - upsilon1->real; minus_conj_upsilon1.imag = - -upsilon1->imag; alpha11->real -= zeta1->real * upsilon1->real - -zeta1->imag * upsilon1->imag + zeta1->real * upsilon1->real - zeta1->imag * -upsilon1->imag; alpha11->imag -= -zeta1->imag * upsilon1->real + zeta1->real * upsilon1->imag + zeta1->imag * upsilon1->real + zeta1->real * -upsilon1->imag; // bli_zaxpyv( BLIS_NO_CONJUGATE, // m_ahead, // &minus_conj_zeta1, // u2, inc_u, // a21, rs_A ); F77_zaxpy( &m_ahead, &minus_conj_zeta1, u2, &inc_u, a21, &rs_A ); // bli_zaxpyv( BLIS_NO_CONJUGATE, // m_ahead, // &minus_conj_upsilon1, // z2, inc_z, // a21, rs_A ); F77_zaxpy( &m_ahead, &minus_conj_upsilon1, z2, &inc_z, a21, &rs_A ); // bli_zmult3( alpha11, chi1, &temp ); // bli_zadd3( &temp, omega1, omega1 ); omega1->real += alpha11->real * chi1->real - alpha11->imag * chi1->imag; omega1->imag += alpha11->imag * chi1->real + alpha11->real * chi1->imag; bli_zdot( BLIS_CONJUGATE, m_ahead, a21, rs_A, x2, inc_x, &temp ); // bli_zadd3( &temp, omega1, omega1 ); omega1->real += temp.real; omega1->imag += temp.imag; // bli_zaxpyv( BLIS_NO_CONJUGATE, // m_ahead, // chi1, // a21, rs_A, // w2, inc_w ); F77_zaxpy( &m_ahead, chi1, a21, &rs_A, w2, &inc_w ); /*------------------------------------------------------------*/ } return FLA_SUCCESS; }
FLA_Error FLA_Fused_UZhu_ZUhu_opc_var1 | ( | int | m_U, |
int | n_U, | ||
scomplex * | buff_delta, | ||
scomplex * | buff_U, | ||
int | rs_U, | ||
int | cs_U, | ||
scomplex * | buff_Z, | ||
int | rs_Z, | ||
int | cs_Z, | ||
scomplex * | buff_u, | ||
int | inc_u, | ||
scomplex * | buff_w, | ||
int | inc_w | ||
) |
References bli_cdot(), and F77_caxpy().
Referenced by FLA_Fused_UZhu_ZUhu_opt_var1(), and FLA_Tridiag_UT_l_step_ofc_var3().
{ int i; for ( i = 0; i < n_U; ++i ) { scomplex* u1 = buff_U + (i )*cs_U + (0 )*rs_U; scomplex* z1 = buff_Z + (i )*cs_Z + (0 )*rs_Z; scomplex* delta = buff_delta; scomplex* u = buff_u; scomplex* w = buff_w; scomplex alpha; scomplex beta; /*------------------------------------------------------------*/ bli_cdot( BLIS_CONJUGATE, m_U, z1, rs_Z, u, inc_u, &alpha ); bli_cdot( BLIS_CONJUGATE, m_U, u1, rs_U, u, inc_u, &beta ); bli_cscals( delta, &alpha ); bli_cscals( delta, &beta ); // bli_caxpyv( BLIS_NO_CONJUGATE, // m_U, // &alpha, // u1, rs_U, // w, inc_w ); F77_caxpy( &m_U, &alpha, u1, &rs_U, w, &inc_w ); // bli_caxpyv( BLIS_NO_CONJUGATE, // m_U, // &beta, // z1, rs_U, // w, inc_w ); F77_caxpy( &m_U, &beta, z1, &rs_Z, w, &inc_w ); /*------------------------------------------------------------*/ } return FLA_SUCCESS; }
FLA_Error FLA_Fused_UZhu_ZUhu_opd_var1 | ( | int | m_U, |
int | n_U, | ||
double * | buff_delta, | ||
double * | buff_U, | ||
int | rs_U, | ||
int | cs_U, | ||
double * | buff_Z, | ||
int | rs_Z, | ||
int | cs_Z, | ||
double * | buff_u, | ||
int | inc_u, | ||
double * | buff_w, | ||
int | inc_w | ||
) |
References F77_daxpy(), and F77_ddot().
Referenced by FLA_Fused_UZhu_ZUhu_opt_var1(), and FLA_Tridiag_UT_l_step_ofd_var3().
{ int i; for ( i = 0; i < n_U; ++i ) { double* u1 = buff_U + (i )*cs_U + (0 )*rs_U; double* z1 = buff_Z + (i )*cs_Z + (0 )*rs_Z; double* delta = buff_delta; double* u = buff_u; double* w = buff_w; double alpha; double beta; /*------------------------------------------------------------*/ // bli_ddot( BLIS_CONJUGATE, // m_U, // z1, rs_Z, // u, inc_u, // &alpha ); alpha = F77_ddot( &m_U, z1, &rs_Z, u, &inc_u ); // bli_ddot( BLIS_CONJUGATE, // m_U, // u1, rs_U, // u, inc_u, // &beta ); beta = F77_ddot( &m_U, u1, &rs_U, u, &inc_u ); // bli_dscals( delta, &alpha ); // bli_dscals( delta, &beta ); alpha *= *delta; beta *= *delta; // bli_daxpyv( BLIS_NO_CONJUGATE, // m_U, // &alpha, // u1, rs_U, // w, inc_w ); F77_daxpy( &m_U, &alpha, u1, &rs_U, w, &inc_w ); // bli_daxpyv( BLIS_NO_CONJUGATE, // m_U, // &beta, // z1, rs_U, // w, inc_w ); F77_daxpy( &m_U, &beta, z1, &rs_Z, w, &inc_w ); /*------------------------------------------------------------*/ } return FLA_SUCCESS; }
FLA_Error FLA_Fused_UZhu_ZUhu_ops_var1 | ( | int | m_U, |
int | n_U, | ||
float * | buff_delta, | ||
float * | buff_U, | ||
int | rs_U, | ||
int | cs_U, | ||
float * | buff_Z, | ||
int | rs_Z, | ||
int | cs_Z, | ||
float * | buff_u, | ||
int | inc_u, | ||
float * | buff_w, | ||
int | inc_w | ||
) |
References F77_saxpy(), and F77_sdot().
Referenced by FLA_Fused_UZhu_ZUhu_opt_var1(), and FLA_Tridiag_UT_l_step_ofs_var3().
{ int i; for ( i = 0; i < n_U; ++i ) { float* u1 = buff_U + (i )*cs_U + (0 )*rs_U; float* z1 = buff_Z + (i )*cs_Z + (0 )*rs_Z; float* delta = buff_delta; float* u = buff_u; float* w = buff_w; float alpha; float beta; /*------------------------------------------------------------*/ // bli_sdot( BLIS_CONJUGATE, // m_U, // z1, rs_Z, // u, inc_u, // &alpha ); alpha = F77_sdot( &m_U, z1, &rs_Z, u, &inc_u ); // bli_sdot( BLIS_CONJUGATE, // m_U, // u1, rs_U, // u, inc_u, // &beta ); beta = F77_sdot( &m_U, u1, &rs_U, u, &inc_u ); // bli_sscals( delta, &alpha ); // bli_sscals( delta, &beta ); alpha *= *delta; beta *= *delta; // bli_saxpyv( BLIS_NO_CONJUGATE, // m_U, // &alpha, // u1, rs_U, // w, inc_w ); F77_saxpy( &m_U, &alpha, u1, &rs_U, w, &inc_w ); // bli_saxpyv( BLIS_NO_CONJUGATE, // m_U, // &beta, // z1, rs_U, // w, inc_w ); F77_saxpy( &m_U, &beta, z1, &rs_Z, w, &inc_w ); /*------------------------------------------------------------*/ } return FLA_SUCCESS; }
FLA_Error FLA_Fused_UZhu_ZUhu_opt_var1 | ( | FLA_Obj | delta, |
FLA_Obj | U, | ||
FLA_Obj | Z, | ||
FLA_Obj | u, | ||
FLA_Obj | w | ||
) |
References FLA_Fused_UZhu_ZUhu_opc_var1(), FLA_Fused_UZhu_ZUhu_opd_var1(), FLA_Fused_UZhu_ZUhu_ops_var1(), FLA_Fused_UZhu_ZUhu_opz_var1(), FLA_Obj_col_stride(), FLA_Obj_datatype(), FLA_Obj_length(), FLA_Obj_row_stride(), FLA_Obj_vector_inc(), and FLA_Obj_width().
{ /* Effective computation: w = w + delta * ( U ( Z' u ) + Z ( U' a ) ); */ FLA_Datatype datatype; int m_U, n_U; int rs_U, cs_U; int rs_Z, cs_Z; int inc_u, inc_w; datatype = FLA_Obj_datatype( U ); m_U = FLA_Obj_length( U ); n_U = FLA_Obj_width( U ); rs_U = FLA_Obj_row_stride( U ); cs_U = FLA_Obj_col_stride( U ); rs_Z = FLA_Obj_row_stride( Z ); cs_Z = FLA_Obj_col_stride( Z ); inc_u = FLA_Obj_vector_inc( u ); inc_w = FLA_Obj_vector_inc( w ); switch ( datatype ) { case FLA_FLOAT: { float* buff_U = FLA_FLOAT_PTR( U ); float* buff_Z = FLA_FLOAT_PTR( Z ); float* buff_u = FLA_FLOAT_PTR( u ); float* buff_w = FLA_FLOAT_PTR( w ); float* buff_delta = FLA_FLOAT_PTR( delta ); FLA_Fused_UZhu_ZUhu_ops_var1( m_U, n_U, buff_delta, buff_U, rs_U, cs_U, buff_Z, rs_Z, cs_Z, buff_u, inc_u, buff_w, inc_w ); break; } case FLA_DOUBLE: { double* buff_U = FLA_DOUBLE_PTR( U ); double* buff_Z = FLA_DOUBLE_PTR( Z ); double* buff_u = FLA_DOUBLE_PTR( u ); double* buff_w = FLA_DOUBLE_PTR( w ); double* buff_delta = FLA_DOUBLE_PTR( delta ); FLA_Fused_UZhu_ZUhu_opd_var1( m_U, n_U, buff_delta, buff_U, rs_U, cs_U, buff_Z, rs_Z, cs_Z, buff_u, inc_u, buff_w, inc_w ); break; } case FLA_COMPLEX: { scomplex* buff_U = FLA_COMPLEX_PTR( U ); scomplex* buff_Z = FLA_COMPLEX_PTR( Z ); scomplex* buff_u = FLA_COMPLEX_PTR( u ); scomplex* buff_w = FLA_COMPLEX_PTR( w ); scomplex* buff_delta = FLA_COMPLEX_PTR( delta ); FLA_Fused_UZhu_ZUhu_opc_var1( m_U, n_U, buff_delta, buff_U, rs_U, cs_U, buff_Z, rs_Z, cs_Z, buff_u, inc_u, buff_w, inc_w ); break; } case FLA_DOUBLE_COMPLEX: { dcomplex* buff_U = FLA_DOUBLE_COMPLEX_PTR( U ); dcomplex* buff_Z = FLA_DOUBLE_COMPLEX_PTR( Z ); dcomplex* buff_u = FLA_DOUBLE_COMPLEX_PTR( u ); dcomplex* buff_w = FLA_DOUBLE_COMPLEX_PTR( w ); dcomplex* buff_delta = FLA_DOUBLE_COMPLEX_PTR( delta ); FLA_Fused_UZhu_ZUhu_opz_var1( m_U, n_U, buff_delta, buff_U, rs_U, cs_U, buff_Z, rs_Z, cs_Z, buff_u, inc_u, buff_w, inc_w ); break; } } return FLA_SUCCESS; }
FLA_Error FLA_Fused_UZhu_ZUhu_opz_var1 | ( | int | m_U, |
int | n_U, | ||
dcomplex * | buff_delta, | ||
dcomplex * | buff_U, | ||
int | rs_U, | ||
int | cs_U, | ||
dcomplex * | buff_Z, | ||
int | rs_Z, | ||
int | cs_Z, | ||
dcomplex * | buff_u, | ||
int | inc_u, | ||
dcomplex * | buff_w, | ||
int | inc_w | ||
) |
References bli_zdot(), and F77_zaxpy().
Referenced by FLA_Fused_UZhu_ZUhu_opt_var1(), and FLA_Tridiag_UT_l_step_ofz_var3().
{ int i; for ( i = 0; i < n_U; ++i ) { dcomplex* u1 = buff_U + (i )*cs_U + (0 )*rs_U; dcomplex* z1 = buff_Z + (i )*cs_Z + (0 )*rs_Z; dcomplex* delta = buff_delta; dcomplex* u = buff_u; dcomplex* w = buff_w; dcomplex alpha; dcomplex beta; /*------------------------------------------------------------*/ bli_zdot( BLIS_CONJUGATE, m_U, z1, rs_Z, u, inc_u, &alpha ); bli_zdot( BLIS_CONJUGATE, m_U, u1, rs_U, u, inc_u, &beta ); bli_zscals( delta, &alpha ); bli_zscals( delta, &beta ); // bli_zaxpyv( BLIS_NO_CONJUGATE, // m_U, // &alpha, // u1, rs_U, // w, inc_w ); F77_zaxpy( &m_U, &alpha, u1, &rs_U, w, &inc_w ); // bli_zaxpyv( BLIS_NO_CONJUGATE, // m_U, // &beta, // z1, rs_U, // w, inc_w ); F77_zaxpy( &m_U, &beta, z1, &rs_Z, w, &inc_w ); /*------------------------------------------------------------*/ } return FLA_SUCCESS; }
References FLA_Cont_with_1x3_to_1x2(), FLA_Cont_with_3x3_to_2x2(), FLA_Obj_length(), FLA_Part_1x2(), FLA_Part_2x2(), FLA_Repart_1x2_to_1x3(), FLA_Repart_2x2_to_3x3(), and FLA_Tridiag_UT_l_step_opt_var1().
Referenced by FLA_Tridiag_UT_l().
{ FLA_Obj ATL, ATR, A00, A01, A02, ABL, ABR, A10, A11, A12, A20, A21, A22; FLA_Obj TL, TR, T0, T1, T2; FLA_Obj T1_tl; FLA_Obj none, none2, none3; 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_1x2( T, &TL, &TR, 0, FLA_LEFT ); while ( FLA_Obj_length( ATL ) < FLA_Obj_length( A ) ) { b = min( FLA_Obj_length( ABR ), 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_1x2_to_1x3( TL, /**/ TR, &T0, /**/ &T1, &T2, b, FLA_RIGHT ); /*------------------------------------------------------------*/ FLA_Part_2x2( T1, &T1_tl, &none, &none2, &none3, b, b, FLA_TL ); // [ ABR, T1 ] = FLA_Tridiag_UT_l_step_unb_var1( ABR, T1, b ); //FLA_Tridiag_UT_l_step_unb_var1( ABR, T1_tl ); //FLA_Tridiag_UT_l_step_ofu_var1( ABR, T1_tl ); FLA_Tridiag_UT_l_step_opt_var1( ABR, T1_tl ); /*------------------------------------------------------------*/ FLA_Cont_with_3x3_to_2x2( &ATL, /**/ &ATR, A00, A01, /**/ A02, A10, A11, /**/ A12, /* ************** */ /* ****************** */ &ABL, /**/ &ABR, A20, A21, /**/ A22, FLA_TL ); FLA_Cont_with_1x3_to_1x2( &TL, /**/ &TR, T0, T1, /**/ T2, FLA_LEFT ); } return FLA_SUCCESS; }
References FLA_Cont_with_1x3_to_1x2(), FLA_Cont_with_3x3_to_2x2(), FLA_Obj_length(), FLA_Part_1x2(), FLA_Part_2x2(), FLA_Repart_1x2_to_1x3(), FLA_Repart_2x2_to_3x3(), and FLA_Tridiag_UT_l_step_opt_var2().
Referenced by FLA_Tridiag_UT_l().
{ FLA_Obj ATL, ATR, A00, A01, A02, ABL, ABR, A10, A11, A12, A20, A21, A22; FLA_Obj TL, TR, T0, T1, T2; FLA_Obj T1_tl; FLA_Obj none, none2, none3; 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_1x2( T, &TL, &TR, 0, FLA_LEFT ); while ( FLA_Obj_length( ATL ) < FLA_Obj_length( A ) ) { b = min( FLA_Obj_length( ABR ), 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_1x2_to_1x3( TL, /**/ TR, &T0, /**/ &T1, &T2, b, FLA_RIGHT ); /*------------------------------------------------------------*/ FLA_Part_2x2( T1, &T1_tl, &none, &none2, &none3, b, b, FLA_TL ); // [ ABR, T1 ] = FLA_Tridiag_UT_l_step_unb_var2( ABR, T1, b ); //FLA_Tridiag_UT_l_step_unb_var2( ABR, T1_tl ); //FLA_Tridiag_UT_l_step_ofu_var2( ABR, T1_tl ); FLA_Tridiag_UT_l_step_opt_var2( ABR, T1_tl ); /*------------------------------------------------------------*/ FLA_Cont_with_3x3_to_2x2( &ATL, /**/ &ATR, A00, A01, /**/ A02, A10, A11, /**/ A12, /* ************** */ /* ****************** */ &ABL, /**/ &ABR, A20, A21, /**/ A22, FLA_TL ); FLA_Cont_with_1x3_to_1x2( &TL, /**/ &TR, T0, T1, /**/ T2, FLA_LEFT ); } return FLA_SUCCESS; }
References FLA_Cont_with_1x3_to_1x2(), FLA_Cont_with_3x1_to_2x1(), FLA_Cont_with_3x3_to_2x2(), FLA_Copy_external(), FLA_Her2k_external(), FLA_MINUS_ONE, FLA_Obj_create(), FLA_Obj_datatype(), FLA_Obj_free(), FLA_Obj_length(), FLA_Obj_set_to_scalar(), 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_Triangularize(), FLA_Tridiag_UT_l_step_opt_var3(), and FLA_ZERO.
Referenced by FLA_Tridiag_UT_l().
{ FLA_Obj ATL, ATR, A00, A01, A02, ABL, ABR, A10, A11, A12, A20, A21, A22; FLA_Obj UT, U0, UB, U1, U2; FLA_Obj ZT, Z0, ZB, Z1, Z2; FLA_Obj TL, TR, T0, T1, T2; FLA_Obj U, Z; FLA_Obj ABR_l; FLA_Obj UB_l, U2_l; FLA_Obj ZB_l, Z2_l; FLA_Obj T1_tl; FLA_Obj none, none2, none3; FLA_Obj UB_tl, UB_bl; FLA_Datatype datatype_A; dim_t m_A; dim_t b_alg, b, bb; b_alg = FLA_Obj_length( T ); datatype_A = FLA_Obj_datatype( A ); m_A = FLA_Obj_length( A ); FLA_Obj_create( datatype_A, m_A, b_alg, 0, 0, &U ); FLA_Obj_create( datatype_A, m_A, b_alg, 0, 0, &Z ); FLA_Part_2x2( A, &ATL, &ATR, &ABL, &ABR, 0, 0, FLA_TL ); FLA_Part_2x1( U, &UT, &UB, 0, FLA_TOP ); FLA_Part_2x1( Z, &ZT, &ZB, 0, FLA_TOP ); FLA_Part_1x2( T, &TL, &TR, 0, FLA_LEFT ); while ( FLA_Obj_length( ATL ) < FLA_Obj_length( A ) ) { b = min( FLA_Obj_length( ABR ), 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( UT, &U0, /* ** */ /* ** */ &U1, UB, &U2, b, FLA_BOTTOM ); FLA_Repart_2x1_to_3x1( ZT, &Z0, /* ** */ /* ** */ &Z1, ZB, &Z2, b, FLA_BOTTOM ); FLA_Repart_1x2_to_1x3( TL, /**/ TR, &T0, /**/ &T1, &T2, b, FLA_RIGHT ); /*------------------------------------------------------------*/ FLA_Part_2x2( T1, &T1_tl, &none, &none2, &none3, b, b, FLA_TL ); bb = min( FLA_Obj_length( ABR ) - 1, b_alg ); FLA_Part_1x2( ABR, &ABR_l, &none, bb, FLA_LEFT ); FLA_Part_1x2( UB, &UB_l, &none, bb, FLA_LEFT ); FLA_Part_1x2( ZB, &ZB_l, &none, bb, FLA_LEFT ); FLA_Part_2x1( UB_l, &none, &U2_l, b, FLA_TOP ); FLA_Part_2x1( ZB_l, &none, &Z2_l, b, FLA_TOP ); // [ ABR, ZB, T1 ] = FLA_Tridiag_UT_l_step_unb_var3( ABR, ZB, T1, b ); //FLA_Tridiag_UT_l_step_unb_var3( ABR, ZB, T1_tl ); //FLA_Tridiag_UT_l_step_ofu_var3( ABR, ZB, T1_tl ); FLA_Tridiag_UT_l_step_opt_var3( ABR, ZB, T1_tl ); if ( FLA_Obj_length( A22 ) > 0 ) { // Build UB from ABR, with explicit unit subdiagonal and zeros. FLA_Copy_external( ABR_l, UB_l ); FLA_Part_2x1( UB_l, &UB_tl, &UB_bl, 1, FLA_TOP ); FLA_Triangularize( FLA_LOWER_TRIANGULAR, FLA_UNIT_DIAG, UB_bl ); FLA_Obj_set_to_scalar( FLA_ZERO, UB_tl ); // A22 = A22 - U2 * Y2' - Z2 * U2'; FLA_Her2k_external( FLA_LOWER_TRIANGULAR, FLA_NO_TRANSPOSE, FLA_MINUS_ONE, U2_l, Z2_l, FLA_ONE, A22 ); } /*------------------------------------------------------------*/ 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( &UT, U0, U1, /* ** */ /* ** */ &UB, U2, FLA_TOP ); FLA_Cont_with_3x1_to_2x1( &ZT, Z0, Z1, /* ** */ /* ** */ &ZB, Z2, FLA_TOP ); FLA_Cont_with_1x3_to_1x2( &TL, /**/ &TR, T0, T1, /**/ T2, FLA_LEFT ); } FLA_Obj_free( &U ); FLA_Obj_free( &Z ); return FLA_SUCCESS; }
References FLA_Tridiag_UT_l_step_ofu_var2().
{ return FLA_Tridiag_UT_l_step_ofu_var2( A, T ); }
References FLA_Obj_create_conf_to(), FLA_Obj_free(), and FLA_Tridiag_UT_l_step_ofu_var3().
{ FLA_Error r_val; FLA_Obj Z; FLA_Obj_create_conf_to( FLA_NO_TRANSPOSE, A, &Z ); r_val = FLA_Tridiag_UT_l_step_ofu_var3( A, Z, T ); FLA_Obj_free( &Z ); return r_val; }
References FLA_Tridiag_UT_l_step_opt_var1().
Referenced by FLA_Tridiag_UT_l().
{ return FLA_Tridiag_UT_l_step_opt_var1( A, T ); }
References FLA_Tridiag_UT_l_step_opt_var2().
Referenced by FLA_Tridiag_UT_l().
{ return FLA_Tridiag_UT_l_step_opt_var2( A, T ); }
References FLA_Obj_create_conf_to(), FLA_Obj_free(), and FLA_Tridiag_UT_l_step_opt_var3().
Referenced by FLA_Tridiag_UT_l().
{ FLA_Error r_val; FLA_Obj Z; FLA_Obj_create_conf_to( FLA_NO_TRANSPOSE, A, &Z ); r_val = FLA_Tridiag_UT_l_step_opt_var3( A, Z, T ); FLA_Obj_free( &Z ); return r_val; }
FLA_Error FLA_Tridiag_UT_l_step_ofc_var1 | ( | int | m_A, |
int | m_T, | ||
scomplex * | buff_A, | ||
int | rs_A, | ||
int | cs_A, | ||
scomplex * | buff_T, | ||
int | rs_T, | ||
int | cs_T | ||
) |
FLA_Error FLA_Tridiag_UT_l_step_ofc_var2 | ( | int | m_A, |
int | m_T, | ||
scomplex * | buff_A, | ||
int | rs_A, | ||
int | cs_A, | ||
scomplex * | buff_T, | ||
int | rs_T, | ||
int | cs_T | ||
) |
References bli_caxpyv(), bli_ccopyv(), bli_cdot(), bli_cgemv(), bli_chemv(), bli_cher2(), bli_cscalv(), FLA_free(), FLA_Fused_Her2_Ax_opc_var1(), FLA_Househ2_UT_l_opc(), FLA_malloc(), FLA_MINUS_ONE, FLA_ONE, FLA_TWO, and FLA_ZERO.
Referenced by FLA_Tridiag_UT_l_step_ofu_var2().
{ scomplex* buff_2 = FLA_COMPLEX_PTR( FLA_TWO ); scomplex* buff_1 = FLA_COMPLEX_PTR( FLA_ONE ); scomplex* buff_0 = FLA_COMPLEX_PTR( FLA_ZERO ); scomplex* buff_m1 = FLA_COMPLEX_PTR( FLA_MINUS_ONE ); scomplex first_elem; scomplex beta; scomplex inv_tau11; scomplex minus_inv_tau11; scomplex minus_upsilon11, minus_conj_upsilon11; scomplex minus_zeta11, minus_conj_zeta11; int i; // b_alg = FLA_Obj_length( T ); int b_alg = m_T; // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &u ); // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &z ); // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &w ); scomplex* buff_u = ( scomplex* ) FLA_malloc( m_A * sizeof( *buff_A ) ); scomplex* buff_z = ( scomplex* ) FLA_malloc( m_A * sizeof( *buff_A ) ); scomplex* buff_w = ( scomplex* ) FLA_malloc( m_A * sizeof( *buff_A ) ); int inc_u = 1; int inc_z = 1; int inc_w = 1; // Initialize some variables (only to prevent compiler warnings). first_elem = *buff_0; minus_inv_tau11 = *buff_0; for ( i = 0; i < b_alg; ++i ) { scomplex* A20 = buff_A + (0 )*cs_A + (i+1)*rs_A; scomplex* alpha11 = buff_A + (i )*cs_A + (i )*rs_A; scomplex* a21 = buff_A + (i )*cs_A + (i+1)*rs_A; scomplex* A22 = buff_A + (i+1)*cs_A + (i+1)*rs_A; scomplex* t01 = buff_T + (i )*cs_T + (0 )*rs_T; scomplex* tau11 = buff_T + (i )*cs_T + (i )*rs_T; scomplex* upsilon11= buff_u + (i )*inc_u; scomplex* u21 = buff_u + (i+1)*inc_u; scomplex* zeta11 = buff_z + (i )*inc_z; scomplex* z21 = buff_z + (i+1)*inc_z; scomplex* w21 = buff_w + (i+1)*inc_w; scomplex* a21_t = a21 + (0 )*cs_A + (0 )*rs_A; scomplex* a21_b = a21 + (0 )*cs_A + (1 )*rs_A; int m_ahead = m_A - i - 1; int m_behind = i; int n_behind = i; /*------------------------------------------------------------*/ if ( m_behind > 0 ) { // FLA_Copy( upsilon11, minus_upsilon11 ); // FLA_Scal( FLA_MINUS_ONE, minus_upsilon11 ); // FLA_Copy( minus_upsilon11, minus_conj_upsilon11 ); bli_cmult3( buff_m1, upsilon11, &minus_upsilon11 ); bli_ccopyconj( &minus_upsilon11, &minus_conj_upsilon11 ); // FLA_Copy( zeta11, minus_zeta11 ); // FLA_Scal( FLA_MINUS_ONE, minus_zeta11 ); // FLA_Copy( minus_zeta11, minus_conj_zeta11 ); bli_cmult3( buff_m1, zeta11, &minus_zeta11 ); bli_ccopyconj( &minus_zeta11, &minus_conj_zeta11 ); // FLA_Axpyt( FLA_CONJ_NO_TRANSPOSE, minus_upsilon11, zeta11, alpha11 ); // FLA_Axpyt( FLA_CONJ_NO_TRANSPOSE, minus_zeta11, upsilon11, alpha11 ); bli_caxpyv( BLIS_CONJUGATE, 1, &minus_upsilon11, zeta11, 1, alpha11, 1 ); bli_caxpyv( BLIS_CONJUGATE, 1, &minus_zeta11, upsilon11, 1, alpha11, 1 ); // FLA_Axpyt( FLA_NO_TRANSPOSE, minus_conj_zeta11, u21, a21 ); // FLA_Axpyt( FLA_NO_TRANSPOSE, minus_conj_upsilon11, z21, a21 ); bli_caxpyv( BLIS_NO_CONJUGATE, m_ahead, &minus_conj_zeta11, u21, inc_u, a21, rs_A ); bli_caxpyv( BLIS_NO_CONJUGATE, m_ahead, &minus_conj_upsilon11, z21, inc_z, a21, rs_A ); } if ( m_ahead > 0 ) { // FLA_Househ2_UT( FLA_LEFT, // a21_t, // a21_b, tau11 ); FLA_Househ2_UT_l_opc( m_ahead - 1, a21_t, a21_b, rs_A, tau11 ); // FLA_Obj_set_to_scalar( FLA_ONE, inv_tau11 ); // FLA_Inv_scalc( FLA_NO_CONJUGATE, tau11, inv_tau11 ); // FLA_Copy( inv_tau11, minus_inv_tau11 ); // FLA_Scal( FLA_MINUS_ONE, minus_inv_tau11 ); bli_cdiv3( buff_1, tau11, &inv_tau11 ); bli_cneg2( &inv_tau11, &minus_inv_tau11 ); // FLA_Copy( a21_t, first_elem ); // FLA_Obj_set_to_scalar( FLA_ONE, a21_t ); first_elem = *a21_t; *a21_t = *buff_1; } if ( m_behind > 0 && m_ahead > 0 ) { // FLA_Her2( FLA_LOWER_TRIANGULAR, FLA_MINUS_ONE, u21, z21, A22 ); // FLA_Hemv( FLA_LOWER_TRIANGULAR, FLA_ONE, A22, a21, FLA_ZERO, w21 ); FLA_Fused_Her2_Ax_opc_var1( m_ahead, buff_m1, u21, inc_u, z21, inc_z, A22, rs_A, cs_A, a21, rs_A, w21, inc_w ); } else if ( m_ahead > 0 ) { // FLA_Hemv( FLA_LOWER_TRIANGULAR, FLA_ONE, A22, a21, FLA_ZERO, w21 ); bli_chemv( BLIS_LOWER_TRIANGULAR, BLIS_NO_CONJUGATE, m_ahead, buff_1, A22, rs_A, cs_A, a21, rs_A, buff_0, w21, inc_w ); } if ( m_ahead > 0 ) { // FLA_Copy( a21, u21 ); // FLA_Copy( w21, z21 ); bli_ccopyv( BLIS_NO_CONJUGATE, m_ahead, a21, rs_A, u21, inc_u ); bli_ccopyv( BLIS_NO_CONJUGATE, m_ahead, w21, inc_w, z21, inc_z ); // FLA_Dotc( FLA_CONJUGATE, a21, z21, beta ); // FLA_Inv_scal( FLA_TWO, beta ); bli_cdot( BLIS_CONJUGATE, m_ahead, a21, rs_A, z21, inc_z, &beta ); bli_cinvscals( buff_2, &beta ); // FLA_Scal( minus_inv_tau11, beta ); // FLA_Axpy( beta, a21, z21 ); // FLA_Scal( inv_tau11, z21 ); bli_cscals( &minus_inv_tau11, &beta ); bli_caxpyv( BLIS_NO_CONJUGATE, m_ahead, &beta, a21, rs_A, z21, inc_z ); bli_cscalv( BLIS_NO_CONJUGATE, m_ahead, &inv_tau11, z21, inc_z ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, t01 ); bli_cgemv( BLIS_CONJ_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_1, A20, rs_A, cs_A, a21, rs_A, buff_0, t01, rs_T ); // FLA_Copy( first_elem, a21_t ); *a21_t = first_elem; } if ( m_behind + 1 == b_alg && m_ahead > 0 ) { // FLA_Her2( FLA_LOWER_TRIANGULAR, FLA_MINUS_ONE, u21, z21, A22 ); bli_cher2( BLIS_LOWER_TRIANGULAR, BLIS_NO_TRANSPOSE, m_ahead, buff_m1, u21, inc_u, z21, inc_z, A22, rs_A, cs_A ); } /*------------------------------------------------------------*/ } // FLA_Obj_free( &u ); // FLA_Obj_free( &z ); // FLA_Obj_free( &w ); FLA_free( buff_u ); FLA_free( buff_z ); FLA_free( buff_w ); return FLA_SUCCESS; }
FLA_Error FLA_Tridiag_UT_l_step_ofc_var3 | ( | int | m_A, |
int | m_T, | ||
scomplex * | buff_A, | ||
int | rs_A, | ||
int | cs_A, | ||
scomplex * | buff_Z, | ||
int | rs_Z, | ||
int | cs_Z, | ||
scomplex * | buff_T, | ||
int | rs_T, | ||
int | cs_T | ||
) |
References bli_caxpyv(), bli_cdot(), bli_cgemv(), bli_chemv(), bli_cscalv(), bli_csetm(), FLA_Fused_UZhu_ZUhu_opc_var1(), FLA_Househ2_UT_l_opc(), FLA_MINUS_ONE, FLA_ONE, FLA_TWO, and FLA_ZERO.
Referenced by FLA_Tridiag_UT_l_step_ofu_var3().
{ scomplex* buff_2 = FLA_COMPLEX_PTR( FLA_TWO ); scomplex* buff_1 = FLA_COMPLEX_PTR( FLA_ONE ); scomplex* buff_0 = FLA_COMPLEX_PTR( FLA_ZERO ); scomplex* buff_m1 = FLA_COMPLEX_PTR( FLA_MINUS_ONE ); scomplex first_elem, last_elem; scomplex beta; scomplex inv_tau11; scomplex minus_inv_tau11; int i; // b_alg = FLA_Obj_length( T ); int b_alg = m_T; // FLA_Obj_set_to_scalar( FLA_ZERO, Z ); bli_csetm( m_A, b_alg, buff_0, buff_Z, rs_Z, cs_Z ); for ( i = 0; i < b_alg; ++i ) { scomplex* a10t = buff_A + (0 )*cs_A + (i )*rs_A; scomplex* A20 = buff_A + (0 )*cs_A + (i+1)*rs_A; scomplex* alpha11 = buff_A + (i )*cs_A + (i )*rs_A; scomplex* a21 = buff_A + (i )*cs_A + (i+1)*rs_A; scomplex* A22 = buff_A + (i+1)*cs_A + (i+1)*rs_A; scomplex* z10t = buff_Z + (0 )*cs_Z + (i )*rs_Z; scomplex* Z20 = buff_Z + (0 )*cs_Z + (i+1)*rs_Z; scomplex* z21 = buff_Z + (i )*cs_Z + (i+1)*rs_Z; scomplex* t01 = buff_T + (i )*cs_T + (0 )*rs_T; scomplex* tau11 = buff_T + (i )*cs_T + (i )*rs_T; scomplex* a10t_r = a10t + (i-1)*cs_A + (0 )*rs_A; scomplex* a21_t = a21 + (0 )*cs_A + (0 )*rs_A; scomplex* a21_b = a21 + (0 )*cs_A + (1 )*rs_A; scomplex* ABL = a10t; scomplex* ZBL = z10t; scomplex* a2 = alpha11; int m_ahead = m_A - i - 1; int m_behind = i; int n_behind = i; /*------------------------------------------------------------*/ if ( m_behind > 0 ) { // FLA_Copy( a10t_r, last_elem ); // FLA_Obj_set_to_scalar( FLA_ONE, a10t_r ); last_elem = *a10t_r; *a10t_r = *buff_1; } // FLA_Gemvc( FLA_NO_TRANSPOSE, FLA_CONJUGATE, FLA_MINUS_ONE, ABL, z10t, FLA_ONE, a2 ); // FLA_Gemvc( FLA_NO_TRANSPOSE, FLA_CONJUGATE, FLA_MINUS_ONE, ZBL, a10t, FLA_ONE, a2 ); bli_cgemv( BLIS_NO_TRANSPOSE, BLIS_CONJUGATE, m_ahead + 1, n_behind, buff_m1, ABL, rs_A, cs_A, z10t, cs_Z, buff_1, a2, rs_A ); bli_cgemv( BLIS_NO_TRANSPOSE, BLIS_CONJUGATE, m_ahead + 1, n_behind, buff_m1, ZBL, rs_Z, cs_Z, a10t, cs_A, buff_1, a2, rs_A ); if ( m_behind > 0 ) { // FLA_Copy( last_elem, a10t_r ); *a10t_r = last_elem; } if ( m_ahead > 0 ) { // FLA_Househ2_UT( FLA_LEFT, // a21_t, // a21_b, tau11 ); FLA_Househ2_UT_l_opc( m_ahead - 1, a21_t, a21_b, rs_A, tau11 ); // FLA_Obj_set_to_scalar( FLA_ONE, inv_tau11 ); // FLA_Inv_scalc( FLA_NO_CONJUGATE, tau11, inv_tau11 ); // FLA_Copy( inv_tau11, minus_inv_tau11 ); // FLA_Scal( FLA_MINUS_ONE, minus_inv_tau11 ); bli_cdiv3( buff_1, tau11, &inv_tau11 ); bli_cneg2( &inv_tau11, &minus_inv_tau11 ); // FLA_Copy( a21_t, first_elem ); // FLA_Obj_set_to_scalar( FLA_ONE, a21_t ); first_elem = *a21_t; *a21_t = *buff_1; // FLA_Hemv( FLA_LOWER_TRIANGULAR, FLA_ONE, A22, a21, FLA_ZERO, z21 ); bli_chemv( BLIS_LOWER_TRIANGULAR, BLIS_NO_CONJUGATE, m_ahead, buff_1, A22, rs_A, cs_A, a21, rs_A, buff_0, z21, rs_Z ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, d01 ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, Z20, a21, FLA_ZERO, f01 ); // FLA_Gemv( FLA_NO_TRANSPOSE, FLA_MINUS_ONE, A20, f01, FLA_ONE, z21 ); // FLA_Gemv( FLA_NO_TRANSPOSE, FLA_MINUS_ONE, Z20, d01, FLA_ONE, z21 ); FLA_Fused_UZhu_ZUhu_opc_var1( m_ahead, n_behind, buff_m1, A20, rs_A, cs_A, Z20, rs_Z, cs_Z, a21, rs_A, z21, rs_Z ); // FLA_Dotc( FLA_CONJUGATE, a21, z21, beta ); // FLA_Inv_scal( FLA_TWO, beta ); bli_cdot( BLIS_CONJUGATE, m_ahead, a21, rs_A, z21, rs_Z, &beta ); bli_cinvscals( buff_2, &beta ); // FLA_Scal( minus_inv_tau11, beta ); // FLA_Axpy( beta, a21, z21 ); // FLA_Scal( inv_tau11, z21 ); bli_cscals( &minus_inv_tau11, &beta ); bli_caxpyv( BLIS_NO_CONJUGATE, m_ahead, &beta, a21, rs_A, z21, rs_Z ); bli_cscalv( BLIS_NO_CONJUGATE, m_ahead, &inv_tau11, z21, rs_Z ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, t01 ); bli_cgemv( BLIS_CONJ_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_1, A20, rs_A, cs_A, a21, rs_A, buff_0, t01, rs_T ); // FLA_Copy( first_elem, a21_t ); *a21_t = first_elem; } /*------------------------------------------------------------*/ } return FLA_SUCCESS; }
FLA_Error FLA_Tridiag_UT_l_step_ofd_var1 | ( | int | m_A, |
int | m_T, | ||
double * | buff_A, | ||
int | rs_A, | ||
int | cs_A, | ||
double * | buff_T, | ||
int | rs_T, | ||
int | cs_T | ||
) |
FLA_Error FLA_Tridiag_UT_l_step_ofd_var2 | ( | int | m_A, |
int | m_T, | ||
double * | buff_A, | ||
int | rs_A, | ||
int | cs_A, | ||
double * | buff_T, | ||
int | rs_T, | ||
int | cs_T | ||
) |
References bli_daxpyv(), bli_dcopyv(), bli_ddot(), bli_dgemv(), bli_dscalv(), bli_dsymv(), bli_dsyr2(), FLA_free(), FLA_Fused_Her2_Ax_opd_var1(), FLA_Househ2_UT_l_opd(), FLA_malloc(), FLA_MINUS_ONE, FLA_ONE, FLA_TWO, and FLA_ZERO.
Referenced by FLA_Tridiag_UT_l_step_ofu_var2().
{ double* buff_2 = FLA_DOUBLE_PTR( FLA_TWO ); double* buff_1 = FLA_DOUBLE_PTR( FLA_ONE ); double* buff_0 = FLA_DOUBLE_PTR( FLA_ZERO ); double* buff_m1 = FLA_DOUBLE_PTR( FLA_MINUS_ONE ); double first_elem; double beta; double inv_tau11; double minus_inv_tau11; double minus_upsilon11, minus_conj_upsilon11; double minus_zeta11, minus_conj_zeta11; int i; // b_alg = FLA_Obj_length( T ); int b_alg = m_T; // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &u ); // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &z ); // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &w ); double* buff_u = ( double* ) FLA_malloc( m_A * sizeof( *buff_A ) ); double* buff_z = ( double* ) FLA_malloc( m_A * sizeof( *buff_A ) ); double* buff_w = ( double* ) FLA_malloc( m_A * sizeof( *buff_A ) ); int inc_u = 1; int inc_z = 1; int inc_w = 1; // Initialize some variables (only to prevent compiler warnings). first_elem = *buff_0; minus_inv_tau11 = *buff_0; for ( i = 0; i < b_alg; ++i ) { double* A20 = buff_A + (0 )*cs_A + (i+1)*rs_A; double* alpha11 = buff_A + (i )*cs_A + (i )*rs_A; double* a21 = buff_A + (i )*cs_A + (i+1)*rs_A; double* A22 = buff_A + (i+1)*cs_A + (i+1)*rs_A; double* t01 = buff_T + (i )*cs_T + (0 )*rs_T; double* tau11 = buff_T + (i )*cs_T + (i )*rs_T; double* upsilon11= buff_u + (i )*inc_u; double* u21 = buff_u + (i+1)*inc_u; double* zeta11 = buff_z + (i )*inc_z; double* z21 = buff_z + (i+1)*inc_z; double* w21 = buff_w + (i+1)*inc_w; double* a21_t = a21 + (0 )*cs_A + (0 )*rs_A; double* a21_b = a21 + (0 )*cs_A + (1 )*rs_A; int m_ahead = m_A - i - 1; int m_behind = i; int n_behind = i; /*------------------------------------------------------------*/ if ( m_behind > 0 ) { // FLA_Copy( upsilon11, minus_upsilon11 ); // FLA_Scal( FLA_MINUS_ONE, minus_upsilon11 ); // FLA_Copy( minus_upsilon11, minus_conj_upsilon11 ); bli_dmult3( buff_m1, upsilon11, &minus_upsilon11 ); bli_dcopyconj( &minus_upsilon11, &minus_conj_upsilon11 ); // FLA_Copy( zeta11, minus_zeta11 ); // FLA_Scal( FLA_MINUS_ONE, minus_zeta11 ); // FLA_Copy( minus_zeta11, minus_conj_zeta11 ); bli_dmult3( buff_m1, zeta11, &minus_zeta11 ); bli_dcopyconj( &minus_zeta11, &minus_conj_zeta11 ); // FLA_Axpyt( FLA_CONJ_NO_TRANSPOSE, minus_upsilon11, zeta11, alpha11 ); // FLA_Axpyt( FLA_CONJ_NO_TRANSPOSE, minus_zeta11, upsilon11, alpha11 ); bli_daxpyv( BLIS_CONJUGATE, 1, &minus_upsilon11, zeta11, 1, alpha11, 1 ); bli_daxpyv( BLIS_CONJUGATE, 1, &minus_zeta11, upsilon11, 1, alpha11, 1 ); // FLA_Axpyt( FLA_NO_TRANSPOSE, minus_conj_zeta11, u21, a21 ); // FLA_Axpyt( FLA_NO_TRANSPOSE, minus_conj_upsilon11, z21, a21 ); bli_daxpyv( BLIS_NO_CONJUGATE, m_ahead, &minus_conj_zeta11, u21, inc_u, a21, rs_A ); bli_daxpyv( BLIS_NO_CONJUGATE, m_ahead, &minus_conj_upsilon11, z21, inc_z, a21, rs_A ); } if ( m_ahead > 0 ) { // FLA_Househ2_UT( FLA_LEFT, // a21_t, // a21_b, tau11 ); FLA_Househ2_UT_l_opd( m_ahead - 1, a21_t, a21_b, rs_A, tau11 ); // FLA_Obj_set_to_scalar( FLA_ONE, inv_tau11 ); // FLA_Inv_scalc( FLA_NO_CONJUGATE, tau11, inv_tau11 ); // FLA_Copy( inv_tau11, minus_inv_tau11 ); // FLA_Scal( FLA_MINUS_ONE, minus_inv_tau11 ); bli_ddiv3( buff_1, tau11, &inv_tau11 ); bli_dneg2( &inv_tau11, &minus_inv_tau11 ); // FLA_Copy( a21_t, first_elem ); // FLA_Obj_set_to_scalar( FLA_ONE, a21_t ); first_elem = *a21_t; *a21_t = *buff_1; } if ( m_behind > 0 && m_ahead > 0 ) { // FLA_Her2( FLA_LOWER_TRIANGULAR, FLA_MINUS_ONE, u21, z21, A22 ); // FLA_Hemv( FLA_LOWER_TRIANGULAR, FLA_ONE, A22, a21, FLA_ZERO, w21 ); FLA_Fused_Her2_Ax_opd_var1( m_ahead, buff_m1, u21, inc_u, z21, inc_z, A22, rs_A, cs_A, a21, rs_A, w21, inc_w ); } else if ( m_ahead > 0 ) { // FLA_Hemv( FLA_LOWER_TRIANGULAR, FLA_ONE, A22, a21, FLA_ZERO, w21 ); bli_dsymv( BLIS_LOWER_TRIANGULAR, m_ahead, buff_1, A22, rs_A, cs_A, a21, rs_A, buff_0, w21, inc_w ); } if ( m_ahead > 0 ) { // FLA_Copy( a21, u21 ); // FLA_Copy( w21, z21 ); bli_dcopyv( BLIS_NO_CONJUGATE, m_ahead, a21, rs_A, u21, inc_u ); bli_dcopyv( BLIS_NO_CONJUGATE, m_ahead, w21, inc_w, z21, inc_z ); // FLA_Dotc( FLA_CONJUGATE, a21, z21, beta ); // FLA_Inv_scal( FLA_TWO, beta ); bli_ddot( BLIS_CONJUGATE, m_ahead, a21, rs_A, z21, inc_z, &beta ); bli_dinvscals( buff_2, &beta ); // FLA_Scal( minus_inv_tau11, beta ); // FLA_Axpy( beta, a21, z21 ); // FLA_Scal( inv_tau11, z21 ); bli_dscals( &minus_inv_tau11, &beta ); bli_daxpyv( BLIS_NO_CONJUGATE, m_ahead, &beta, a21, rs_A, z21, inc_z ); bli_dscalv( BLIS_NO_CONJUGATE, m_ahead, &inv_tau11, z21, inc_z ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, t01 ); bli_dgemv( BLIS_CONJ_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_1, A20, rs_A, cs_A, a21, rs_A, buff_0, t01, rs_T ); // FLA_Copy( first_elem, a21_t ); *a21_t = first_elem; } if ( m_behind + 1 == b_alg && m_ahead > 0 ) { // FLA_Her2( FLA_LOWER_TRIANGULAR, FLA_MINUS_ONE, u21, z21, A22 ); bli_dsyr2( BLIS_LOWER_TRIANGULAR, m_ahead, buff_m1, u21, inc_u, z21, inc_z, A22, rs_A, cs_A ); } /*------------------------------------------------------------*/ } // FLA_Obj_free( &u ); // FLA_Obj_free( &z ); // FLA_Obj_free( &w ); FLA_free( buff_u ); FLA_free( buff_z ); FLA_free( buff_w ); return FLA_SUCCESS; }
FLA_Error FLA_Tridiag_UT_l_step_ofd_var3 | ( | int | m_A, |
int | m_T, | ||
double * | buff_A, | ||
int | rs_A, | ||
int | cs_A, | ||
double * | buff_Z, | ||
int | rs_Z, | ||
int | cs_Z, | ||
double * | buff_T, | ||
int | rs_T, | ||
int | cs_T | ||
) |
References bli_daxpyv(), bli_ddot(), bli_dgemv(), bli_dscalv(), bli_dsetm(), bli_dsymv(), FLA_Fused_UZhu_ZUhu_opd_var1(), FLA_Househ2_UT_l_opd(), FLA_MINUS_ONE, FLA_ONE, FLA_TWO, and FLA_ZERO.
Referenced by FLA_Tridiag_UT_l_step_ofu_var3().
{ double* buff_2 = FLA_DOUBLE_PTR( FLA_TWO ); double* buff_1 = FLA_DOUBLE_PTR( FLA_ONE ); double* buff_0 = FLA_DOUBLE_PTR( FLA_ZERO ); double* buff_m1 = FLA_DOUBLE_PTR( FLA_MINUS_ONE ); double first_elem, last_elem; double beta; double inv_tau11; double minus_inv_tau11; int i; // b_alg = FLA_Obj_length( T ); int b_alg = m_T; // FLA_Obj_set_to_scalar( FLA_ZERO, Z ); bli_dsetm( m_A, b_alg, buff_0, buff_Z, rs_Z, cs_Z ); for ( i = 0; i < b_alg; ++i ) { double* a10t = buff_A + (0 )*cs_A + (i )*rs_A; double* A20 = buff_A + (0 )*cs_A + (i+1)*rs_A; double* alpha11 = buff_A + (i )*cs_A + (i )*rs_A; double* a21 = buff_A + (i )*cs_A + (i+1)*rs_A; double* A22 = buff_A + (i+1)*cs_A + (i+1)*rs_A; double* z10t = buff_Z + (0 )*cs_Z + (i )*rs_Z; double* Z20 = buff_Z + (0 )*cs_Z + (i+1)*rs_Z; double* z21 = buff_Z + (i )*cs_Z + (i+1)*rs_Z; double* t01 = buff_T + (i )*cs_T + (0 )*rs_T; double* tau11 = buff_T + (i )*cs_T + (i )*rs_T; double* a10t_r = a10t + (i-1)*cs_A + (0 )*rs_A; double* a21_t = a21 + (0 )*cs_A + (0 )*rs_A; double* a21_b = a21 + (0 )*cs_A + (1 )*rs_A; double* ABL = a10t; double* ZBL = z10t; double* a2 = alpha11; int m_ahead = m_A - i - 1; int m_behind = i; int n_behind = i; /*------------------------------------------------------------*/ if ( m_behind > 0 ) { // FLA_Copy( a10t_r, last_elem ); // FLA_Obj_set_to_scalar( FLA_ONE, a10t_r ); last_elem = *a10t_r; *a10t_r = *buff_1; } // FLA_Gemvc( FLA_NO_TRANSPOSE, FLA_CONJUGATE, FLA_MINUS_ONE, ABL, z10t, FLA_ONE, a2 ); // FLA_Gemvc( FLA_NO_TRANSPOSE, FLA_CONJUGATE, FLA_MINUS_ONE, ZBL, a10t, FLA_ONE, a2 ); bli_dgemv( BLIS_NO_TRANSPOSE, BLIS_CONJUGATE, m_ahead + 1, n_behind, buff_m1, ABL, rs_A, cs_A, z10t, cs_Z, buff_1, a2, rs_A ); bli_dgemv( BLIS_NO_TRANSPOSE, BLIS_CONJUGATE, m_ahead + 1, n_behind, buff_m1, ZBL, rs_Z, cs_Z, a10t, cs_A, buff_1, a2, rs_A ); if ( m_behind > 0 ) { // FLA_Copy( last_elem, a10t_r ); *a10t_r = last_elem; } if ( m_ahead > 0 ) { // FLA_Househ2_UT( FLA_LEFT, // a21_t, // a21_b, tau11 ); FLA_Househ2_UT_l_opd( m_ahead - 1, a21_t, a21_b, rs_A, tau11 ); // FLA_Obj_set_to_scalar( FLA_ONE, inv_tau11 ); // FLA_Inv_scalc( FLA_NO_CONJUGATE, tau11, inv_tau11 ); // FLA_Copy( inv_tau11, minus_inv_tau11 ); // FLA_Scal( FLA_MINUS_ONE, minus_inv_tau11 ); bli_ddiv3( buff_1, tau11, &inv_tau11 ); bli_dneg2( &inv_tau11, &minus_inv_tau11 ); // FLA_Copy( a21_t, first_elem ); // FLA_Obj_set_to_scalar( FLA_ONE, a21_t ); first_elem = *a21_t; *a21_t = *buff_1; // FLA_Hemv( FLA_LOWER_TRIANGULAR, FLA_ONE, A22, a21, FLA_ZERO, z21 ); bli_dsymv( BLIS_LOWER_TRIANGULAR, m_ahead, buff_1, A22, rs_A, cs_A, a21, rs_A, buff_0, z21, rs_Z ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, d01 ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, Z20, a21, FLA_ZERO, f01 ); // FLA_Gemv( FLA_NO_TRANSPOSE, FLA_MINUS_ONE, A20, f01, FLA_ONE, z21 ); // FLA_Gemv( FLA_NO_TRANSPOSE, FLA_MINUS_ONE, Z20, d01, FLA_ONE, z21 ); FLA_Fused_UZhu_ZUhu_opd_var1( m_ahead, n_behind, buff_m1, A20, rs_A, cs_A, Z20, rs_Z, cs_Z, a21, rs_A, z21, rs_Z ); // FLA_Dotc( FLA_CONJUGATE, a21, z21, beta ); // FLA_Inv_scal( FLA_TWO, beta ); bli_ddot( BLIS_CONJUGATE, m_ahead, a21, rs_A, z21, rs_Z, &beta ); bli_dinvscals( buff_2, &beta ); // FLA_Scal( minus_inv_tau11, beta ); // FLA_Axpy( beta, a21, z21 ); // FLA_Scal( inv_tau11, z21 ); bli_dscals( &minus_inv_tau11, &beta ); bli_daxpyv( BLIS_NO_CONJUGATE, m_ahead, &beta, a21, rs_A, z21, rs_Z ); bli_dscalv( BLIS_NO_CONJUGATE, m_ahead, &inv_tau11, z21, rs_Z ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, t01 ); bli_dgemv( BLIS_CONJ_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_1, A20, rs_A, cs_A, a21, rs_A, buff_0, t01, rs_T ); // FLA_Copy( first_elem, a21_t ); *a21_t = first_elem; } /*------------------------------------------------------------*/ } return FLA_SUCCESS; }
FLA_Error FLA_Tridiag_UT_l_step_ofs_var1 | ( | int | m_A, |
int | m_T, | ||
float * | buff_A, | ||
int | rs_A, | ||
int | cs_A, | ||
float * | buff_T, | ||
int | rs_T, | ||
int | cs_T | ||
) |
FLA_Error FLA_Tridiag_UT_l_step_ofs_var2 | ( | int | m_A, |
int | m_T, | ||
float * | buff_A, | ||
int | rs_A, | ||
int | cs_A, | ||
float * | buff_T, | ||
int | rs_T, | ||
int | cs_T | ||
) |
References bli_saxpyv(), bli_scopyv(), bli_sdot(), bli_sgemv(), bli_sscalv(), bli_ssymv(), bli_ssyr2(), FLA_free(), FLA_Fused_Her2_Ax_ops_var1(), FLA_Househ2_UT_l_ops(), FLA_malloc(), FLA_MINUS_ONE, FLA_ONE, FLA_TWO, and FLA_ZERO.
Referenced by FLA_Tridiag_UT_l_step_ofu_var2().
{ float* buff_2 = FLA_FLOAT_PTR( FLA_TWO ); float* buff_1 = FLA_FLOAT_PTR( FLA_ONE ); float* buff_0 = FLA_FLOAT_PTR( FLA_ZERO ); float* buff_m1 = FLA_FLOAT_PTR( FLA_MINUS_ONE ); float first_elem; float beta; float inv_tau11; float minus_inv_tau11; float minus_upsilon11, minus_conj_upsilon11; float minus_zeta11, minus_conj_zeta11; int i; // b_alg = FLA_Obj_length( T ); int b_alg = m_T; // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &u ); // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &z ); // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &w ); float* buff_u = ( float* ) FLA_malloc( m_A * sizeof( *buff_A ) ); float* buff_z = ( float* ) FLA_malloc( m_A * sizeof( *buff_A ) ); float* buff_w = ( float* ) FLA_malloc( m_A * sizeof( *buff_A ) ); int inc_u = 1; int inc_z = 1; int inc_w = 1; // Initialize some variables (only to prevent compiler warnings). first_elem = *buff_0; minus_inv_tau11 = *buff_0; for ( i = 0; i < b_alg; ++i ) { float* A20 = buff_A + (0 )*cs_A + (i+1)*rs_A; float* alpha11 = buff_A + (i )*cs_A + (i )*rs_A; float* a21 = buff_A + (i )*cs_A + (i+1)*rs_A; float* A22 = buff_A + (i+1)*cs_A + (i+1)*rs_A; float* t01 = buff_T + (i )*cs_T + (0 )*rs_T; float* tau11 = buff_T + (i )*cs_T + (i )*rs_T; float* upsilon11= buff_u + (i )*inc_u; float* u21 = buff_u + (i+1)*inc_u; float* zeta11 = buff_z + (i )*inc_z; float* z21 = buff_z + (i+1)*inc_z; float* w21 = buff_w + (i+1)*inc_w; float* a21_t = a21 + (0 )*cs_A + (0 )*rs_A; float* a21_b = a21 + (0 )*cs_A + (1 )*rs_A; int m_ahead = m_A - i - 1; int m_behind = i; int n_behind = i; /*------------------------------------------------------------*/ if ( m_behind > 0 ) { // FLA_Copy( upsilon11, minus_upsilon11 ); // FLA_Scal( FLA_MINUS_ONE, minus_upsilon11 ); // FLA_Copy( minus_upsilon11, minus_conj_upsilon11 ); bli_smult3( buff_m1, upsilon11, &minus_upsilon11 ); bli_scopyconj( &minus_upsilon11, &minus_conj_upsilon11 ); // FLA_Copy( zeta11, minus_zeta11 ); // FLA_Scal( FLA_MINUS_ONE, minus_zeta11 ); // FLA_Copy( minus_zeta11, minus_conj_zeta11 ); bli_smult3( buff_m1, zeta11, &minus_zeta11 ); bli_scopyconj( &minus_zeta11, &minus_conj_zeta11 ); // FLA_Axpyt( FLA_CONJ_NO_TRANSPOSE, minus_upsilon11, zeta11, alpha11 ); // FLA_Axpyt( FLA_CONJ_NO_TRANSPOSE, minus_zeta11, upsilon11, alpha11 ); bli_saxpyv( BLIS_CONJUGATE, 1, &minus_upsilon11, zeta11, 1, alpha11, 1 ); bli_saxpyv( BLIS_CONJUGATE, 1, &minus_zeta11, upsilon11, 1, alpha11, 1 ); // FLA_Axpyt( FLA_NO_TRANSPOSE, minus_conj_zeta11, u21, a21 ); // FLA_Axpyt( FLA_NO_TRANSPOSE, minus_conj_upsilon11, z21, a21 ); bli_saxpyv( BLIS_NO_CONJUGATE, m_ahead, &minus_conj_zeta11, u21, inc_u, a21, rs_A ); bli_saxpyv( BLIS_NO_CONJUGATE, m_ahead, &minus_conj_upsilon11, z21, inc_z, a21, rs_A ); } if ( m_ahead > 0 ) { // FLA_Househ2_UT( FLA_LEFT, // a21_t, // a21_b, tau11 ); FLA_Househ2_UT_l_ops( m_ahead - 1, a21_t, a21_b, rs_A, tau11 ); // FLA_Obj_set_to_scalar( FLA_ONE, inv_tau11 ); // FLA_Inv_scalc( FLA_NO_CONJUGATE, tau11, inv_tau11 ); // FLA_Copy( inv_tau11, minus_inv_tau11 ); // FLA_Scal( FLA_MINUS_ONE, minus_inv_tau11 ); bli_sdiv3( buff_1, tau11, &inv_tau11 ); bli_sneg2( &inv_tau11, &minus_inv_tau11 ); // FLA_Copy( a21_t, first_elem ); // FLA_Obj_set_to_scalar( FLA_ONE, a21_t ); first_elem = *a21_t; *a21_t = *buff_1; } if ( m_behind > 0 && m_ahead > 0 ) { // FLA_Her2( FLA_LOWER_TRIANGULAR, FLA_MINUS_ONE, u21, z21, A22 ); // FLA_Hemv( FLA_LOWER_TRIANGULAR, FLA_ONE, A22, a21, FLA_ZERO, w21 ); FLA_Fused_Her2_Ax_ops_var1( m_ahead, buff_m1, u21, inc_u, z21, inc_z, A22, rs_A, cs_A, a21, rs_A, w21, inc_w ); } else if ( m_ahead > 0 ) { // FLA_Hemv( FLA_LOWER_TRIANGULAR, FLA_ONE, A22, a21, FLA_ZERO, w21 ); bli_ssymv( BLIS_LOWER_TRIANGULAR, m_ahead, buff_1, A22, rs_A, cs_A, a21, rs_A, buff_0, w21, inc_w ); } if ( m_ahead > 0 ) { // FLA_Copy( a21, u21 ); // FLA_Copy( w21, z21 ); bli_scopyv( BLIS_NO_CONJUGATE, m_ahead, a21, rs_A, u21, inc_u ); bli_scopyv( BLIS_NO_CONJUGATE, m_ahead, w21, inc_w, z21, inc_z ); // FLA_Dotc( FLA_CONJUGATE, a21, z21, beta ); // FLA_Inv_scal( FLA_TWO, beta ); bli_sdot( BLIS_CONJUGATE, m_ahead, a21, rs_A, z21, inc_z, &beta ); bli_sinvscals( buff_2, &beta ); // FLA_Scal( minus_inv_tau11, beta ); // FLA_Axpy( beta, a21, z21 ); // FLA_Scal( inv_tau11, z21 ); bli_sscals( &minus_inv_tau11, &beta ); bli_saxpyv( BLIS_NO_CONJUGATE, m_ahead, &beta, a21, rs_A, z21, inc_z ); bli_sscalv( BLIS_NO_CONJUGATE, m_ahead, &inv_tau11, z21, inc_z ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, t01 ); bli_sgemv( BLIS_CONJ_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_1, A20, rs_A, cs_A, a21, rs_A, buff_0, t01, rs_T ); // FLA_Copy( first_elem, a21_t ); *a21_t = first_elem; } if ( m_behind + 1 == b_alg && m_ahead > 0 ) { // FLA_Her2( FLA_LOWER_TRIANGULAR, FLA_MINUS_ONE, u21, z21, A22 ); bli_ssyr2( BLIS_LOWER_TRIANGULAR, m_ahead, buff_m1, u21, inc_u, z21, inc_z, A22, rs_A, cs_A ); } /*------------------------------------------------------------*/ } // FLA_Obj_free( &u ); // FLA_Obj_free( &z ); // FLA_Obj_free( &w ); FLA_free( buff_u ); FLA_free( buff_z ); FLA_free( buff_w ); return FLA_SUCCESS; }
FLA_Error FLA_Tridiag_UT_l_step_ofs_var3 | ( | int | m_A, |
int | m_T, | ||
float * | buff_A, | ||
int | rs_A, | ||
int | cs_A, | ||
float * | buff_Z, | ||
int | rs_Z, | ||
int | cs_Z, | ||
float * | buff_T, | ||
int | rs_T, | ||
int | cs_T | ||
) |
References bli_saxpyv(), bli_sdot(), bli_sgemv(), bli_sscalv(), bli_ssetm(), bli_ssymv(), FLA_Fused_UZhu_ZUhu_ops_var1(), FLA_Househ2_UT_l_ops(), FLA_MINUS_ONE, FLA_ONE, FLA_TWO, and FLA_ZERO.
Referenced by FLA_Tridiag_UT_l_step_ofu_var3().
{ float* buff_2 = FLA_FLOAT_PTR( FLA_TWO ); float* buff_1 = FLA_FLOAT_PTR( FLA_ONE ); float* buff_0 = FLA_FLOAT_PTR( FLA_ZERO ); float* buff_m1 = FLA_FLOAT_PTR( FLA_MINUS_ONE ); float first_elem, last_elem; float beta; float inv_tau11; float minus_inv_tau11; int i; // b_alg = FLA_Obj_length( T ); int b_alg = m_T; // FLA_Obj_set_to_scalar( FLA_ZERO, Z ); bli_ssetm( m_A, b_alg, buff_0, buff_Z, rs_Z, cs_Z ); for ( i = 0; i < b_alg; ++i ) { float* a10t = buff_A + (0 )*cs_A + (i )*rs_A; float* A20 = buff_A + (0 )*cs_A + (i+1)*rs_A; float* alpha11 = buff_A + (i )*cs_A + (i )*rs_A; float* a21 = buff_A + (i )*cs_A + (i+1)*rs_A; float* A22 = buff_A + (i+1)*cs_A + (i+1)*rs_A; float* z10t = buff_Z + (0 )*cs_Z + (i )*rs_Z; float* Z20 = buff_Z + (0 )*cs_Z + (i+1)*rs_Z; float* z21 = buff_Z + (i )*cs_Z + (i+1)*rs_Z; float* t01 = buff_T + (i )*cs_T + (0 )*rs_T; float* tau11 = buff_T + (i )*cs_T + (i )*rs_T; float* a10t_r = a10t + (i-1)*cs_A + (0 )*rs_A; float* a21_t = a21 + (0 )*cs_A + (0 )*rs_A; float* a21_b = a21 + (0 )*cs_A + (1 )*rs_A; float* ABL = a10t; float* ZBL = z10t; float* a2 = alpha11; int m_ahead = m_A - i - 1; int m_behind = i; int n_behind = i; /*------------------------------------------------------------*/ if ( m_behind > 0 ) { // FLA_Copy( a10t_r, last_elem ); // FLA_Obj_set_to_scalar( FLA_ONE, a10t_r ); last_elem = *a10t_r; *a10t_r = *buff_1; } // FLA_Gemvc( FLA_NO_TRANSPOSE, FLA_CONJUGATE, FLA_MINUS_ONE, ABL, z10t, FLA_ONE, a2 ); // FLA_Gemvc( FLA_NO_TRANSPOSE, FLA_CONJUGATE, FLA_MINUS_ONE, ZBL, a10t, FLA_ONE, a2 ); bli_sgemv( BLIS_NO_TRANSPOSE, BLIS_CONJUGATE, m_ahead + 1, n_behind, buff_m1, ABL, rs_A, cs_A, z10t, cs_Z, buff_1, a2, rs_A ); bli_sgemv( BLIS_NO_TRANSPOSE, BLIS_CONJUGATE, m_ahead + 1, n_behind, buff_m1, ZBL, rs_Z, cs_Z, a10t, cs_A, buff_1, a2, rs_A ); if ( m_behind > 0 ) { // FLA_Copy( last_elem, a10t_r ); *a10t_r = last_elem; } if ( m_ahead > 0 ) { // FLA_Househ2_UT( FLA_LEFT, // a21_t, // a21_b, tau11 ); FLA_Househ2_UT_l_ops( m_ahead - 1, a21_t, a21_b, rs_A, tau11 ); // FLA_Obj_set_to_scalar( FLA_ONE, inv_tau11 ); // FLA_Inv_scalc( FLA_NO_CONJUGATE, tau11, inv_tau11 ); // FLA_Copy( inv_tau11, minus_inv_tau11 ); // FLA_Scal( FLA_MINUS_ONE, minus_inv_tau11 ); bli_sdiv3( buff_1, tau11, &inv_tau11 ); bli_sneg2( &inv_tau11, &minus_inv_tau11 ); // FLA_Copy( a21_t, first_elem ); // FLA_Obj_set_to_scalar( FLA_ONE, a21_t ); first_elem = *a21_t; *a21_t = *buff_1; // FLA_Hemv( FLA_LOWER_TRIANGULAR, FLA_ONE, A22, a21, FLA_ZERO, z21 ); bli_ssymv( BLIS_LOWER_TRIANGULAR, m_ahead, buff_1, A22, rs_A, cs_A, a21, rs_A, buff_0, z21, rs_Z ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, d01 ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, Z20, a21, FLA_ZERO, f01 ); // FLA_Gemv( FLA_NO_TRANSPOSE, FLA_MINUS_ONE, A20, f01, FLA_ONE, z21 ); // FLA_Gemv( FLA_NO_TRANSPOSE, FLA_MINUS_ONE, Z20, d01, FLA_ONE, z21 ); FLA_Fused_UZhu_ZUhu_ops_var1( m_ahead, n_behind, buff_m1, A20, rs_A, cs_A, Z20, rs_Z, cs_Z, a21, rs_A, z21, rs_Z ); // FLA_Dotc( FLA_CONJUGATE, a21, z21, beta ); // FLA_Inv_scal( FLA_TWO, beta ); bli_sdot( BLIS_CONJUGATE, m_ahead, a21, rs_A, z21, rs_Z, &beta ); bli_sinvscals( buff_2, &beta ); // FLA_Scal( minus_inv_tau11, beta ); // FLA_Axpy( beta, a21, z21 ); // FLA_Scal( inv_tau11, z21 ); bli_sscals( &minus_inv_tau11, &beta ); bli_saxpyv( BLIS_NO_CONJUGATE, m_ahead, &beta, a21, rs_A, z21, rs_Z ); bli_sscalv( BLIS_NO_CONJUGATE, m_ahead, &inv_tau11, z21, rs_Z ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, t01 ); bli_sgemv( BLIS_CONJ_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_1, A20, rs_A, cs_A, a21, rs_A, buff_0, t01, rs_T ); // FLA_Copy( first_elem, a21_t ); *a21_t = first_elem; } /*------------------------------------------------------------*/ } return FLA_SUCCESS; }
References FLA_Obj_col_stride(), FLA_Obj_datatype(), FLA_Obj_length(), FLA_Obj_row_stride(), FLA_Tridiag_UT_l_step_ofc_var2(), FLA_Tridiag_UT_l_step_ofd_var2(), FLA_Tridiag_UT_l_step_ofs_var2(), and FLA_Tridiag_UT_l_step_ofz_var2().
Referenced by FLA_Tridiag_UT_l_ofu_var2().
{ FLA_Datatype datatype; int m_A, m_T; int rs_A, cs_A; int rs_T, cs_T; datatype = FLA_Obj_datatype( A ); m_A = FLA_Obj_length( A ); m_T = FLA_Obj_length( T ); rs_A = FLA_Obj_row_stride( A ); cs_A = FLA_Obj_col_stride( A ); 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 ); FLA_Tridiag_UT_l_step_ofs_var2( m_A, m_T, buff_A, rs_A, cs_A, buff_T, rs_T, cs_T ); break; } case FLA_DOUBLE: { double* buff_A = FLA_DOUBLE_PTR( A ); double* buff_T = FLA_DOUBLE_PTR( T ); FLA_Tridiag_UT_l_step_ofd_var2( m_A, m_T, buff_A, rs_A, cs_A, buff_T, rs_T, cs_T ); break; } case FLA_COMPLEX: { scomplex* buff_A = FLA_COMPLEX_PTR( A ); scomplex* buff_T = FLA_COMPLEX_PTR( T ); FLA_Tridiag_UT_l_step_ofc_var2( m_A, m_T, buff_A, rs_A, cs_A, 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 ); FLA_Tridiag_UT_l_step_ofz_var2( m_A, m_T, buff_A, rs_A, cs_A, buff_T, rs_T, cs_T ); break; } } return FLA_SUCCESS; }
References FLA_Obj_col_stride(), FLA_Obj_datatype(), FLA_Obj_length(), FLA_Obj_row_stride(), FLA_Tridiag_UT_l_step_ofc_var3(), FLA_Tridiag_UT_l_step_ofd_var3(), FLA_Tridiag_UT_l_step_ofs_var3(), and FLA_Tridiag_UT_l_step_ofz_var3().
Referenced by FLA_Tridiag_UT_l_ofu_var3().
{ FLA_Datatype datatype; int m_A, m_T; int rs_A, cs_A; int rs_Z, cs_Z; int rs_T, cs_T; datatype = FLA_Obj_datatype( A ); m_A = FLA_Obj_length( A ); m_T = FLA_Obj_length( T ); rs_A = FLA_Obj_row_stride( A ); cs_A = FLA_Obj_col_stride( A ); rs_Z = FLA_Obj_row_stride( Z ); cs_Z = FLA_Obj_col_stride( Z ); 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_Z = FLA_FLOAT_PTR( Z ); float* buff_T = FLA_FLOAT_PTR( T ); FLA_Tridiag_UT_l_step_ofs_var3( m_A, m_T, buff_A, rs_A, cs_A, buff_Z, rs_Z, cs_Z, buff_T, rs_T, cs_T ); break; } case FLA_DOUBLE: { double* buff_A = FLA_DOUBLE_PTR( A ); double* buff_Z = FLA_DOUBLE_PTR( Z ); double* buff_T = FLA_DOUBLE_PTR( T ); FLA_Tridiag_UT_l_step_ofd_var3( m_A, m_T, buff_A, rs_A, cs_A, buff_Z, rs_Z, cs_Z, buff_T, rs_T, cs_T ); break; } case FLA_COMPLEX: { scomplex* buff_A = FLA_COMPLEX_PTR( A ); scomplex* buff_Z = FLA_COMPLEX_PTR( Z ); scomplex* buff_T = FLA_COMPLEX_PTR( T ); FLA_Tridiag_UT_l_step_ofc_var3( m_A, m_T, buff_A, rs_A, cs_A, buff_Z, rs_Z, cs_Z, buff_T, rs_T, cs_T ); break; } case FLA_DOUBLE_COMPLEX: { dcomplex* buff_A = FLA_DOUBLE_COMPLEX_PTR( A ); dcomplex* buff_Z = FLA_DOUBLE_COMPLEX_PTR( Z ); dcomplex* buff_T = FLA_DOUBLE_COMPLEX_PTR( T ); FLA_Tridiag_UT_l_step_ofz_var3( m_A, m_T, buff_A, rs_A, cs_A, buff_Z, rs_Z, cs_Z, buff_T, rs_T, cs_T ); break; } } return FLA_SUCCESS; }
FLA_Error FLA_Tridiag_UT_l_step_ofz_var1 | ( | int | m_A, |
int | m_T, | ||
dcomplex * | buff_A, | ||
int | rs_A, | ||
int | cs_A, | ||
dcomplex * | buff_T, | ||
int | rs_T, | ||
int | cs_T | ||
) |
FLA_Error FLA_Tridiag_UT_l_step_ofz_var2 | ( | int | m_A, |
int | m_T, | ||
dcomplex * | buff_A, | ||
int | rs_A, | ||
int | cs_A, | ||
dcomplex * | buff_T, | ||
int | rs_T, | ||
int | cs_T | ||
) |
References bli_zaxpyv(), bli_zcopyv(), bli_zdot(), bli_zgemv(), bli_zhemv(), bli_zher2(), bli_zscalv(), FLA_free(), FLA_Fused_Her2_Ax_opz_var1(), FLA_Househ2_UT_l_opz(), FLA_malloc(), FLA_MINUS_ONE, FLA_ONE, FLA_TWO, and FLA_ZERO.
Referenced by FLA_Tridiag_UT_l_step_ofu_var2().
{ dcomplex* buff_2 = FLA_DOUBLE_COMPLEX_PTR( FLA_TWO ); dcomplex* buff_1 = FLA_DOUBLE_COMPLEX_PTR( FLA_ONE ); dcomplex* buff_0 = FLA_DOUBLE_COMPLEX_PTR( FLA_ZERO ); dcomplex* buff_m1 = FLA_DOUBLE_COMPLEX_PTR( FLA_MINUS_ONE ); dcomplex first_elem; dcomplex beta; dcomplex inv_tau11; dcomplex minus_inv_tau11; dcomplex minus_upsilon11, minus_conj_upsilon11; dcomplex minus_zeta11, minus_conj_zeta11; int i; // b_alg = FLA_Obj_length( T ); int b_alg = m_T; // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &u ); // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &z ); // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &w ); dcomplex* buff_u = ( dcomplex* ) FLA_malloc( m_A * sizeof( *buff_A ) ); dcomplex* buff_z = ( dcomplex* ) FLA_malloc( m_A * sizeof( *buff_A ) ); dcomplex* buff_w = ( dcomplex* ) FLA_malloc( m_A * sizeof( *buff_A ) ); int inc_u = 1; int inc_z = 1; int inc_w = 1; // Initialize some variables (only to prevent compiler warnings). first_elem = *buff_0; minus_inv_tau11 = *buff_0; for ( i = 0; i < b_alg; ++i ) { dcomplex* A20 = buff_A + (0 )*cs_A + (i+1)*rs_A; dcomplex* alpha11 = buff_A + (i )*cs_A + (i )*rs_A; dcomplex* a21 = buff_A + (i )*cs_A + (i+1)*rs_A; dcomplex* A22 = buff_A + (i+1)*cs_A + (i+1)*rs_A; dcomplex* t01 = buff_T + (i )*cs_T + (0 )*rs_T; dcomplex* tau11 = buff_T + (i )*cs_T + (i )*rs_T; dcomplex* upsilon11= buff_u + (i )*inc_u; dcomplex* u21 = buff_u + (i+1)*inc_u; dcomplex* zeta11 = buff_z + (i )*inc_z; dcomplex* z21 = buff_z + (i+1)*inc_z; dcomplex* w21 = buff_w + (i+1)*inc_w; dcomplex* a21_t = a21 + (0 )*cs_A + (0 )*rs_A; dcomplex* a21_b = a21 + (0 )*cs_A + (1 )*rs_A; int m_ahead = m_A - i - 1; int m_behind = i; int n_behind = i; /*------------------------------------------------------------*/ if ( m_behind > 0 ) { // FLA_Copy( upsilon11, minus_upsilon11 ); // FLA_Scal( FLA_MINUS_ONE, minus_upsilon11 ); // FLA_Copy( minus_upsilon11, minus_conj_upsilon11 ); bli_zmult3( buff_m1, upsilon11, &minus_upsilon11 ); bli_zcopyconj( &minus_upsilon11, &minus_conj_upsilon11 ); // FLA_Copy( zeta11, minus_zeta11 ); // FLA_Scal( FLA_MINUS_ONE, minus_zeta11 ); // FLA_Copy( minus_zeta11, minus_conj_zeta11 ); bli_zmult3( buff_m1, zeta11, &minus_zeta11 ); bli_zcopyconj( &minus_zeta11, &minus_conj_zeta11 ); // FLA_Axpyt( FLA_CONJ_NO_TRANSPOSE, minus_upsilon11, zeta11, alpha11 ); // FLA_Axpyt( FLA_CONJ_NO_TRANSPOSE, minus_zeta11, upsilon11, alpha11 ); bli_zaxpyv( BLIS_CONJUGATE, 1, &minus_upsilon11, zeta11, 1, alpha11, 1 ); bli_zaxpyv( BLIS_CONJUGATE, 1, &minus_zeta11, upsilon11, 1, alpha11, 1 ); // FLA_Axpyt( FLA_NO_TRANSPOSE, minus_conj_zeta11, u21, a21 ); // FLA_Axpyt( FLA_NO_TRANSPOSE, minus_conj_upsilon11, z21, a21 ); bli_zaxpyv( BLIS_NO_CONJUGATE, m_ahead, &minus_conj_zeta11, u21, inc_u, a21, rs_A ); bli_zaxpyv( BLIS_NO_CONJUGATE, m_ahead, &minus_conj_upsilon11, z21, inc_z, a21, rs_A ); } if ( m_ahead > 0 ) { // FLA_Househ2_UT( FLA_LEFT, // a21_t, // a21_b, tau11 ); FLA_Househ2_UT_l_opz( m_ahead - 1, a21_t, a21_b, rs_A, tau11 ); // FLA_Obj_set_to_scalar( FLA_ONE, inv_tau11 ); // FLA_Inv_scalc( FLA_NO_CONJUGATE, tau11, inv_tau11 ); // FLA_Copy( inv_tau11, minus_inv_tau11 ); // FLA_Scal( FLA_MINUS_ONE, minus_inv_tau11 ); bli_zdiv3( buff_1, tau11, &inv_tau11 ); bli_zneg2( &inv_tau11, &minus_inv_tau11 ); // FLA_Copy( a21_t, first_elem ); // FLA_Obj_set_to_scalar( FLA_ONE, a21_t ); first_elem = *a21_t; *a21_t = *buff_1; } if ( m_behind > 0 && m_ahead > 0 ) { // FLA_Her2( FLA_LOWER_TRIANGULAR, FLA_MINUS_ONE, u21, z21, A22 ); // FLA_Hemv( FLA_LOWER_TRIANGULAR, FLA_ONE, A22, a21, FLA_ZERO, w21 ); FLA_Fused_Her2_Ax_opz_var1( m_ahead, buff_m1, u21, inc_u, z21, inc_z, A22, rs_A, cs_A, a21, rs_A, w21, inc_w ); } else if ( m_ahead > 0 ) { // FLA_Hemv( FLA_LOWER_TRIANGULAR, FLA_ONE, A22, a21, FLA_ZERO, w21 ); bli_zhemv( BLIS_LOWER_TRIANGULAR, BLIS_NO_CONJUGATE, m_ahead, buff_1, A22, rs_A, cs_A, a21, rs_A, buff_0, w21, inc_w ); } if ( m_ahead > 0 ) { // FLA_Copy( a21, u21 ); // FLA_Copy( w21, z21 ); bli_zcopyv( BLIS_NO_CONJUGATE, m_ahead, a21, rs_A, u21, inc_u ); bli_zcopyv( BLIS_NO_CONJUGATE, m_ahead, w21, inc_w, z21, inc_z ); // FLA_Dotc( FLA_CONJUGATE, a21, z21, beta ); // FLA_Inv_scal( FLA_TWO, beta ); bli_zdot( BLIS_CONJUGATE, m_ahead, a21, rs_A, z21, inc_z, &beta ); bli_zinvscals( buff_2, &beta ); // FLA_Scal( minus_inv_tau11, beta ); // FLA_Axpy( beta, a21, z21 ); // FLA_Scal( inv_tau11, z21 ); bli_zscals( &minus_inv_tau11, &beta ); bli_zaxpyv( BLIS_NO_CONJUGATE, m_ahead, &beta, a21, rs_A, z21, inc_z ); bli_zscalv( BLIS_NO_CONJUGATE, m_ahead, &inv_tau11, z21, inc_z ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, t01 ); bli_zgemv( BLIS_CONJ_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_1, A20, rs_A, cs_A, a21, rs_A, buff_0, t01, rs_T ); // FLA_Copy( first_elem, a21_t ); *a21_t = first_elem; } if ( m_behind + 1 == b_alg && m_ahead > 0 ) { // FLA_Her2( FLA_LOWER_TRIANGULAR, FLA_MINUS_ONE, u21, z21, A22 ); bli_zher2( BLIS_LOWER_TRIANGULAR, BLIS_NO_TRANSPOSE, m_ahead, buff_m1, u21, inc_u, z21, inc_z, A22, rs_A, cs_A ); } /*------------------------------------------------------------*/ } // FLA_Obj_free( &u ); // FLA_Obj_free( &z ); // FLA_Obj_free( &w ); FLA_free( buff_u ); FLA_free( buff_z ); FLA_free( buff_w ); return FLA_SUCCESS; }
FLA_Error FLA_Tridiag_UT_l_step_ofz_var3 | ( | int | m_A, |
int | m_T, | ||
dcomplex * | buff_A, | ||
int | rs_A, | ||
int | cs_A, | ||
dcomplex * | buff_Z, | ||
int | rs_Z, | ||
int | cs_Z, | ||
dcomplex * | buff_T, | ||
int | rs_T, | ||
int | cs_T | ||
) |
References bli_zaxpyv(), bli_zdot(), bli_zgemv(), bli_zhemv(), bli_zscalv(), bli_zsetm(), FLA_Fused_UZhu_ZUhu_opz_var1(), FLA_Househ2_UT_l_opz(), FLA_MINUS_ONE, FLA_ONE, FLA_TWO, and FLA_ZERO.
Referenced by FLA_Tridiag_UT_l_step_ofu_var3().
{ dcomplex* buff_2 = FLA_DOUBLE_COMPLEX_PTR( FLA_TWO ); dcomplex* buff_1 = FLA_DOUBLE_COMPLEX_PTR( FLA_ONE ); dcomplex* buff_0 = FLA_DOUBLE_COMPLEX_PTR( FLA_ZERO ); dcomplex* buff_m1 = FLA_DOUBLE_COMPLEX_PTR( FLA_MINUS_ONE ); dcomplex first_elem, last_elem; dcomplex beta; dcomplex inv_tau11; dcomplex minus_inv_tau11; int i; // b_alg = FLA_Obj_length( T ); int b_alg = m_T; // FLA_Obj_set_to_scalar( FLA_ZERO, Z ); bli_zsetm( m_A, b_alg, buff_0, buff_Z, rs_Z, cs_Z ); for ( i = 0; i < b_alg; ++i ) { dcomplex* a10t = buff_A + (0 )*cs_A + (i )*rs_A; dcomplex* A20 = buff_A + (0 )*cs_A + (i+1)*rs_A; dcomplex* alpha11 = buff_A + (i )*cs_A + (i )*rs_A; dcomplex* a21 = buff_A + (i )*cs_A + (i+1)*rs_A; dcomplex* A22 = buff_A + (i+1)*cs_A + (i+1)*rs_A; dcomplex* z10t = buff_Z + (0 )*cs_Z + (i )*rs_Z; dcomplex* Z20 = buff_Z + (0 )*cs_Z + (i+1)*rs_Z; dcomplex* z21 = buff_Z + (i )*cs_Z + (i+1)*rs_Z; dcomplex* t01 = buff_T + (i )*cs_T + (0 )*rs_T; dcomplex* tau11 = buff_T + (i )*cs_T + (i )*rs_T; dcomplex* a10t_r = a10t + (i-1)*cs_A + (0 )*rs_A; dcomplex* a21_t = a21 + (0 )*cs_A + (0 )*rs_A; dcomplex* a21_b = a21 + (0 )*cs_A + (1 )*rs_A; dcomplex* ABL = a10t; dcomplex* ZBL = z10t; dcomplex* a2 = alpha11; int m_ahead = m_A - i - 1; int m_behind = i; int n_behind = i; /*------------------------------------------------------------*/ if ( m_behind > 0 ) { // FLA_Copy( a10t_r, last_elem ); // FLA_Obj_set_to_scalar( FLA_ONE, a10t_r ); last_elem = *a10t_r; *a10t_r = *buff_1; } // FLA_Gemvc( FLA_NO_TRANSPOSE, FLA_CONJUGATE, FLA_MINUS_ONE, ABL, z10t, FLA_ONE, a2 ); // FLA_Gemvc( FLA_NO_TRANSPOSE, FLA_CONJUGATE, FLA_MINUS_ONE, ZBL, a10t, FLA_ONE, a2 ); bli_zgemv( BLIS_NO_TRANSPOSE, BLIS_CONJUGATE, m_ahead + 1, n_behind, buff_m1, ABL, rs_A, cs_A, z10t, cs_Z, buff_1, a2, rs_A ); bli_zgemv( BLIS_NO_TRANSPOSE, BLIS_CONJUGATE, m_ahead + 1, n_behind, buff_m1, ZBL, rs_Z, cs_Z, a10t, cs_A, buff_1, a2, rs_A ); if ( m_behind > 0 ) { // FLA_Copy( last_elem, a10t_r ); *a10t_r = last_elem; } if ( m_ahead > 0 ) { // FLA_Househ2_UT( FLA_LEFT, // a21_t, // a21_b, tau11 ); FLA_Househ2_UT_l_opz( m_ahead - 1, a21_t, a21_b, rs_A, tau11 ); // FLA_Obj_set_to_scalar( FLA_ONE, inv_tau11 ); // FLA_Inv_scalc( FLA_NO_CONJUGATE, tau11, inv_tau11 ); // FLA_Copy( inv_tau11, minus_inv_tau11 ); // FLA_Scal( FLA_MINUS_ONE, minus_inv_tau11 ); bli_zdiv3( buff_1, tau11, &inv_tau11 ); bli_zneg2( &inv_tau11, &minus_inv_tau11 ); // FLA_Copy( a21_t, first_elem ); // FLA_Obj_set_to_scalar( FLA_ONE, a21_t ); first_elem = *a21_t; *a21_t = *buff_1; // FLA_Hemv( FLA_LOWER_TRIANGULAR, FLA_ONE, A22, a21, FLA_ZERO, z21 ); bli_zhemv( BLIS_LOWER_TRIANGULAR, BLIS_NO_CONJUGATE, m_ahead, buff_1, A22, rs_A, cs_A, a21, rs_A, buff_0, z21, rs_Z ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, d01 ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, Z20, a21, FLA_ZERO, f01 ); // FLA_Gemv( FLA_NO_TRANSPOSE, FLA_MINUS_ONE, A20, f01, FLA_ONE, z21 ); // FLA_Gemv( FLA_NO_TRANSPOSE, FLA_MINUS_ONE, Z20, d01, FLA_ONE, z21 ); FLA_Fused_UZhu_ZUhu_opz_var1( m_ahead, n_behind, buff_m1, A20, rs_A, cs_A, Z20, rs_Z, cs_Z, a21, rs_A, z21, rs_Z ); // FLA_Dotc( FLA_CONJUGATE, a21, z21, beta ); // FLA_Inv_scal( FLA_TWO, beta ); bli_zdot( BLIS_CONJUGATE, m_ahead, a21, rs_A, z21, rs_Z, &beta ); bli_zinvscals( buff_2, &beta ); // FLA_Scal( minus_inv_tau11, beta ); // FLA_Axpy( beta, a21, z21 ); // FLA_Scal( inv_tau11, z21 ); bli_zscals( &minus_inv_tau11, &beta ); bli_zaxpyv( BLIS_NO_CONJUGATE, m_ahead, &beta, a21, rs_A, z21, rs_Z ); bli_zscalv( BLIS_NO_CONJUGATE, m_ahead, &inv_tau11, z21, rs_Z ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, t01 ); bli_zgemv( BLIS_CONJ_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_1, A20, rs_A, cs_A, a21, rs_A, buff_0, t01, rs_T ); // FLA_Copy( first_elem, a21_t ); *a21_t = first_elem; } /*------------------------------------------------------------*/ } return FLA_SUCCESS; }
FLA_Error FLA_Tridiag_UT_l_step_opc_var1 | ( | int | m_A, |
int | m_T, | ||
scomplex * | buff_A, | ||
int | rs_A, | ||
int | cs_A, | ||
scomplex * | buff_T, | ||
int | rs_T, | ||
int | cs_T | ||
) |
References bli_caxpyv(), bli_cdot(), bli_cgemv(), bli_chemv(), bli_cher2(), bli_cscalv(), FLA_free(), FLA_Househ2_UT_l_opc(), FLA_malloc(), FLA_MINUS_ONE, FLA_ONE, FLA_TWO, and FLA_ZERO.
Referenced by FLA_Tridiag_UT_l_step_opt_var1().
{ scomplex* buff_2 = FLA_COMPLEX_PTR( FLA_TWO ); scomplex* buff_1 = FLA_COMPLEX_PTR( FLA_ONE ); scomplex* buff_0 = FLA_COMPLEX_PTR( FLA_ZERO ); scomplex* buff_m1 = FLA_COMPLEX_PTR( FLA_MINUS_ONE ); scomplex first_elem; scomplex beta; scomplex inv_tau11; scomplex minus_inv_tau11; int i; // b_alg = FLA_Obj_length( T ); int b_alg = m_T; // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &z ); scomplex* buff_z = ( scomplex* ) FLA_malloc( m_A * sizeof( *buff_A ) ); int inc_z = 1; for ( i = 0; i < b_alg; ++i ) { scomplex* A20 = buff_A + (0 )*cs_A + (i+1)*rs_A; scomplex* a21 = buff_A + (i )*cs_A + (i+1)*rs_A; scomplex* A22 = buff_A + (i+1)*cs_A + (i+1)*rs_A; scomplex* t01 = buff_T + (i )*cs_T + (0 )*rs_T; scomplex* tau11 = buff_T + (i )*cs_T + (i )*rs_T; scomplex* z21 = buff_z + (i+1)*inc_z; scomplex* a21_t = a21 + (0 )*cs_A + (0 )*rs_A; scomplex* a21_b = a21 + (0 )*cs_A + (1 )*rs_A; int m_ahead = m_A - i - 1; int n_behind = i; /*------------------------------------------------------------*/ if ( m_ahead > 0 ) { // FLA_Househ2_UT( FLA_LEFT, // a21_t, // a21_b, tau11 ); FLA_Househ2_UT_l_opc( m_ahead - 1, a21_t, a21_b, rs_A, tau11 ); // FLA_Obj_set_to_scalar( FLA_ONE, inv_tau11 ); // FLA_Inv_scalc( FLA_NO_CONJUGATE, tau11, inv_tau11 ); // FLA_Copy( inv_tau11, minus_inv_tau11 ); // FLA_Scal( FLA_MINUS_ONE, minus_inv_tau11 ); bli_cdiv3( buff_1, tau11, &inv_tau11 ); bli_cneg2( &inv_tau11, &minus_inv_tau11 ); // FLA_Copy( a21_t, first_elem ); // FLA_Obj_set_to_scalar( FLA_ONE, a21_t ); first_elem = *a21_t; *a21_t = *buff_1; // FLA_Hemv( FLA_LOWER_TRIANGULAR, FLA_ONE, A22, a21, FLA_ZERO, z21 ); bli_chemv( BLIS_LOWER_TRIANGULAR, BLIS_NO_CONJUGATE, m_ahead, buff_1, A22, rs_A, cs_A, a21, rs_A, buff_0, z21, inc_z ); // FLA_Dotc( FLA_CONJUGATE, a21, z21, beta ); // FLA_Inv_scal( FLA_TWO, beta ); bli_cdot( BLIS_CONJUGATE, m_ahead, a21, rs_A, z21, inc_z, &beta ); bli_cinvscals( buff_2, &beta ); // FLA_Scal( minus_inv_tau11, beta ); // FLA_Axpy( beta, a21, z21 ); // FLA_Scal( inv_tau11, z21 ); bli_cscals( &minus_inv_tau11, &beta ); bli_caxpyv( BLIS_NO_CONJUGATE, m_ahead, &beta, a21, rs_A, z21, inc_z ); bli_cscalv( BLIS_NO_CONJUGATE, m_ahead, &inv_tau11, z21, inc_z ); // FLA_Her2( FLA_LOWER_TRIANGULAR, FLA_MINUS_ONE, a21, z21, A22 ); bli_cher2( BLIS_LOWER_TRIANGULAR, BLIS_NO_TRANSPOSE, m_ahead, buff_m1, a21, rs_A, z21, inc_z, A22, rs_A, cs_A ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, t01 ); bli_cgemv( BLIS_CONJ_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_1, A20, rs_A, cs_A, a21, rs_A, buff_0, t01, rs_T ); // FLA_Copy( first_elem, a21_t ); *a21_t = first_elem; } /*------------------------------------------------------------*/ } // FLA_Obj_free( &z ); FLA_free( buff_z ); return FLA_SUCCESS; }
FLA_Error FLA_Tridiag_UT_l_step_opc_var2 | ( | int | m_A, |
int | m_T, | ||
scomplex * | buff_A, | ||
int | rs_A, | ||
int | cs_A, | ||
scomplex * | buff_T, | ||
int | rs_T, | ||
int | cs_T | ||
) |
References bli_caxpyv(), bli_ccopyv(), bli_cdot(), bli_cgemv(), bli_chemv(), bli_cher2(), bli_cscalv(), FLA_free(), FLA_Househ2_UT_l_opc(), FLA_malloc(), FLA_MINUS_ONE, FLA_ONE, FLA_TWO, and FLA_ZERO.
Referenced by FLA_Tridiag_UT_l_step_opt_var2().
{ scomplex* buff_2 = FLA_COMPLEX_PTR( FLA_TWO ); scomplex* buff_1 = FLA_COMPLEX_PTR( FLA_ONE ); scomplex* buff_0 = FLA_COMPLEX_PTR( FLA_ZERO ); scomplex* buff_m1 = FLA_COMPLEX_PTR( FLA_MINUS_ONE ); scomplex first_elem; scomplex beta; scomplex inv_tau11; scomplex minus_inv_tau11; scomplex minus_upsilon11, minus_conj_upsilon11; scomplex minus_zeta11, minus_conj_zeta11; int i; // b_alg = FLA_Obj_length( T ); int b_alg = m_T; // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &u ); // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &z ); // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &w ); scomplex* buff_u = ( scomplex* ) FLA_malloc( m_A * sizeof( *buff_A ) ); scomplex* buff_z = ( scomplex* ) FLA_malloc( m_A * sizeof( *buff_A ) ); scomplex* buff_w = ( scomplex* ) FLA_malloc( m_A * sizeof( *buff_A ) ); int inc_u = 1; int inc_z = 1; int inc_w = 1; // Initialize some variables (only to prevent compiler warnings). first_elem = *buff_0; minus_inv_tau11 = *buff_0; for ( i = 0; i < b_alg; ++i ) { scomplex* A20 = buff_A + (0 )*cs_A + (i+1)*rs_A; scomplex* alpha11 = buff_A + (i )*cs_A + (i )*rs_A; scomplex* a21 = buff_A + (i )*cs_A + (i+1)*rs_A; scomplex* A22 = buff_A + (i+1)*cs_A + (i+1)*rs_A; scomplex* t01 = buff_T + (i )*cs_T + (0 )*rs_T; scomplex* tau11 = buff_T + (i )*cs_T + (i )*rs_T; scomplex* upsilon11= buff_u + (i )*inc_u; scomplex* u21 = buff_u + (i+1)*inc_u; scomplex* zeta11 = buff_z + (i )*inc_z; scomplex* z21 = buff_z + (i+1)*inc_z; scomplex* w21 = buff_w + (i+1)*inc_w; scomplex* a21_t = a21 + (0 )*cs_A + (0 )*rs_A; scomplex* a21_b = a21 + (0 )*cs_A + (1 )*rs_A; int m_ahead = m_A - i - 1; int m_behind = i; int n_behind = i; /*------------------------------------------------------------*/ if ( m_behind > 0 ) { // FLA_Copy( upsilon11, minus_upsilon11 ); // FLA_Scal( FLA_MINUS_ONE, minus_upsilon11 ); // FLA_Copy( minus_upsilon11, minus_conj_upsilon11 ); bli_cmult3( buff_m1, upsilon11, &minus_upsilon11 ); bli_ccopyconj( &minus_upsilon11, &minus_conj_upsilon11 ); // FLA_Copy( zeta11, minus_zeta11 ); // FLA_Scal( FLA_MINUS_ONE, minus_zeta11 ); // FLA_Copy( minus_zeta11, minus_conj_zeta11 ); bli_cmult3( buff_m1, zeta11, &minus_zeta11 ); bli_ccopyconj( &minus_zeta11, &minus_conj_zeta11 ); // FLA_Axpyt( FLA_CONJ_NO_TRANSPOSE, minus_upsilon11, zeta11, alpha11 ); // FLA_Axpyt( FLA_CONJ_NO_TRANSPOSE, minus_zeta11, upsilon11, alpha11 ); bli_caxpyv( BLIS_CONJUGATE, 1, &minus_upsilon11, zeta11, 1, alpha11, 1 ); bli_caxpyv( BLIS_CONJUGATE, 1, &minus_zeta11, upsilon11, 1, alpha11, 1 ); // FLA_Axpyt( FLA_NO_TRANSPOSE, minus_conj_zeta11, u21, a21 ); // FLA_Axpyt( FLA_NO_TRANSPOSE, minus_conj_upsilon11, z21, a21 ); bli_caxpyv( BLIS_NO_CONJUGATE, m_ahead, &minus_conj_zeta11, u21, inc_u, a21, rs_A ); bli_caxpyv( BLIS_NO_CONJUGATE, m_ahead, &minus_conj_upsilon11, z21, inc_z, a21, rs_A ); } if ( m_ahead > 0 ) { // FLA_Househ2_UT( FLA_LEFT, // a21_t, // a21_b, tau11 ); FLA_Househ2_UT_l_opc( m_ahead - 1, a21_t, a21_b, rs_A, tau11 ); // FLA_Obj_set_to_scalar( FLA_ONE, inv_tau11 ); // FLA_Inv_scalc( FLA_NO_CONJUGATE, tau11, inv_tau11 ); // FLA_Copy( inv_tau11, minus_inv_tau11 ); // FLA_Scal( FLA_MINUS_ONE, minus_inv_tau11 ); bli_cdiv3( buff_1, tau11, &inv_tau11 ); bli_cneg2( &inv_tau11, &minus_inv_tau11 ); // FLA_Copy( a21_t, first_elem ); // FLA_Obj_set_to_scalar( FLA_ONE, a21_t ); first_elem = *a21_t; *a21_t = *buff_1; } if ( m_behind > 0 ) { // FLA_Her2( FLA_LOWER_TRIANGULAR, FLA_MINUS_ONE, u21, z21, A22 ); bli_cher2( BLIS_LOWER_TRIANGULAR, BLIS_NO_TRANSPOSE, m_ahead, buff_m1, u21, inc_u, z21, inc_z, A22, rs_A, cs_A ); } if ( m_ahead > 0 ) { // FLA_Hemv( FLA_LOWER_TRIANGULAR, FLA_ONE, A22, a21, FLA_ZERO, w21 ); bli_chemv( BLIS_LOWER_TRIANGULAR, BLIS_NO_CONJUGATE, m_ahead, buff_1, A22, rs_A, cs_A, a21, rs_A, buff_0, w21, inc_w ); // FLA_Copy( a21, u21 ); // FLA_Copy( w21, z21 ); bli_ccopyv( BLIS_NO_CONJUGATE, m_ahead, a21, rs_A, u21, inc_u ); bli_ccopyv( BLIS_NO_CONJUGATE, m_ahead, w21, inc_w, z21, inc_z ); // FLA_Dotc( FLA_CONJUGATE, a21, z21, beta ); // FLA_Inv_scal( FLA_TWO, beta ); bli_cdot( BLIS_CONJUGATE, m_ahead, a21, rs_A, z21, inc_z, &beta ); bli_cinvscals( buff_2, &beta ); // FLA_Scal( minus_inv_tau11, beta ); // FLA_Axpy( beta, a21, z21 ); // FLA_Scal( inv_tau11, z21 ); bli_cscals( &minus_inv_tau11, &beta ); bli_caxpyv( BLIS_NO_CONJUGATE, m_ahead, &beta, a21, rs_A, z21, inc_z ); bli_cscalv( BLIS_NO_CONJUGATE, m_ahead, &inv_tau11, z21, inc_z ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, t01 ); bli_cgemv( BLIS_CONJ_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_1, A20, rs_A, cs_A, a21, rs_A, buff_0, t01, rs_T ); // FLA_Copy( first_elem, a21_t ); *a21_t = first_elem; } if ( m_behind + 1 == b_alg && m_ahead > 0 ) { // FLA_Her2( FLA_LOWER_TRIANGULAR, FLA_MINUS_ONE, u21, z21, A22 ); bli_cher2( BLIS_LOWER_TRIANGULAR, BLIS_NO_TRANSPOSE, m_ahead, buff_m1, u21, inc_u, z21, inc_z, A22, rs_A, cs_A ); } /*------------------------------------------------------------*/ } // FLA_Obj_free( &u ); // FLA_Obj_free( &z ); // FLA_Obj_free( &w ); FLA_free( buff_u ); FLA_free( buff_z ); FLA_free( buff_w ); return FLA_SUCCESS; }
FLA_Error FLA_Tridiag_UT_l_step_opc_var3 | ( | int | m_A, |
int | m_T, | ||
scomplex * | buff_A, | ||
int | rs_A, | ||
int | cs_A, | ||
scomplex * | buff_Z, | ||
int | rs_Z, | ||
int | cs_Z, | ||
scomplex * | buff_T, | ||
int | rs_T, | ||
int | cs_T | ||
) |
References bli_caxpyv(), bli_cdot(), bli_cgemv(), bli_chemv(), bli_cscalv(), bli_csetm(), FLA_free(), FLA_Househ2_UT_l_opc(), FLA_malloc(), FLA_MINUS_ONE, FLA_ONE, FLA_TWO, and FLA_ZERO.
Referenced by FLA_Tridiag_UT_l_step_opt_var3().
{ scomplex* buff_2 = FLA_COMPLEX_PTR( FLA_TWO ); scomplex* buff_1 = FLA_COMPLEX_PTR( FLA_ONE ); scomplex* buff_0 = FLA_COMPLEX_PTR( FLA_ZERO ); scomplex* buff_m1 = FLA_COMPLEX_PTR( FLA_MINUS_ONE ); scomplex first_elem, last_elem; scomplex beta; scomplex inv_tau11; scomplex minus_inv_tau11; int i; // b_alg = FLA_Obj_length( T ); int b_alg = m_T; // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &d ); // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &f ); scomplex* buff_d = ( scomplex* ) FLA_malloc( m_A * sizeof( *buff_A ) ); scomplex* buff_f = ( scomplex* ) FLA_malloc( m_A * sizeof( *buff_A ) ); int inc_d = 1; int inc_f = 1; // FLA_Obj_set_to_scalar( FLA_ZERO, Z ); bli_csetm( m_A, b_alg, buff_0, buff_Z, rs_Z, cs_Z ); for ( i = 0; i < b_alg; ++i ) { scomplex* a10t = buff_A + (0 )*cs_A + (i )*rs_A; scomplex* A20 = buff_A + (0 )*cs_A + (i+1)*rs_A; scomplex* alpha11 = buff_A + (i )*cs_A + (i )*rs_A; scomplex* a21 = buff_A + (i )*cs_A + (i+1)*rs_A; scomplex* A22 = buff_A + (i+1)*cs_A + (i+1)*rs_A; scomplex* z10t = buff_Z + (0 )*cs_Z + (i )*rs_Z; scomplex* Z20 = buff_Z + (0 )*cs_Z + (i+1)*rs_Z; scomplex* z21 = buff_Z + (i )*cs_Z + (i+1)*rs_Z; scomplex* t01 = buff_T + (i )*cs_T + (0 )*rs_T; scomplex* tau11 = buff_T + (i )*cs_T + (i )*rs_T; scomplex* d01 = buff_d + (0 )*inc_d; scomplex* f01 = buff_f + (0 )*inc_f; scomplex* a10t_r = a10t + (i-1)*cs_A + (0 )*rs_A; scomplex* a21_t = a21 + (0 )*cs_A + (0 )*rs_A; scomplex* a21_b = a21 + (0 )*cs_A + (1 )*rs_A; scomplex* ABL = a10t; scomplex* ZBL = z10t; scomplex* a2 = alpha11; int m_ahead = m_A - i - 1; int m_behind = i; int n_behind = i; /*------------------------------------------------------------*/ if ( m_behind > 0 ) { // FLA_Copy( a10t_r, last_elem ); // FLA_Obj_set_to_scalar( FLA_ONE, a10t_r ); last_elem = *a10t_r; *a10t_r = *buff_1; } // FLA_Gemvc( FLA_NO_TRANSPOSE, FLA_CONJUGATE, FLA_MINUS_ONE, ABL, z10t, FLA_ONE, a2 ); // FLA_Gemvc( FLA_NO_TRANSPOSE, FLA_CONJUGATE, FLA_MINUS_ONE, ZBL, a10t, FLA_ONE, a2 ); bli_cgemv( BLIS_NO_TRANSPOSE, BLIS_CONJUGATE, m_ahead + 1, n_behind, buff_m1, ABL, rs_A, cs_A, z10t, cs_Z, buff_1, a2, rs_A ); bli_cgemv( BLIS_NO_TRANSPOSE, BLIS_CONJUGATE, m_ahead + 1, n_behind, buff_m1, ZBL, rs_Z, cs_Z, a10t, cs_A, buff_1, a2, rs_A ); if ( m_behind > 0 ) { // FLA_Copy( last_elem, a10t_r ); *a10t_r = last_elem; } if ( m_ahead > 0 ) { // FLA_Househ2_UT( FLA_LEFT, // a21_t, // a21_b, tau11 ); FLA_Househ2_UT_l_opc( m_ahead - 1, a21_t, a21_b, rs_A, tau11 ); // FLA_Obj_set_to_scalar( FLA_ONE, inv_tau11 ); // FLA_Inv_scalc( FLA_NO_CONJUGATE, tau11, inv_tau11 ); // FLA_Copy( inv_tau11, minus_inv_tau11 ); // FLA_Scal( FLA_MINUS_ONE, minus_inv_tau11 ); bli_cdiv3( buff_1, tau11, &inv_tau11 ); bli_cneg2( &inv_tau11, &minus_inv_tau11 ); // FLA_Copy( a21_t, first_elem ); // FLA_Obj_set_to_scalar( FLA_ONE, a21_t ); first_elem = *a21_t; *a21_t = *buff_1; // FLA_Hemv( FLA_LOWER_TRIANGULAR, FLA_ONE, A22, a21, FLA_ZERO, z21 ); bli_chemv( BLIS_LOWER_TRIANGULAR, BLIS_NO_CONJUGATE, m_ahead, buff_1, A22, rs_A, cs_A, a21, rs_A, buff_0, z21, rs_Z ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, d01 ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, Z20, a21, FLA_ZERO, f01 ); bli_cgemv( BLIS_CONJ_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_1, A20, rs_A, cs_A, a21, rs_A, buff_0, d01, inc_d ); bli_cgemv( BLIS_CONJ_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_1, Z20, rs_Z, cs_Z, a21, rs_A, buff_0, f01, inc_f ); // FLA_Gemv( FLA_NO_TRANSPOSE, FLA_MINUS_ONE, A20, f01, FLA_ONE, z21 ); // FLA_Gemv( FLA_NO_TRANSPOSE, FLA_MINUS_ONE, Z20, d01, FLA_ONE, z21 ); bli_cgemv( BLIS_NO_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_m1, A20, rs_A, cs_A, f01, inc_f, buff_1, z21, rs_Z ); bli_cgemv( BLIS_NO_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_m1, Z20, rs_Z, cs_Z, d01, inc_d, buff_1, z21, rs_Z ); // FLA_Dotc( FLA_CONJUGATE, a21, z21, beta ); // FLA_Inv_scal( FLA_TWO, beta ); bli_cdot( BLIS_CONJUGATE, m_ahead, a21, rs_A, z21, rs_Z, &beta ); bli_cinvscals( buff_2, &beta ); // FLA_Scal( minus_inv_tau11, beta ); // FLA_Axpy( beta, a21, z21 ); // FLA_Scal( inv_tau11, z21 ); bli_cscals( &minus_inv_tau11, &beta ); bli_caxpyv( BLIS_NO_CONJUGATE, m_ahead, &beta, a21, rs_A, z21, rs_Z ); bli_cscalv( BLIS_NO_CONJUGATE, m_ahead, &inv_tau11, z21, rs_Z ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, t01 ); bli_cgemv( BLIS_CONJ_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_1, A20, rs_A, cs_A, a21, rs_A, buff_0, t01, rs_T ); // FLA_Copy( first_elem, a21_t ); *a21_t = first_elem; } /*------------------------------------------------------------*/ } // FLA_Obj_free( &d ); // FLA_Obj_free( &f ); FLA_free( buff_d ); FLA_free( buff_f ); return FLA_SUCCESS; }
FLA_Error FLA_Tridiag_UT_l_step_opd_var1 | ( | int | m_A, |
int | m_T, | ||
double * | buff_A, | ||
int | rs_A, | ||
int | cs_A, | ||
double * | buff_T, | ||
int | rs_T, | ||
int | cs_T | ||
) |
References bli_daxpyv(), bli_ddot(), bli_dgemv(), bli_dscalv(), bli_dsymv(), bli_dsyr2(), FLA_free(), FLA_Househ2_UT_l_opd(), FLA_malloc(), FLA_MINUS_ONE, FLA_ONE, FLA_TWO, and FLA_ZERO.
Referenced by FLA_Tridiag_UT_l_step_opt_var1().
{ double* buff_2 = FLA_DOUBLE_PTR( FLA_TWO ); double* buff_1 = FLA_DOUBLE_PTR( FLA_ONE ); double* buff_0 = FLA_DOUBLE_PTR( FLA_ZERO ); double* buff_m1 = FLA_DOUBLE_PTR( FLA_MINUS_ONE ); double first_elem; double beta; double inv_tau11; double minus_inv_tau11; int i; // b_alg = FLA_Obj_length( T ); int b_alg = m_T; // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &z ); double* buff_z = ( double* ) FLA_malloc( m_A * sizeof( *buff_A ) ); int inc_z = 1; for ( i = 0; i < b_alg; ++i ) { double* A20 = buff_A + (0 )*cs_A + (i+1)*rs_A; double* a21 = buff_A + (i )*cs_A + (i+1)*rs_A; double* A22 = buff_A + (i+1)*cs_A + (i+1)*rs_A; double* t01 = buff_T + (i )*cs_T + (0 )*rs_T; double* tau11 = buff_T + (i )*cs_T + (i )*rs_T; double* z21 = buff_z + (i+1)*inc_z; double* a21_t = a21 + (0 )*cs_A + (0 )*rs_A; double* a21_b = a21 + (0 )*cs_A + (1 )*rs_A; int m_ahead = m_A - i - 1; int n_behind = i; /*------------------------------------------------------------*/ if ( m_ahead > 0 ) { // FLA_Househ2_UT( FLA_LEFT, // a21_t, // a21_b, tau11 ); FLA_Househ2_UT_l_opd( m_ahead - 1, a21_t, a21_b, rs_A, tau11 ); // FLA_Obj_set_to_scalar( FLA_ONE, inv_tau11 ); // FLA_Inv_scalc( FLA_NO_CONJUGATE, tau11, inv_tau11 ); // FLA_Copy( inv_tau11, minus_inv_tau11 ); // FLA_Scal( FLA_MINUS_ONE, minus_inv_tau11 ); bli_ddiv3( buff_1, tau11, &inv_tau11 ); bli_dneg2( &inv_tau11, &minus_inv_tau11 ); // FLA_Copy( a21_t, first_elem ); // FLA_Obj_set_to_scalar( FLA_ONE, a21_t ); first_elem = *a21_t; *a21_t = *buff_1; // FLA_Hemv( FLA_LOWER_TRIANGULAR, FLA_ONE, A22, a21, FLA_ZERO, z21 ); bli_dsymv( BLIS_LOWER_TRIANGULAR, m_ahead, buff_1, A22, rs_A, cs_A, a21, rs_A, buff_0, z21, inc_z ); // FLA_Dotc( FLA_CONJUGATE, a21, z21, beta ); // FLA_Inv_scal( FLA_TWO, beta ); bli_ddot( BLIS_CONJUGATE, m_ahead, a21, rs_A, z21, inc_z, &beta ); bli_dinvscals( buff_2, &beta ); // FLA_Scal( minus_inv_tau11, beta ); // FLA_Axpy( beta, a21, z21 ); // FLA_Scal( inv_tau11, z21 ); bli_dscals( &minus_inv_tau11, &beta ); bli_daxpyv( BLIS_NO_CONJUGATE, m_ahead, &beta, a21, rs_A, z21, inc_z ); bli_dscalv( BLIS_NO_CONJUGATE, m_ahead, &inv_tau11, z21, inc_z ); // FLA_Her2( FLA_LOWER_TRIANGULAR, FLA_MINUS_ONE, a21, z21, A22 ); bli_dsyr2( BLIS_LOWER_TRIANGULAR, m_ahead, buff_m1, a21, rs_A, z21, inc_z, A22, rs_A, cs_A ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, t01 ); bli_dgemv( BLIS_CONJ_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_1, A20, rs_A, cs_A, a21, rs_A, buff_0, t01, rs_T ); // FLA_Copy( first_elem, a21_t ); *a21_t = first_elem; } /*------------------------------------------------------------*/ } // FLA_Obj_free( &z ); FLA_free( buff_z ); return FLA_SUCCESS; }
FLA_Error FLA_Tridiag_UT_l_step_opd_var2 | ( | int | m_A, |
int | m_T, | ||
double * | buff_A, | ||
int | rs_A, | ||
int | cs_A, | ||
double * | buff_T, | ||
int | rs_T, | ||
int | cs_T | ||
) |
References bli_daxpyv(), bli_dcopyv(), bli_ddot(), bli_dgemv(), bli_dscalv(), bli_dsymv(), bli_dsyr2(), FLA_free(), FLA_Househ2_UT_l_opd(), FLA_malloc(), FLA_MINUS_ONE, FLA_ONE, FLA_TWO, and FLA_ZERO.
Referenced by FLA_Tridiag_UT_l_step_opt_var2().
{ double* buff_2 = FLA_DOUBLE_PTR( FLA_TWO ); double* buff_1 = FLA_DOUBLE_PTR( FLA_ONE ); double* buff_0 = FLA_DOUBLE_PTR( FLA_ZERO ); double* buff_m1 = FLA_DOUBLE_PTR( FLA_MINUS_ONE ); double first_elem; double beta; double inv_tau11; double minus_inv_tau11; double minus_upsilon11, minus_conj_upsilon11; double minus_zeta11, minus_conj_zeta11; int i; // b_alg = FLA_Obj_length( T ); int b_alg = m_T; // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &u ); // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &z ); // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &w ); double* buff_u = ( double* ) FLA_malloc( m_A * sizeof( *buff_A ) ); double* buff_z = ( double* ) FLA_malloc( m_A * sizeof( *buff_A ) ); double* buff_w = ( double* ) FLA_malloc( m_A * sizeof( *buff_A ) ); int inc_u = 1; int inc_z = 1; int inc_w = 1; // Initialize some variables (only to prevent compiler warnings). first_elem = *buff_0; minus_inv_tau11 = *buff_0; for ( i = 0; i < b_alg; ++i ) { double* A20 = buff_A + (0 )*cs_A + (i+1)*rs_A; double* alpha11 = buff_A + (i )*cs_A + (i )*rs_A; double* a21 = buff_A + (i )*cs_A + (i+1)*rs_A; double* A22 = buff_A + (i+1)*cs_A + (i+1)*rs_A; double* t01 = buff_T + (i )*cs_T + (0 )*rs_T; double* tau11 = buff_T + (i )*cs_T + (i )*rs_T; double* upsilon11= buff_u + (i )*inc_u; double* u21 = buff_u + (i+1)*inc_u; double* zeta11 = buff_z + (i )*inc_z; double* z21 = buff_z + (i+1)*inc_z; double* w21 = buff_w + (i+1)*inc_w; double* a21_t = a21 + (0 )*cs_A + (0 )*rs_A; double* a21_b = a21 + (0 )*cs_A + (1 )*rs_A; int m_ahead = m_A - i - 1; int m_behind = i; int n_behind = i; /*------------------------------------------------------------*/ if ( m_behind > 0 ) { // FLA_Copy( upsilon11, minus_upsilon11 ); // FLA_Scal( FLA_MINUS_ONE, minus_upsilon11 ); // FLA_Copy( minus_upsilon11, minus_conj_upsilon11 ); bli_dmult3( buff_m1, upsilon11, &minus_upsilon11 ); bli_dcopyconj( &minus_upsilon11, &minus_conj_upsilon11 ); // FLA_Copy( zeta11, minus_zeta11 ); // FLA_Scal( FLA_MINUS_ONE, minus_zeta11 ); // FLA_Copy( minus_zeta11, minus_conj_zeta11 ); bli_dmult3( buff_m1, zeta11, &minus_zeta11 ); bli_dcopyconj( &minus_zeta11, &minus_conj_zeta11 ); // FLA_Axpyt( FLA_CONJ_NO_TRANSPOSE, minus_upsilon11, zeta11, alpha11 ); // FLA_Axpyt( FLA_CONJ_NO_TRANSPOSE, minus_zeta11, upsilon11, alpha11 ); bli_daxpyv( BLIS_CONJUGATE, 1, &minus_upsilon11, zeta11, 1, alpha11, 1 ); bli_daxpyv( BLIS_CONJUGATE, 1, &minus_zeta11, upsilon11, 1, alpha11, 1 ); // FLA_Axpyt( FLA_NO_TRANSPOSE, minus_conj_zeta11, u21, a21 ); // FLA_Axpyt( FLA_NO_TRANSPOSE, minus_conj_upsilon11, z21, a21 ); bli_daxpyv( BLIS_NO_CONJUGATE, m_ahead, &minus_conj_zeta11, u21, inc_u, a21, rs_A ); bli_daxpyv( BLIS_NO_CONJUGATE, m_ahead, &minus_conj_upsilon11, z21, inc_z, a21, rs_A ); } if ( m_ahead > 0 ) { // FLA_Househ2_UT( FLA_LEFT, // a21_t, // a21_b, tau11 ); FLA_Househ2_UT_l_opd( m_ahead - 1, a21_t, a21_b, rs_A, tau11 ); // FLA_Obj_set_to_scalar( FLA_ONE, inv_tau11 ); // FLA_Inv_scalc( FLA_NO_CONJUGATE, tau11, inv_tau11 ); // FLA_Copy( inv_tau11, minus_inv_tau11 ); // FLA_Scal( FLA_MINUS_ONE, minus_inv_tau11 ); bli_ddiv3( buff_1, tau11, &inv_tau11 ); bli_dneg2( &inv_tau11, &minus_inv_tau11 ); // FLA_Copy( a21_t, first_elem ); // FLA_Obj_set_to_scalar( FLA_ONE, a21_t ); first_elem = *a21_t; *a21_t = *buff_1; } if ( m_behind > 0 ) { // FLA_Her2( FLA_LOWER_TRIANGULAR, FLA_MINUS_ONE, u21, z21, A22 ); bli_dsyr2( BLIS_LOWER_TRIANGULAR, m_ahead, buff_m1, u21, inc_u, z21, inc_z, A22, rs_A, cs_A ); } if ( m_ahead > 0 ) { // FLA_Hemv( FLA_LOWER_TRIANGULAR, FLA_ONE, A22, a21, FLA_ZERO, w21 ); bli_dsymv( BLIS_LOWER_TRIANGULAR, m_ahead, buff_1, A22, rs_A, cs_A, a21, rs_A, buff_0, w21, inc_w ); // FLA_Copy( a21, u21 ); // FLA_Copy( w21, z21 ); bli_dcopyv( BLIS_NO_CONJUGATE, m_ahead, a21, rs_A, u21, inc_u ); bli_dcopyv( BLIS_NO_CONJUGATE, m_ahead, w21, inc_w, z21, inc_z ); // FLA_Dotc( FLA_CONJUGATE, a21, z21, beta ); // FLA_Inv_scal( FLA_TWO, beta ); bli_ddot( BLIS_CONJUGATE, m_ahead, a21, rs_A, z21, inc_z, &beta ); bli_dinvscals( buff_2, &beta ); // FLA_Scal( minus_inv_tau11, beta ); // FLA_Axpy( beta, a21, z21 ); // FLA_Scal( inv_tau11, z21 ); bli_dscals( &minus_inv_tau11, &beta ); bli_daxpyv( BLIS_NO_CONJUGATE, m_ahead, &beta, a21, rs_A, z21, inc_z ); bli_dscalv( BLIS_NO_CONJUGATE, m_ahead, &inv_tau11, z21, inc_z ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, t01 ); bli_dgemv( BLIS_CONJ_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_1, A20, rs_A, cs_A, a21, rs_A, buff_0, t01, rs_T ); // FLA_Copy( first_elem, a21_t ); *a21_t = first_elem; } if ( m_behind + 1 == b_alg && m_ahead > 0 ) { // FLA_Her2( FLA_LOWER_TRIANGULAR, FLA_MINUS_ONE, u21, z21, A22 ); bli_dsyr2( BLIS_LOWER_TRIANGULAR, m_ahead, buff_m1, u21, inc_u, z21, inc_z, A22, rs_A, cs_A ); } /*------------------------------------------------------------*/ } // FLA_Obj_free( &u ); // FLA_Obj_free( &z ); // FLA_Obj_free( &w ); FLA_free( buff_u ); FLA_free( buff_z ); FLA_free( buff_w ); return FLA_SUCCESS; }
FLA_Error FLA_Tridiag_UT_l_step_opd_var3 | ( | int | m_A, |
int | m_T, | ||
double * | buff_A, | ||
int | rs_A, | ||
int | cs_A, | ||
double * | buff_Z, | ||
int | rs_Z, | ||
int | cs_Z, | ||
double * | buff_T, | ||
int | rs_T, | ||
int | cs_T | ||
) |
References bli_daxpyv(), bli_ddot(), bli_dgemv(), bli_dscalv(), bli_dsetm(), bli_dsymv(), FLA_free(), FLA_Househ2_UT_l_opd(), FLA_malloc(), FLA_MINUS_ONE, FLA_ONE, FLA_TWO, and FLA_ZERO.
Referenced by FLA_Tridiag_UT_l_step_opt_var3().
{ double* buff_2 = FLA_DOUBLE_PTR( FLA_TWO ); double* buff_1 = FLA_DOUBLE_PTR( FLA_ONE ); double* buff_0 = FLA_DOUBLE_PTR( FLA_ZERO ); double* buff_m1 = FLA_DOUBLE_PTR( FLA_MINUS_ONE ); double first_elem, last_elem; double beta; double inv_tau11; double minus_inv_tau11; int i; // b_alg = FLA_Obj_length( T ); int b_alg = m_T; // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &d ); // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &f ); double* buff_d = ( double* ) FLA_malloc( m_A * sizeof( *buff_A ) ); double* buff_f = ( double* ) FLA_malloc( m_A * sizeof( *buff_A ) ); int inc_d = 1; int inc_f = 1; // FLA_Obj_set_to_scalar( FLA_ZERO, Z ); bli_dsetm( m_A, b_alg, buff_0, buff_Z, rs_Z, cs_Z ); for ( i = 0; i < b_alg; ++i ) { double* a10t = buff_A + (0 )*cs_A + (i )*rs_A; double* A20 = buff_A + (0 )*cs_A + (i+1)*rs_A; double* alpha11 = buff_A + (i )*cs_A + (i )*rs_A; double* a21 = buff_A + (i )*cs_A + (i+1)*rs_A; double* A22 = buff_A + (i+1)*cs_A + (i+1)*rs_A; double* z10t = buff_Z + (0 )*cs_Z + (i )*rs_Z; double* Z20 = buff_Z + (0 )*cs_Z + (i+1)*rs_Z; double* z21 = buff_Z + (i )*cs_Z + (i+1)*rs_Z; double* t01 = buff_T + (i )*cs_T + (0 )*rs_T; double* tau11 = buff_T + (i )*cs_T + (i )*rs_T; double* d01 = buff_d + (0 )*inc_d; double* f01 = buff_f + (0 )*inc_f; double* a10t_r = a10t + (i-1)*cs_A + (0 )*rs_A; double* a21_t = a21 + (0 )*cs_A + (0 )*rs_A; double* a21_b = a21 + (0 )*cs_A + (1 )*rs_A; double* ABL = a10t; double* ZBL = z10t; double* a2 = alpha11; int m_ahead = m_A - i - 1; int m_behind = i; int n_behind = i; /*------------------------------------------------------------*/ if ( m_behind > 0 ) { // FLA_Copy( a10t_r, last_elem ); // FLA_Obj_set_to_scalar( FLA_ONE, a10t_r ); last_elem = *a10t_r; *a10t_r = *buff_1; } // FLA_Gemvc( FLA_NO_TRANSPOSE, FLA_CONJUGATE, FLA_MINUS_ONE, ABL, z10t, FLA_ONE, a2 ); // FLA_Gemvc( FLA_NO_TRANSPOSE, FLA_CONJUGATE, FLA_MINUS_ONE, ZBL, a10t, FLA_ONE, a2 ); bli_dgemv( BLIS_NO_TRANSPOSE, BLIS_CONJUGATE, m_ahead + 1, n_behind, buff_m1, ABL, rs_A, cs_A, z10t, cs_Z, buff_1, a2, rs_A ); bli_dgemv( BLIS_NO_TRANSPOSE, BLIS_CONJUGATE, m_ahead + 1, n_behind, buff_m1, ZBL, rs_Z, cs_Z, a10t, cs_A, buff_1, a2, rs_A ); if ( m_behind > 0 ) { // FLA_Copy( last_elem, a10t_r ); *a10t_r = last_elem; } if ( m_ahead > 0 ) { // FLA_Househ2_UT( FLA_LEFT, // a21_t, // a21_b, tau11 ); FLA_Househ2_UT_l_opd( m_ahead - 1, a21_t, a21_b, rs_A, tau11 ); // FLA_Obj_set_to_scalar( FLA_ONE, inv_tau11 ); // FLA_Inv_scalc( FLA_NO_CONJUGATE, tau11, inv_tau11 ); // FLA_Copy( inv_tau11, minus_inv_tau11 ); // FLA_Scal( FLA_MINUS_ONE, minus_inv_tau11 ); bli_ddiv3( buff_1, tau11, &inv_tau11 ); bli_dneg2( &inv_tau11, &minus_inv_tau11 ); // FLA_Copy( a21_t, first_elem ); // FLA_Obj_set_to_scalar( FLA_ONE, a21_t ); first_elem = *a21_t; *a21_t = *buff_1; // FLA_Hemv( FLA_LOWER_TRIANGULAR, FLA_ONE, A22, a21, FLA_ZERO, z21 ); bli_dsymv( BLIS_LOWER_TRIANGULAR, m_ahead, buff_1, A22, rs_A, cs_A, a21, rs_A, buff_0, z21, rs_Z ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, d01 ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, Z20, a21, FLA_ZERO, f01 ); bli_dgemv( BLIS_CONJ_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_1, A20, rs_A, cs_A, a21, rs_A, buff_0, d01, inc_d ); bli_dgemv( BLIS_CONJ_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_1, Z20, rs_Z, cs_Z, a21, rs_A, buff_0, f01, inc_f ); // FLA_Gemv( FLA_NO_TRANSPOSE, FLA_MINUS_ONE, A20, f01, FLA_ONE, z21 ); // FLA_Gemv( FLA_NO_TRANSPOSE, FLA_MINUS_ONE, Z20, d01, FLA_ONE, z21 ); bli_dgemv( BLIS_NO_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_m1, A20, rs_A, cs_A, f01, inc_f, buff_1, z21, rs_Z ); bli_dgemv( BLIS_NO_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_m1, Z20, rs_Z, cs_Z, d01, inc_d, buff_1, z21, rs_Z ); // FLA_Dotc( FLA_CONJUGATE, a21, z21, beta ); // FLA_Inv_scal( FLA_TWO, beta ); bli_ddot( BLIS_CONJUGATE, m_ahead, a21, rs_A, z21, rs_Z, &beta ); bli_dinvscals( buff_2, &beta ); // FLA_Scal( minus_inv_tau11, beta ); // FLA_Axpy( beta, a21, z21 ); // FLA_Scal( inv_tau11, z21 ); bli_dscals( &minus_inv_tau11, &beta ); bli_daxpyv( BLIS_NO_CONJUGATE, m_ahead, &beta, a21, rs_A, z21, rs_Z ); bli_dscalv( BLIS_NO_CONJUGATE, m_ahead, &inv_tau11, z21, rs_Z ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, t01 ); bli_dgemv( BLIS_CONJ_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_1, A20, rs_A, cs_A, a21, rs_A, buff_0, t01, rs_T ); // FLA_Copy( first_elem, a21_t ); *a21_t = first_elem; } /*------------------------------------------------------------*/ } // FLA_Obj_free( &d ); // FLA_Obj_free( &f ); FLA_free( buff_d ); FLA_free( buff_f ); return FLA_SUCCESS; }
FLA_Error FLA_Tridiag_UT_l_step_ops_var1 | ( | int | m_A, |
int | m_T, | ||
float * | buff_A, | ||
int | rs_A, | ||
int | cs_A, | ||
float * | buff_T, | ||
int | rs_T, | ||
int | cs_T | ||
) |
References bli_saxpyv(), bli_sdot(), bli_sgemv(), bli_sscalv(), bli_ssymv(), bli_ssyr2(), FLA_free(), FLA_Househ2_UT_l_ops(), FLA_malloc(), FLA_MINUS_ONE, FLA_ONE, FLA_TWO, and FLA_ZERO.
Referenced by FLA_Tridiag_UT_l_step_opt_var1().
{ float* buff_2 = FLA_FLOAT_PTR( FLA_TWO ); float* buff_1 = FLA_FLOAT_PTR( FLA_ONE ); float* buff_0 = FLA_FLOAT_PTR( FLA_ZERO ); float* buff_m1 = FLA_FLOAT_PTR( FLA_MINUS_ONE ); float first_elem; float beta; float inv_tau11; float minus_inv_tau11; int i; // b_alg = FLA_Obj_length( T ); int b_alg = m_T; // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &z ); float* buff_z = ( float* ) FLA_malloc( m_A * sizeof( *buff_A ) ); int inc_z = 1; for ( i = 0; i < b_alg; ++i ) { float* A20 = buff_A + (0 )*cs_A + (i+1)*rs_A; float* a21 = buff_A + (i )*cs_A + (i+1)*rs_A; float* A22 = buff_A + (i+1)*cs_A + (i+1)*rs_A; float* t01 = buff_T + (i )*cs_T + (0 )*rs_T; float* tau11 = buff_T + (i )*cs_T + (i )*rs_T; float* z21 = buff_z + (i+1)*inc_z; float* a21_t = a21 + (0 )*cs_A + (0 )*rs_A; float* a21_b = a21 + (0 )*cs_A + (1 )*rs_A; int m_ahead = m_A - i - 1; int n_behind = i; /*------------------------------------------------------------*/ if ( m_ahead > 0 ) { // FLA_Househ2_UT( FLA_LEFT, // a21_t, // a21_b, tau11 ); FLA_Househ2_UT_l_ops( m_ahead - 1, a21_t, a21_b, rs_A, tau11 ); // FLA_Obj_set_to_scalar( FLA_ONE, inv_tau11 ); // FLA_Inv_scalc( FLA_NO_CONJUGATE, tau11, inv_tau11 ); // FLA_Copy( inv_tau11, minus_inv_tau11 ); // FLA_Scal( FLA_MINUS_ONE, minus_inv_tau11 ); bli_sdiv3( buff_1, tau11, &inv_tau11 ); bli_sneg2( &inv_tau11, &minus_inv_tau11 ); // FLA_Copy( a21_t, first_elem ); // FLA_Obj_set_to_scalar( FLA_ONE, a21_t ); first_elem = *a21_t; *a21_t = *buff_1; // FLA_Hemv( FLA_LOWER_TRIANGULAR, FLA_ONE, A22, a21, FLA_ZERO, z21 ); bli_ssymv( BLIS_LOWER_TRIANGULAR, m_ahead, buff_1, A22, rs_A, cs_A, a21, rs_A, buff_0, z21, inc_z ); // FLA_Dotc( FLA_CONJUGATE, a21, z21, beta ); // FLA_Inv_scal( FLA_TWO, beta ); bli_sdot( BLIS_CONJUGATE, m_ahead, a21, rs_A, z21, inc_z, &beta ); bli_sinvscals( buff_2, &beta ); // FLA_Scal( minus_inv_tau11, beta ); // FLA_Axpy( beta, a21, z21 ); // FLA_Scal( inv_tau11, z21 ); bli_sscals( &minus_inv_tau11, &beta ); bli_saxpyv( BLIS_NO_CONJUGATE, m_ahead, &beta, a21, rs_A, z21, inc_z ); bli_sscalv( BLIS_NO_CONJUGATE, m_ahead, &inv_tau11, z21, inc_z ); // FLA_Her2( FLA_LOWER_TRIANGULAR, FLA_MINUS_ONE, a21, z21, A22 ); bli_ssyr2( BLIS_LOWER_TRIANGULAR, m_ahead, buff_m1, a21, rs_A, z21, inc_z, A22, rs_A, cs_A ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, t01 ); bli_sgemv( BLIS_CONJ_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_1, A20, rs_A, cs_A, a21, rs_A, buff_0, t01, rs_T ); // FLA_Copy( first_elem, a21_t ); *a21_t = first_elem; } /*------------------------------------------------------------*/ } // FLA_Obj_free( &z ); FLA_free( buff_z ); return FLA_SUCCESS; }
FLA_Error FLA_Tridiag_UT_l_step_ops_var2 | ( | int | m_A, |
int | m_T, | ||
float * | buff_A, | ||
int | rs_A, | ||
int | cs_A, | ||
float * | buff_T, | ||
int | rs_T, | ||
int | cs_T | ||
) |
References bli_saxpyv(), bli_scopyv(), bli_sdot(), bli_sgemv(), bli_sscalv(), bli_ssymv(), bli_ssyr2(), FLA_free(), FLA_Househ2_UT_l_ops(), FLA_malloc(), FLA_MINUS_ONE, FLA_ONE, FLA_TWO, and FLA_ZERO.
Referenced by FLA_Tridiag_UT_l_step_opt_var2().
{ float* buff_2 = FLA_FLOAT_PTR( FLA_TWO ); float* buff_1 = FLA_FLOAT_PTR( FLA_ONE ); float* buff_0 = FLA_FLOAT_PTR( FLA_ZERO ); float* buff_m1 = FLA_FLOAT_PTR( FLA_MINUS_ONE ); float first_elem; float beta; float inv_tau11; float minus_inv_tau11; float minus_upsilon11, minus_conj_upsilon11; float minus_zeta11, minus_conj_zeta11; int i; // b_alg = FLA_Obj_length( T ); int b_alg = m_T; // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &u ); // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &z ); // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &w ); float* buff_u = ( float* ) FLA_malloc( m_A * sizeof( *buff_A ) ); float* buff_z = ( float* ) FLA_malloc( m_A * sizeof( *buff_A ) ); float* buff_w = ( float* ) FLA_malloc( m_A * sizeof( *buff_A ) ); int inc_u = 1; int inc_z = 1; int inc_w = 1; // Initialize some variables (only to prevent compiler warnings). first_elem = *buff_0; minus_inv_tau11 = *buff_0; for ( i = 0; i < b_alg; ++i ) { float* A20 = buff_A + (0 )*cs_A + (i+1)*rs_A; float* alpha11 = buff_A + (i )*cs_A + (i )*rs_A; float* a21 = buff_A + (i )*cs_A + (i+1)*rs_A; float* A22 = buff_A + (i+1)*cs_A + (i+1)*rs_A; float* t01 = buff_T + (i )*cs_T + (0 )*rs_T; float* tau11 = buff_T + (i )*cs_T + (i )*rs_T; float* upsilon11= buff_u + (i )*inc_u; float* u21 = buff_u + (i+1)*inc_u; float* zeta11 = buff_z + (i )*inc_z; float* z21 = buff_z + (i+1)*inc_z; float* w21 = buff_w + (i+1)*inc_w; float* a21_t = a21 + (0 )*cs_A + (0 )*rs_A; float* a21_b = a21 + (0 )*cs_A + (1 )*rs_A; int m_ahead = m_A - i - 1; int m_behind = i; int n_behind = i; /*------------------------------------------------------------*/ if ( m_behind > 0 ) { // FLA_Copy( upsilon11, minus_upsilon11 ); // FLA_Scal( FLA_MINUS_ONE, minus_upsilon11 ); // FLA_Copy( minus_upsilon11, minus_conj_upsilon11 ); bli_smult3( buff_m1, upsilon11, &minus_upsilon11 ); bli_scopyconj( &minus_upsilon11, &minus_conj_upsilon11 ); // FLA_Copy( zeta11, minus_zeta11 ); // FLA_Scal( FLA_MINUS_ONE, minus_zeta11 ); // FLA_Copy( minus_zeta11, minus_conj_zeta11 ); bli_smult3( buff_m1, zeta11, &minus_zeta11 ); bli_scopyconj( &minus_zeta11, &minus_conj_zeta11 ); // FLA_Axpyt( FLA_CONJ_NO_TRANSPOSE, minus_upsilon11, zeta11, alpha11 ); // FLA_Axpyt( FLA_CONJ_NO_TRANSPOSE, minus_zeta11, upsilon11, alpha11 ); bli_saxpyv( BLIS_CONJUGATE, 1, &minus_upsilon11, zeta11, 1, alpha11, 1 ); bli_saxpyv( BLIS_CONJUGATE, 1, &minus_zeta11, upsilon11, 1, alpha11, 1 ); // FLA_Axpyt( FLA_NO_TRANSPOSE, minus_conj_zeta11, u21, a21 ); // FLA_Axpyt( FLA_NO_TRANSPOSE, minus_conj_upsilon11, z21, a21 ); bli_saxpyv( BLIS_NO_CONJUGATE, m_ahead, &minus_conj_zeta11, u21, inc_u, a21, rs_A ); bli_saxpyv( BLIS_NO_CONJUGATE, m_ahead, &minus_conj_upsilon11, z21, inc_z, a21, rs_A ); } if ( m_ahead > 0 ) { // FLA_Househ2_UT( FLA_LEFT, // a21_t, // a21_b, tau11 ); FLA_Househ2_UT_l_ops( m_ahead - 1, a21_t, a21_b, rs_A, tau11 ); // FLA_Obj_set_to_scalar( FLA_ONE, inv_tau11 ); // FLA_Inv_scalc( FLA_NO_CONJUGATE, tau11, inv_tau11 ); // FLA_Copy( inv_tau11, minus_inv_tau11 ); // FLA_Scal( FLA_MINUS_ONE, minus_inv_tau11 ); bli_sdiv3( buff_1, tau11, &inv_tau11 ); bli_sneg2( &inv_tau11, &minus_inv_tau11 ); // FLA_Copy( a21_t, first_elem ); // FLA_Obj_set_to_scalar( FLA_ONE, a21_t ); first_elem = *a21_t; *a21_t = *buff_1; } if ( m_behind > 0 ) { // FLA_Her2( FLA_LOWER_TRIANGULAR, FLA_MINUS_ONE, u21, z21, A22 ); bli_ssyr2( BLIS_LOWER_TRIANGULAR, m_ahead, buff_m1, u21, inc_u, z21, inc_z, A22, rs_A, cs_A ); } if ( m_ahead > 0 ) { // FLA_Hemv( FLA_LOWER_TRIANGULAR, FLA_ONE, A22, a21, FLA_ZERO, w21 ); bli_ssymv( BLIS_LOWER_TRIANGULAR, m_ahead, buff_1, A22, rs_A, cs_A, a21, rs_A, buff_0, w21, inc_w ); // FLA_Copy( a21, u21 ); // FLA_Copy( w21, z21 ); bli_scopyv( BLIS_NO_CONJUGATE, m_ahead, a21, rs_A, u21, inc_u ); bli_scopyv( BLIS_NO_CONJUGATE, m_ahead, w21, inc_w, z21, inc_z ); // FLA_Dotc( FLA_CONJUGATE, a21, z21, beta ); // FLA_Inv_scal( FLA_TWO, beta ); bli_sdot( BLIS_CONJUGATE, m_ahead, a21, rs_A, z21, inc_z, &beta ); bli_sinvscals( buff_2, &beta ); // FLA_Scal( minus_inv_tau11, beta ); // FLA_Axpy( beta, a21, z21 ); // FLA_Scal( inv_tau11, z21 ); bli_sscals( &minus_inv_tau11, &beta ); bli_saxpyv( BLIS_NO_CONJUGATE, m_ahead, &beta, a21, rs_A, z21, inc_z ); bli_sscalv( BLIS_NO_CONJUGATE, m_ahead, &inv_tau11, z21, inc_z ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, t01 ); bli_sgemv( BLIS_CONJ_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_1, A20, rs_A, cs_A, a21, rs_A, buff_0, t01, rs_T ); // FLA_Copy( first_elem, a21_t ); *a21_t = first_elem; } if ( m_behind + 1 == b_alg && m_ahead > 0 ) { // FLA_Her2( FLA_LOWER_TRIANGULAR, FLA_MINUS_ONE, u21, z21, A22 ); bli_ssyr2( BLIS_LOWER_TRIANGULAR, m_ahead, buff_m1, u21, inc_u, z21, inc_z, A22, rs_A, cs_A ); } /*------------------------------------------------------------*/ } // FLA_Obj_free( &u ); // FLA_Obj_free( &z ); // FLA_Obj_free( &w ); FLA_free( buff_u ); FLA_free( buff_z ); FLA_free( buff_w ); return FLA_SUCCESS; }
FLA_Error FLA_Tridiag_UT_l_step_ops_var3 | ( | int | m_A, |
int | m_T, | ||
float * | buff_A, | ||
int | rs_A, | ||
int | cs_A, | ||
float * | buff_Z, | ||
int | rs_Z, | ||
int | cs_Z, | ||
float * | buff_T, | ||
int | rs_T, | ||
int | cs_T | ||
) |
References bli_saxpyv(), bli_sdot(), bli_sgemv(), bli_sscalv(), bli_ssetm(), bli_ssymv(), FLA_free(), FLA_Househ2_UT_l_ops(), FLA_malloc(), FLA_MINUS_ONE, FLA_ONE, FLA_TWO, and FLA_ZERO.
Referenced by FLA_Tridiag_UT_l_step_opt_var3().
{ float* buff_2 = FLA_FLOAT_PTR( FLA_TWO ); float* buff_1 = FLA_FLOAT_PTR( FLA_ONE ); float* buff_0 = FLA_FLOAT_PTR( FLA_ZERO ); float* buff_m1 = FLA_FLOAT_PTR( FLA_MINUS_ONE ); float first_elem, last_elem; float beta; float inv_tau11; float minus_inv_tau11; int i; // b_alg = FLA_Obj_length( T ); int b_alg = m_T; // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &d ); // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &f ); float* buff_d = ( float* ) FLA_malloc( m_A * sizeof( *buff_A ) ); float* buff_f = ( float* ) FLA_malloc( m_A * sizeof( *buff_A ) ); int inc_d = 1; int inc_f = 1; // FLA_Obj_set_to_scalar( FLA_ZERO, Z ); bli_ssetm( m_A, b_alg, buff_0, buff_Z, rs_Z, cs_Z ); for ( i = 0; i < b_alg; ++i ) { float* a10t = buff_A + (0 )*cs_A + (i )*rs_A; float* A20 = buff_A + (0 )*cs_A + (i+1)*rs_A; float* alpha11 = buff_A + (i )*cs_A + (i )*rs_A; float* a21 = buff_A + (i )*cs_A + (i+1)*rs_A; float* A22 = buff_A + (i+1)*cs_A + (i+1)*rs_A; float* z10t = buff_Z + (0 )*cs_Z + (i )*rs_Z; float* Z20 = buff_Z + (0 )*cs_Z + (i+1)*rs_Z; float* z21 = buff_Z + (i )*cs_Z + (i+1)*rs_Z; float* t01 = buff_T + (i )*cs_T + (0 )*rs_T; float* tau11 = buff_T + (i )*cs_T + (i )*rs_T; float* d01 = buff_d + (0 )*inc_d; float* f01 = buff_f + (0 )*inc_f; float* a10t_r = a10t + (i-1)*cs_A + (0 )*rs_A; float* a21_t = a21 + (0 )*cs_A + (0 )*rs_A; float* a21_b = a21 + (0 )*cs_A + (1 )*rs_A; float* ABL = a10t; float* ZBL = z10t; float* a2 = alpha11; int m_ahead = m_A - i - 1; int m_behind = i; int n_behind = i; /*------------------------------------------------------------*/ if ( m_behind > 0 ) { // FLA_Copy( a10t_r, last_elem ); // FLA_Obj_set_to_scalar( FLA_ONE, a10t_r ); last_elem = *a10t_r; *a10t_r = *buff_1; } // FLA_Gemvc( FLA_NO_TRANSPOSE, FLA_CONJUGATE, FLA_MINUS_ONE, ABL, z10t, FLA_ONE, a2 ); // FLA_Gemvc( FLA_NO_TRANSPOSE, FLA_CONJUGATE, FLA_MINUS_ONE, ZBL, a10t, FLA_ONE, a2 ); bli_sgemv( BLIS_NO_TRANSPOSE, BLIS_CONJUGATE, m_ahead + 1, n_behind, buff_m1, ABL, rs_A, cs_A, z10t, cs_Z, buff_1, a2, rs_A ); bli_sgemv( BLIS_NO_TRANSPOSE, BLIS_CONJUGATE, m_ahead + 1, n_behind, buff_m1, ZBL, rs_Z, cs_Z, a10t, cs_A, buff_1, a2, rs_A ); if ( m_behind > 0 ) { // FLA_Copy( last_elem, a10t_r ); *a10t_r = last_elem; } if ( m_ahead > 0 ) { // FLA_Househ2_UT( FLA_LEFT, // a21_t, // a21_b, tau11 ); FLA_Househ2_UT_l_ops( m_ahead - 1, a21_t, a21_b, rs_A, tau11 ); // FLA_Obj_set_to_scalar( FLA_ONE, inv_tau11 ); // FLA_Inv_scalc( FLA_NO_CONJUGATE, tau11, inv_tau11 ); // FLA_Copy( inv_tau11, minus_inv_tau11 ); // FLA_Scal( FLA_MINUS_ONE, minus_inv_tau11 ); bli_sdiv3( buff_1, tau11, &inv_tau11 ); bli_sneg2( &inv_tau11, &minus_inv_tau11 ); // FLA_Copy( a21_t, first_elem ); // FLA_Obj_set_to_scalar( FLA_ONE, a21_t ); first_elem = *a21_t; *a21_t = *buff_1; // FLA_Hemv( FLA_LOWER_TRIANGULAR, FLA_ONE, A22, a21, FLA_ZERO, z21 ); bli_ssymv( BLIS_LOWER_TRIANGULAR, m_ahead, buff_1, A22, rs_A, cs_A, a21, rs_A, buff_0, z21, rs_Z ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, d01 ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, Z20, a21, FLA_ZERO, f01 ); bli_sgemv( BLIS_CONJ_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_1, A20, rs_A, cs_A, a21, rs_A, buff_0, d01, inc_d ); bli_sgemv( BLIS_CONJ_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_1, Z20, rs_Z, cs_Z, a21, rs_A, buff_0, f01, inc_f ); // FLA_Gemv( FLA_NO_TRANSPOSE, FLA_MINUS_ONE, A20, f01, FLA_ONE, z21 ); // FLA_Gemv( FLA_NO_TRANSPOSE, FLA_MINUS_ONE, Z20, d01, FLA_ONE, z21 ); bli_sgemv( BLIS_NO_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_m1, A20, rs_A, cs_A, f01, inc_f, buff_1, z21, rs_Z ); bli_sgemv( BLIS_NO_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_m1, Z20, rs_Z, cs_Z, d01, inc_d, buff_1, z21, rs_Z ); // FLA_Dotc( FLA_CONJUGATE, a21, z21, beta ); // FLA_Inv_scal( FLA_TWO, beta ); bli_sdot( BLIS_CONJUGATE, m_ahead, a21, rs_A, z21, rs_Z, &beta ); bli_sinvscals( buff_2, &beta ); // FLA_Scal( minus_inv_tau11, beta ); // FLA_Axpy( beta, a21, z21 ); // FLA_Scal( inv_tau11, z21 ); bli_sscals( &minus_inv_tau11, &beta ); bli_saxpyv( BLIS_NO_CONJUGATE, m_ahead, &beta, a21, rs_A, z21, rs_Z ); bli_sscalv( BLIS_NO_CONJUGATE, m_ahead, &inv_tau11, z21, rs_Z ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, t01 ); bli_sgemv( BLIS_CONJ_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_1, A20, rs_A, cs_A, a21, rs_A, buff_0, t01, rs_T ); // FLA_Copy( first_elem, a21_t ); *a21_t = first_elem; } /*------------------------------------------------------------*/ } // FLA_Obj_free( &d ); // FLA_Obj_free( &f ); FLA_free( buff_d ); FLA_free( buff_f ); return FLA_SUCCESS; }
References FLA_Obj_col_stride(), FLA_Obj_datatype(), FLA_Obj_length(), FLA_Obj_row_stride(), FLA_Tridiag_UT_l_step_opc_var1(), FLA_Tridiag_UT_l_step_opd_var1(), FLA_Tridiag_UT_l_step_ops_var1(), and FLA_Tridiag_UT_l_step_opz_var1().
Referenced by FLA_Tridiag_UT_l_blk_var1(), and FLA_Tridiag_UT_l_opt_var1().
{ FLA_Datatype datatype; int m_A, m_T; int rs_A, cs_A; int rs_T, cs_T; datatype = FLA_Obj_datatype( A ); m_A = FLA_Obj_length( A ); m_T = FLA_Obj_length( T ); rs_A = FLA_Obj_row_stride( A ); cs_A = FLA_Obj_col_stride( A ); 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 ); FLA_Tridiag_UT_l_step_ops_var1( m_A, m_T, buff_A, rs_A, cs_A, buff_T, rs_T, cs_T ); break; } case FLA_DOUBLE: { double* buff_A = FLA_DOUBLE_PTR( A ); double* buff_T = FLA_DOUBLE_PTR( T ); FLA_Tridiag_UT_l_step_opd_var1( m_A, m_T, buff_A, rs_A, cs_A, buff_T, rs_T, cs_T ); break; } case FLA_COMPLEX: { scomplex* buff_A = FLA_COMPLEX_PTR( A ); scomplex* buff_T = FLA_COMPLEX_PTR( T ); FLA_Tridiag_UT_l_step_opc_var1( m_A, m_T, buff_A, rs_A, cs_A, 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 ); FLA_Tridiag_UT_l_step_opz_var1( m_A, m_T, buff_A, rs_A, cs_A, buff_T, rs_T, cs_T ); break; } } return FLA_SUCCESS; }
References FLA_Obj_col_stride(), FLA_Obj_datatype(), FLA_Obj_length(), FLA_Obj_row_stride(), FLA_Tridiag_UT_l_step_opc_var2(), FLA_Tridiag_UT_l_step_opd_var2(), FLA_Tridiag_UT_l_step_ops_var2(), and FLA_Tridiag_UT_l_step_opz_var2().
Referenced by FLA_Tridiag_UT_l_blk_var2(), and FLA_Tridiag_UT_l_opt_var2().
{ FLA_Datatype datatype; int m_A, m_T; int rs_A, cs_A; int rs_T, cs_T; datatype = FLA_Obj_datatype( A ); m_A = FLA_Obj_length( A ); m_T = FLA_Obj_length( T ); rs_A = FLA_Obj_row_stride( A ); cs_A = FLA_Obj_col_stride( A ); 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 ); FLA_Tridiag_UT_l_step_ops_var2( m_A, m_T, buff_A, rs_A, cs_A, buff_T, rs_T, cs_T ); break; } case FLA_DOUBLE: { double* buff_A = FLA_DOUBLE_PTR( A ); double* buff_T = FLA_DOUBLE_PTR( T ); FLA_Tridiag_UT_l_step_opd_var2( m_A, m_T, buff_A, rs_A, cs_A, buff_T, rs_T, cs_T ); break; } case FLA_COMPLEX: { scomplex* buff_A = FLA_COMPLEX_PTR( A ); scomplex* buff_T = FLA_COMPLEX_PTR( T ); FLA_Tridiag_UT_l_step_opc_var2( m_A, m_T, buff_A, rs_A, cs_A, 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 ); FLA_Tridiag_UT_l_step_opz_var2( m_A, m_T, buff_A, rs_A, cs_A, buff_T, rs_T, cs_T ); break; } } return FLA_SUCCESS; }
References FLA_Obj_col_stride(), FLA_Obj_datatype(), FLA_Obj_length(), FLA_Obj_row_stride(), FLA_Tridiag_UT_l_step_opc_var3(), FLA_Tridiag_UT_l_step_opd_var3(), FLA_Tridiag_UT_l_step_ops_var3(), and FLA_Tridiag_UT_l_step_opz_var3().
Referenced by FLA_Tridiag_UT_l_blk_var3(), and FLA_Tridiag_UT_l_opt_var3().
{ FLA_Datatype datatype; int m_A, m_T; int rs_A, cs_A; int rs_Z, cs_Z; int rs_T, cs_T; datatype = FLA_Obj_datatype( A ); m_A = FLA_Obj_length( A ); m_T = FLA_Obj_length( T ); rs_A = FLA_Obj_row_stride( A ); cs_A = FLA_Obj_col_stride( A ); rs_Z = FLA_Obj_row_stride( Z ); cs_Z = FLA_Obj_col_stride( Z ); 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_Z = FLA_FLOAT_PTR( Z ); float* buff_T = FLA_FLOAT_PTR( T ); FLA_Tridiag_UT_l_step_ops_var3( m_A, m_T, buff_A, rs_A, cs_A, buff_Z, rs_Z, cs_Z, buff_T, rs_T, cs_T ); break; } case FLA_DOUBLE: { double* buff_A = FLA_DOUBLE_PTR( A ); double* buff_Z = FLA_DOUBLE_PTR( Z ); double* buff_T = FLA_DOUBLE_PTR( T ); FLA_Tridiag_UT_l_step_opd_var3( m_A, m_T, buff_A, rs_A, cs_A, buff_Z, rs_Z, cs_Z, buff_T, rs_T, cs_T ); break; } case FLA_COMPLEX: { scomplex* buff_A = FLA_COMPLEX_PTR( A ); scomplex* buff_Z = FLA_COMPLEX_PTR( Z ); scomplex* buff_T = FLA_COMPLEX_PTR( T ); FLA_Tridiag_UT_l_step_opc_var3( m_A, m_T, buff_A, rs_A, cs_A, buff_Z, rs_Z, cs_Z, buff_T, rs_T, cs_T ); break; } case FLA_DOUBLE_COMPLEX: { dcomplex* buff_A = FLA_DOUBLE_COMPLEX_PTR( A ); dcomplex* buff_Z = FLA_DOUBLE_COMPLEX_PTR( Z ); dcomplex* buff_T = FLA_DOUBLE_COMPLEX_PTR( T ); FLA_Tridiag_UT_l_step_opz_var3( m_A, m_T, buff_A, rs_A, cs_A, buff_Z, rs_Z, cs_Z, buff_T, rs_T, cs_T ); break; } } return FLA_SUCCESS; }
FLA_Error FLA_Tridiag_UT_l_step_opz_var1 | ( | int | m_A, |
int | m_T, | ||
dcomplex * | buff_A, | ||
int | rs_A, | ||
int | cs_A, | ||
dcomplex * | buff_T, | ||
int | rs_T, | ||
int | cs_T | ||
) |
References bli_zaxpyv(), bli_zdot(), bli_zgemv(), bli_zhemv(), bli_zher2(), bli_zscalv(), FLA_free(), FLA_Househ2_UT_l_opz(), FLA_malloc(), FLA_MINUS_ONE, FLA_ONE, FLA_TWO, and FLA_ZERO.
Referenced by FLA_Tridiag_UT_l_step_opt_var1().
{ dcomplex* buff_2 = FLA_DOUBLE_COMPLEX_PTR( FLA_TWO ); dcomplex* buff_1 = FLA_DOUBLE_COMPLEX_PTR( FLA_ONE ); dcomplex* buff_0 = FLA_DOUBLE_COMPLEX_PTR( FLA_ZERO ); dcomplex* buff_m1 = FLA_DOUBLE_COMPLEX_PTR( FLA_MINUS_ONE ); dcomplex first_elem; dcomplex beta; dcomplex inv_tau11; dcomplex minus_inv_tau11; int i; // b_alg = FLA_Obj_length( T ); int b_alg = m_T; // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &z ); dcomplex* buff_z = ( dcomplex* ) FLA_malloc( m_A * sizeof( *buff_A ) ); int inc_z = 1; for ( i = 0; i < b_alg; ++i ) { dcomplex* A20 = buff_A + (0 )*cs_A + (i+1)*rs_A; dcomplex* a21 = buff_A + (i )*cs_A + (i+1)*rs_A; dcomplex* A22 = buff_A + (i+1)*cs_A + (i+1)*rs_A; dcomplex* t01 = buff_T + (i )*cs_T + (0 )*rs_T; dcomplex* tau11 = buff_T + (i )*cs_T + (i )*rs_T; dcomplex* z21 = buff_z + (i+1)*inc_z; dcomplex* a21_t = a21 + (0 )*cs_A + (0 )*rs_A; dcomplex* a21_b = a21 + (0 )*cs_A + (1 )*rs_A; int m_ahead = m_A - i - 1; int n_behind = i; /*------------------------------------------------------------*/ if ( m_ahead > 0 ) { // FLA_Househ2_UT( FLA_LEFT, // a21_t, // a21_b, tau11 ); FLA_Househ2_UT_l_opz( m_ahead - 1, a21_t, a21_b, rs_A, tau11 ); // FLA_Obj_set_to_scalar( FLA_ONE, inv_tau11 ); // FLA_Inv_scalc( FLA_NO_CONJUGATE, tau11, inv_tau11 ); // FLA_Copy( inv_tau11, minus_inv_tau11 ); // FLA_Scal( FLA_MINUS_ONE, minus_inv_tau11 ); bli_zdiv3( buff_1, tau11, &inv_tau11 ); bli_zneg2( &inv_tau11, &minus_inv_tau11 ); // FLA_Copy( a21_t, first_elem ); // FLA_Obj_set_to_scalar( FLA_ONE, a21_t ); first_elem = *a21_t; *a21_t = *buff_1; // FLA_Hemv( FLA_LOWER_TRIANGULAR, FLA_ONE, A22, a21, FLA_ZERO, z21 ); bli_zhemv( BLIS_LOWER_TRIANGULAR, BLIS_NO_CONJUGATE, m_ahead, buff_1, A22, rs_A, cs_A, a21, rs_A, buff_0, z21, inc_z ); // FLA_Dotc( FLA_CONJUGATE, a21, z21, beta ); // FLA_Inv_scal( FLA_TWO, beta ); bli_zdot( BLIS_CONJUGATE, m_ahead, a21, rs_A, z21, inc_z, &beta ); bli_zinvscals( buff_2, &beta ); // FLA_Scal( minus_inv_tau11, beta ); // FLA_Axpy( beta, a21, z21 ); // FLA_Scal( inv_tau11, z21 ); bli_zscals( &minus_inv_tau11, &beta ); bli_zaxpyv( BLIS_NO_CONJUGATE, m_ahead, &beta, a21, rs_A, z21, inc_z ); bli_zscalv( BLIS_NO_CONJUGATE, m_ahead, &inv_tau11, z21, inc_z ); // FLA_Her2( FLA_LOWER_TRIANGULAR, FLA_MINUS_ONE, a21, z21, A22 ); bli_zher2( BLIS_LOWER_TRIANGULAR, BLIS_NO_TRANSPOSE, m_ahead, buff_m1, a21, rs_A, z21, inc_z, A22, rs_A, cs_A ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, t01 ); bli_zgemv( BLIS_CONJ_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_1, A20, rs_A, cs_A, a21, rs_A, buff_0, t01, rs_T ); // FLA_Copy( first_elem, a21_t ); *a21_t = first_elem; } /*------------------------------------------------------------*/ } // FLA_Obj_free( &z ); FLA_free( buff_z ); return FLA_SUCCESS; }
FLA_Error FLA_Tridiag_UT_l_step_opz_var2 | ( | int | m_A, |
int | m_T, | ||
dcomplex * | buff_A, | ||
int | rs_A, | ||
int | cs_A, | ||
dcomplex * | buff_T, | ||
int | rs_T, | ||
int | cs_T | ||
) |
References bli_zaxpyv(), bli_zcopyv(), bli_zdot(), bli_zgemv(), bli_zhemv(), bli_zher2(), bli_zscalv(), FLA_free(), FLA_Househ2_UT_l_opz(), FLA_malloc(), FLA_MINUS_ONE, FLA_ONE, FLA_TWO, and FLA_ZERO.
Referenced by FLA_Tridiag_UT_l_step_opt_var2().
{ dcomplex* buff_2 = FLA_DOUBLE_COMPLEX_PTR( FLA_TWO ); dcomplex* buff_1 = FLA_DOUBLE_COMPLEX_PTR( FLA_ONE ); dcomplex* buff_0 = FLA_DOUBLE_COMPLEX_PTR( FLA_ZERO ); dcomplex* buff_m1 = FLA_DOUBLE_COMPLEX_PTR( FLA_MINUS_ONE ); dcomplex first_elem; dcomplex beta; dcomplex inv_tau11; dcomplex minus_inv_tau11; dcomplex minus_upsilon11, minus_conj_upsilon11; dcomplex minus_zeta11, minus_conj_zeta11; int i; // b_alg = FLA_Obj_length( T ); int b_alg = m_T; // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &u ); // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &z ); // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &w ); dcomplex* buff_u = ( dcomplex* ) FLA_malloc( m_A * sizeof( *buff_A ) ); dcomplex* buff_z = ( dcomplex* ) FLA_malloc( m_A * sizeof( *buff_A ) ); dcomplex* buff_w = ( dcomplex* ) FLA_malloc( m_A * sizeof( *buff_A ) ); int inc_u = 1; int inc_z = 1; int inc_w = 1; // Initialize some variables (only to prevent compiler warnings). first_elem = *buff_0; minus_inv_tau11 = *buff_0; for ( i = 0; i < b_alg; ++i ) { dcomplex* A20 = buff_A + (0 )*cs_A + (i+1)*rs_A; dcomplex* alpha11 = buff_A + (i )*cs_A + (i )*rs_A; dcomplex* a21 = buff_A + (i )*cs_A + (i+1)*rs_A; dcomplex* A22 = buff_A + (i+1)*cs_A + (i+1)*rs_A; dcomplex* t01 = buff_T + (i )*cs_T + (0 )*rs_T; dcomplex* tau11 = buff_T + (i )*cs_T + (i )*rs_T; dcomplex* upsilon11= buff_u + (i )*inc_u; dcomplex* u21 = buff_u + (i+1)*inc_u; dcomplex* zeta11 = buff_z + (i )*inc_z; dcomplex* z21 = buff_z + (i+1)*inc_z; dcomplex* w21 = buff_w + (i+1)*inc_w; dcomplex* a21_t = a21 + (0 )*cs_A + (0 )*rs_A; dcomplex* a21_b = a21 + (0 )*cs_A + (1 )*rs_A; int m_ahead = m_A - i - 1; int m_behind = i; int n_behind = i; /*------------------------------------------------------------*/ if ( m_behind > 0 ) { // FLA_Copy( upsilon11, minus_upsilon11 ); // FLA_Scal( FLA_MINUS_ONE, minus_upsilon11 ); // FLA_Copy( minus_upsilon11, minus_conj_upsilon11 ); bli_zmult3( buff_m1, upsilon11, &minus_upsilon11 ); bli_zcopyconj( &minus_upsilon11, &minus_conj_upsilon11 ); // FLA_Copy( zeta11, minus_zeta11 ); // FLA_Scal( FLA_MINUS_ONE, minus_zeta11 ); // FLA_Copy( minus_zeta11, minus_conj_zeta11 ); bli_zmult3( buff_m1, zeta11, &minus_zeta11 ); bli_zcopyconj( &minus_zeta11, &minus_conj_zeta11 ); // FLA_Axpyt( FLA_CONJ_NO_TRANSPOSE, minus_upsilon11, zeta11, alpha11 ); // FLA_Axpyt( FLA_CONJ_NO_TRANSPOSE, minus_zeta11, upsilon11, alpha11 ); bli_zaxpyv( BLIS_CONJUGATE, 1, &minus_upsilon11, zeta11, 1, alpha11, 1 ); bli_zaxpyv( BLIS_CONJUGATE, 1, &minus_zeta11, upsilon11, 1, alpha11, 1 ); // FLA_Axpyt( FLA_NO_TRANSPOSE, minus_conj_zeta11, u21, a21 ); // FLA_Axpyt( FLA_NO_TRANSPOSE, minus_conj_upsilon11, z21, a21 ); bli_zaxpyv( BLIS_NO_CONJUGATE, m_ahead, &minus_conj_zeta11, u21, inc_u, a21, rs_A ); bli_zaxpyv( BLIS_NO_CONJUGATE, m_ahead, &minus_conj_upsilon11, z21, inc_z, a21, rs_A ); } if ( m_ahead > 0 ) { // FLA_Househ2_UT( FLA_LEFT, // a21_t, // a21_b, tau11 ); FLA_Househ2_UT_l_opz( m_ahead - 1, a21_t, a21_b, rs_A, tau11 ); // FLA_Obj_set_to_scalar( FLA_ONE, inv_tau11 ); // FLA_Inv_scalc( FLA_NO_CONJUGATE, tau11, inv_tau11 ); // FLA_Copy( inv_tau11, minus_inv_tau11 ); // FLA_Scal( FLA_MINUS_ONE, minus_inv_tau11 ); bli_zdiv3( buff_1, tau11, &inv_tau11 ); bli_zneg2( &inv_tau11, &minus_inv_tau11 ); // FLA_Copy( a21_t, first_elem ); // FLA_Obj_set_to_scalar( FLA_ONE, a21_t ); first_elem = *a21_t; *a21_t = *buff_1; } if ( m_behind > 0 ) { // FLA_Her2( FLA_LOWER_TRIANGULAR, FLA_MINUS_ONE, u21, z21, A22 ); bli_zher2( BLIS_LOWER_TRIANGULAR, BLIS_NO_TRANSPOSE, m_ahead, buff_m1, u21, inc_u, z21, inc_z, A22, rs_A, cs_A ); } if ( m_ahead > 0 ) { // FLA_Hemv( FLA_LOWER_TRIANGULAR, FLA_ONE, A22, a21, FLA_ZERO, w21 ); bli_zhemv( BLIS_LOWER_TRIANGULAR, BLIS_NO_CONJUGATE, m_ahead, buff_1, A22, rs_A, cs_A, a21, rs_A, buff_0, w21, inc_w ); // FLA_Copy( a21, u21 ); // FLA_Copy( w21, z21 ); bli_zcopyv( BLIS_NO_CONJUGATE, m_ahead, a21, rs_A, u21, inc_u ); bli_zcopyv( BLIS_NO_CONJUGATE, m_ahead, w21, inc_w, z21, inc_z ); // FLA_Dotc( FLA_CONJUGATE, a21, z21, beta ); // FLA_Inv_scal( FLA_TWO, beta ); bli_zdot( BLIS_CONJUGATE, m_ahead, a21, rs_A, z21, inc_z, &beta ); bli_zinvscals( buff_2, &beta ); // FLA_Scal( minus_inv_tau11, beta ); // FLA_Axpy( beta, a21, z21 ); // FLA_Scal( inv_tau11, z21 ); bli_zscals( &minus_inv_tau11, &beta ); bli_zaxpyv( BLIS_NO_CONJUGATE, m_ahead, &beta, a21, rs_A, z21, inc_z ); bli_zscalv( BLIS_NO_CONJUGATE, m_ahead, &inv_tau11, z21, inc_z ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, t01 ); bli_zgemv( BLIS_CONJ_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_1, A20, rs_A, cs_A, a21, rs_A, buff_0, t01, rs_T ); // FLA_Copy( first_elem, a21_t ); *a21_t = first_elem; } if ( m_behind + 1 == b_alg && m_ahead > 0 ) { // FLA_Her2( FLA_LOWER_TRIANGULAR, FLA_MINUS_ONE, u21, z21, A22 ); bli_zher2( BLIS_LOWER_TRIANGULAR, BLIS_NO_TRANSPOSE, m_ahead, buff_m1, u21, inc_u, z21, inc_z, A22, rs_A, cs_A ); } /*------------------------------------------------------------*/ } // FLA_Obj_free( &u ); // FLA_Obj_free( &z ); // FLA_Obj_free( &w ); FLA_free( buff_u ); FLA_free( buff_z ); FLA_free( buff_w ); return FLA_SUCCESS; }
FLA_Error FLA_Tridiag_UT_l_step_opz_var3 | ( | int | m_A, |
int | m_T, | ||
dcomplex * | buff_A, | ||
int | rs_A, | ||
int | cs_A, | ||
dcomplex * | buff_Z, | ||
int | rs_Z, | ||
int | cs_Z, | ||
dcomplex * | buff_T, | ||
int | rs_T, | ||
int | cs_T | ||
) |
References bli_zaxpyv(), bli_zdot(), bli_zgemv(), bli_zhemv(), bli_zscalv(), bli_zsetm(), FLA_free(), FLA_Househ2_UT_l_opz(), FLA_malloc(), FLA_MINUS_ONE, FLA_ONE, FLA_TWO, and FLA_ZERO.
Referenced by FLA_Tridiag_UT_l_step_opt_var3().
{ dcomplex* buff_2 = FLA_DOUBLE_COMPLEX_PTR( FLA_TWO ); dcomplex* buff_1 = FLA_DOUBLE_COMPLEX_PTR( FLA_ONE ); dcomplex* buff_0 = FLA_DOUBLE_COMPLEX_PTR( FLA_ZERO ); dcomplex* buff_m1 = FLA_DOUBLE_COMPLEX_PTR( FLA_MINUS_ONE ); dcomplex first_elem, last_elem; dcomplex beta; dcomplex inv_tau11; dcomplex minus_inv_tau11; int i; // b_alg = FLA_Obj_length( T ); int b_alg = m_T; // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &d ); // FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &f ); dcomplex* buff_d = ( dcomplex* ) FLA_malloc( m_A * sizeof( *buff_A ) ); dcomplex* buff_f = ( dcomplex* ) FLA_malloc( m_A * sizeof( *buff_A ) ); int inc_d = 1; int inc_f = 1; // FLA_Obj_set_to_scalar( FLA_ZERO, Z ); bli_zsetm( m_A, b_alg, buff_0, buff_Z, rs_Z, cs_Z ); for ( i = 0; i < b_alg; ++i ) { dcomplex* a10t = buff_A + (0 )*cs_A + (i )*rs_A; dcomplex* A20 = buff_A + (0 )*cs_A + (i+1)*rs_A; dcomplex* alpha11 = buff_A + (i )*cs_A + (i )*rs_A; dcomplex* a21 = buff_A + (i )*cs_A + (i+1)*rs_A; dcomplex* A22 = buff_A + (i+1)*cs_A + (i+1)*rs_A; dcomplex* z10t = buff_Z + (0 )*cs_Z + (i )*rs_Z; dcomplex* Z20 = buff_Z + (0 )*cs_Z + (i+1)*rs_Z; dcomplex* z21 = buff_Z + (i )*cs_Z + (i+1)*rs_Z; dcomplex* t01 = buff_T + (i )*cs_T + (0 )*rs_T; dcomplex* tau11 = buff_T + (i )*cs_T + (i )*rs_T; dcomplex* d01 = buff_d + (0 )*inc_d; dcomplex* f01 = buff_f + (0 )*inc_f; dcomplex* a10t_r = a10t + (i-1)*cs_A + (0 )*rs_A; dcomplex* a21_t = a21 + (0 )*cs_A + (0 )*rs_A; dcomplex* a21_b = a21 + (0 )*cs_A + (1 )*rs_A; dcomplex* ABL = a10t; dcomplex* ZBL = z10t; dcomplex* a2 = alpha11; int m_ahead = m_A - i - 1; int m_behind = i; int n_behind = i; /*------------------------------------------------------------*/ if ( m_behind > 0 ) { // FLA_Copy( a10t_r, last_elem ); // FLA_Obj_set_to_scalar( FLA_ONE, a10t_r ); last_elem = *a10t_r; *a10t_r = *buff_1; } // FLA_Gemvc( FLA_NO_TRANSPOSE, FLA_CONJUGATE, FLA_MINUS_ONE, ABL, z10t, FLA_ONE, a2 ); // FLA_Gemvc( FLA_NO_TRANSPOSE, FLA_CONJUGATE, FLA_MINUS_ONE, ZBL, a10t, FLA_ONE, a2 ); bli_zgemv( BLIS_NO_TRANSPOSE, BLIS_CONJUGATE, m_ahead + 1, n_behind, buff_m1, ABL, rs_A, cs_A, z10t, cs_Z, buff_1, a2, rs_A ); bli_zgemv( BLIS_NO_TRANSPOSE, BLIS_CONJUGATE, m_ahead + 1, n_behind, buff_m1, ZBL, rs_Z, cs_Z, a10t, cs_A, buff_1, a2, rs_A ); if ( m_behind > 0 ) { // FLA_Copy( last_elem, a10t_r ); *a10t_r = last_elem; } if ( m_ahead > 0 ) { // FLA_Househ2_UT( FLA_LEFT, // a21_t, // a21_b, tau11 ); FLA_Househ2_UT_l_opz( m_ahead - 1, a21_t, a21_b, rs_A, tau11 ); // FLA_Obj_set_to_scalar( FLA_ONE, inv_tau11 ); // FLA_Inv_scalc( FLA_NO_CONJUGATE, tau11, inv_tau11 ); // FLA_Copy( inv_tau11, minus_inv_tau11 ); // FLA_Scal( FLA_MINUS_ONE, minus_inv_tau11 ); bli_zdiv3( buff_1, tau11, &inv_tau11 ); bli_zneg2( &inv_tau11, &minus_inv_tau11 ); // FLA_Copy( a21_t, first_elem ); // FLA_Obj_set_to_scalar( FLA_ONE, a21_t ); first_elem = *a21_t; *a21_t = *buff_1; // FLA_Hemv( FLA_LOWER_TRIANGULAR, FLA_ONE, A22, a21, FLA_ZERO, z21 ); bli_zhemv( BLIS_LOWER_TRIANGULAR, BLIS_NO_CONJUGATE, m_ahead, buff_1, A22, rs_A, cs_A, a21, rs_A, buff_0, z21, rs_Z ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, d01 ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, Z20, a21, FLA_ZERO, f01 ); bli_zgemv( BLIS_CONJ_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_1, A20, rs_A, cs_A, a21, rs_A, buff_0, d01, inc_d ); bli_zgemv( BLIS_CONJ_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_1, Z20, rs_Z, cs_Z, a21, rs_A, buff_0, f01, inc_f ); // FLA_Gemv( FLA_NO_TRANSPOSE, FLA_MINUS_ONE, A20, f01, FLA_ONE, z21 ); // FLA_Gemv( FLA_NO_TRANSPOSE, FLA_MINUS_ONE, Z20, d01, FLA_ONE, z21 ); bli_zgemv( BLIS_NO_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_m1, A20, rs_A, cs_A, f01, inc_f, buff_1, z21, rs_Z ); bli_zgemv( BLIS_NO_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_m1, Z20, rs_Z, cs_Z, d01, inc_d, buff_1, z21, rs_Z ); // FLA_Dotc( FLA_CONJUGATE, a21, z21, beta ); // FLA_Inv_scal( FLA_TWO, beta ); bli_zdot( BLIS_CONJUGATE, m_ahead, a21, rs_A, z21, rs_Z, &beta ); bli_zinvscals( buff_2, &beta ); // FLA_Scal( minus_inv_tau11, beta ); // FLA_Axpy( beta, a21, z21 ); // FLA_Scal( inv_tau11, z21 ); bli_zscals( &minus_inv_tau11, &beta ); bli_zaxpyv( BLIS_NO_CONJUGATE, m_ahead, &beta, a21, rs_A, z21, rs_Z ); bli_zscalv( BLIS_NO_CONJUGATE, m_ahead, &inv_tau11, z21, rs_Z ); // FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, t01 ); bli_zgemv( BLIS_CONJ_TRANSPOSE, BLIS_NO_CONJUGATE, m_ahead, n_behind, buff_1, A20, rs_A, cs_A, a21, rs_A, buff_0, t01, rs_T ); // FLA_Copy( first_elem, a21_t ); *a21_t = first_elem; } /*------------------------------------------------------------*/ } // FLA_Obj_free( &d ); // FLA_Obj_free( &f ); FLA_free( buff_d ); FLA_free( buff_f ); return FLA_SUCCESS; }
References FLA_Axpy(), FLA_Cont_with_3x1_to_2x1(), FLA_Cont_with_3x3_to_2x2(), FLA_Copy(), FLA_Dotc(), FLA_Gemv(), FLA_Hemv(), FLA_Her2(), FLA_Househ2_UT(), FLA_Inv_scal(), FLA_Inv_scalc(), FLA_MINUS_ONE, FLA_Obj_create(), FLA_Obj_datatype(), FLA_Obj_free(), FLA_Obj_length(), FLA_Obj_set_to_scalar(), FLA_ONE, FLA_Part_2x1(), FLA_Part_2x2(), FLA_Repart_2x1_to_3x1(), FLA_Repart_2x2_to_3x3(), FLA_Scal(), FLA_TWO, and FLA_ZERO.
Referenced by FLA_Tridiag_UT_l_unb_var1().
{ FLA_Obj ATL, ATR, A00, a01, A02, ABL, ABR, a10t, alpha11, a12t, A20, a21, A22; FLA_Obj TTL, TTR, T00, t01, T02, TBL, TBR, t10t, tau11, t12t, T20, t21, T22; FLA_Obj zT, z01, zB, zeta11, z21; FLA_Obj z; FLA_Obj inv_tau11; FLA_Obj minus_inv_tau11; FLA_Obj first_elem; FLA_Obj beta; FLA_Obj a21_t, a21_b; FLA_Datatype datatype_A; dim_t m_A; dim_t b_alg; b_alg = FLA_Obj_length( T ); datatype_A = FLA_Obj_datatype( A ); m_A = FLA_Obj_length( A ); FLA_Obj_create( datatype_A, 1, 1, 0, 0, &inv_tau11 ); FLA_Obj_create( datatype_A, 1, 1, 0, 0, &minus_inv_tau11 ); FLA_Obj_create( datatype_A, 1, 1, 0, 0, &first_elem ); FLA_Obj_create( datatype_A, 1, 1, 0, 0, &beta ); FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &z ); FLA_Part_2x2( A, &ATL, &ATR, &ABL, &ABR, 0, 0, FLA_TL ); FLA_Part_2x2( T, &TTL, &TTR, &TBL, &TBR, 0, 0, FLA_TL ); FLA_Part_2x1( z, &zT, &zB, 0, FLA_TOP ); while ( FLA_Obj_length( ATL ) < b_alg ) { FLA_Repart_2x2_to_3x3( ATL, /**/ ATR, &A00, /**/ &a01, &A02, /* ************* */ /* ************************** */ &a10t, /**/ &alpha11, &a12t, ABL, /**/ ABR, &A20, /**/ &a21, &A22, 1, 1, FLA_BR ); FLA_Repart_2x2_to_3x3( TTL, /**/ TTR, &T00, /**/ &t01, &T02, /* ************* */ /* ************************ */ &t10t, /**/ &tau11, &t12t, TBL, /**/ TBR, &T20, /**/ &t21, &T22, 1, 1, FLA_BR ); FLA_Repart_2x1_to_3x1( zT, &z01, /* ** */ /* ****** */ &zeta11, zB, &z21, 1, FLA_BOTTOM ); /*------------------------------------------------------------*/ if ( FLA_Obj_length( A22 ) > 0 ) { FLA_Part_2x1( a21, &a21_t, &a21_b, 1, FLA_TOP ); // [ u21, tau11, a21 ] = House( a21 ); FLA_Househ2_UT( FLA_LEFT, a21_t, a21_b, tau11 ); // inv_tau11 = 1 / tau11; // minus_inv_tau11 = -1 / tau11; FLA_Obj_set_to_scalar( FLA_ONE, inv_tau11 ); FLA_Inv_scalc( FLA_NO_CONJUGATE, tau11, inv_tau11 ); FLA_Copy( inv_tau11, minus_inv_tau11 ); FLA_Scal( FLA_MINUS_ONE, minus_inv_tau11 ); // Save first element of a21_t and set it to one so we can use a21 as // u21 in subsequent computations. We will restore a21_t later on. FLA_Copy( a21_t, first_elem ); FLA_Obj_set_to_scalar( FLA_ONE, a21_t ); // z21 = A22 * u21; FLA_Hemv( FLA_LOWER_TRIANGULAR, FLA_ONE, A22, a21, FLA_ZERO, z21 ); // beta = u21' * z21 / 2; FLA_Dotc( FLA_CONJUGATE, a21, z21, beta ); FLA_Inv_scal( FLA_TWO, beta ); // z21 = ( z21 - beta / tau * u21 ) / tau; FLA_Scal( minus_inv_tau11, beta ); FLA_Axpy( beta, a21, z21 ); FLA_Scal( inv_tau11, z21 ); // A22 = A22 - u21 * z21' - z21 * u21'; FLA_Her2( FLA_LOWER_TRIANGULAR, FLA_MINUS_ONE, a21, z21, A22 ); // t01 = U20' * u21; FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, t01 ); // Restore first element of a21. FLA_Copy( first_elem, a21_t ); } /*------------------------------------------------------------*/ FLA_Cont_with_3x3_to_2x2( &ATL, /**/ &ATR, A00, a01, /**/ A02, a10t, alpha11, /**/ a12t, /* ************** */ /* ************************ */ &ABL, /**/ &ABR, A20, a21, /**/ A22, FLA_TL ); FLA_Cont_with_3x3_to_2x2( &TTL, /**/ &TTR, T00, t01, /**/ T02, t10t, tau11, /**/ t12t, /* ************** */ /* ********************** */ &TBL, /**/ &TBR, T20, t21, /**/ T22, FLA_TL ); FLA_Cont_with_3x1_to_2x1( &zT, z01, zeta11, /* ** */ /* ****** */ &zB, z21, FLA_TOP ); } FLA_Obj_free( &inv_tau11 ); FLA_Obj_free( &minus_inv_tau11 ); FLA_Obj_free( &first_elem ); FLA_Obj_free( &beta ); FLA_Obj_free( &z ); return FLA_SUCCESS; }
References FLA_Axpy(), FLA_Axpyt(), FLA_Cont_with_3x1_to_2x1(), FLA_Cont_with_3x3_to_2x2(), FLA_Copy(), FLA_Copyt(), FLA_Dotc(), FLA_Gemv(), FLA_Hemv(), FLA_Her2(), FLA_Househ2_UT(), FLA_Inv_scal(), FLA_Inv_scalc(), FLA_MINUS_ONE, FLA_Obj_create(), FLA_Obj_datatype(), FLA_Obj_free(), FLA_Obj_length(), FLA_Obj_set_to_scalar(), FLA_ONE, FLA_Part_2x1(), FLA_Part_2x2(), FLA_Repart_2x1_to_3x1(), FLA_Repart_2x2_to_3x3(), FLA_Scal(), FLA_TWO, and FLA_ZERO.
Referenced by FLA_Tridiag_UT_l_unb_var2().
{ FLA_Obj ATL, ATR, A00, a01, A02, ABL, ABR, a10t, alpha11, a12t, A20, a21, A22; FLA_Obj TTL, TTR, T00, t01, T02, TBL, TBR, t10t, tau11, t12t, T20, t21, T22; FLA_Obj uT, u01, uB, upsilon11, u21; FLA_Obj zT, z01, zB, zeta11, z21; FLA_Obj wT, w01, wB, omega11, w21; FLA_Obj u, z, w; FLA_Obj inv_tau11; FLA_Obj minus_inv_tau11; FLA_Obj first_elem; FLA_Obj beta; FLA_Obj minus_upsilon11; FLA_Obj minus_conj_upsilon11; FLA_Obj minus_zeta11; FLA_Obj minus_conj_zeta11; FLA_Obj a21_t, a21_b; FLA_Datatype datatype_A; dim_t m_A; dim_t b_alg; b_alg = FLA_Obj_length( T ); datatype_A = FLA_Obj_datatype( A ); m_A = FLA_Obj_length( A ); FLA_Obj_create( datatype_A, 1, 1, 0, 0, &inv_tau11 ); FLA_Obj_create( datatype_A, 1, 1, 0, 0, &minus_inv_tau11 ); FLA_Obj_create( datatype_A, 1, 1, 0, 0, &first_elem ); FLA_Obj_create( datatype_A, 1, 1, 0, 0, &beta ); FLA_Obj_create( datatype_A, 1, 1, 0, 0, &minus_upsilon11 ); FLA_Obj_create( datatype_A, 1, 1, 0, 0, &minus_conj_upsilon11 ); FLA_Obj_create( datatype_A, 1, 1, 0, 0, &minus_zeta11 ); FLA_Obj_create( datatype_A, 1, 1, 0, 0, &minus_conj_zeta11 ); FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &u ); FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &z ); FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &w ); FLA_Part_2x2( A, &ATL, &ATR, &ABL, &ABR, 0, 0, FLA_TL ); FLA_Part_2x2( T, &TTL, &TTR, &TBL, &TBR, 0, 0, FLA_TL ); FLA_Part_2x1( u, &uT, &uB, 0, FLA_TOP ); FLA_Part_2x1( z, &zT, &zB, 0, FLA_TOP ); FLA_Part_2x1( w, &wT, &wB, 0, FLA_TOP ); while ( FLA_Obj_length( ATL ) < b_alg ) { FLA_Repart_2x2_to_3x3( ATL, /**/ ATR, &A00, /**/ &a01, &A02, /* ************* */ /* ************************** */ &a10t, /**/ &alpha11, &a12t, ABL, /**/ ABR, &A20, /**/ &a21, &A22, 1, 1, FLA_BR ); FLA_Repart_2x2_to_3x3( TTL, /**/ TTR, &T00, /**/ &t01, &T02, /* ************* */ /* ************************ */ &t10t, /**/ &tau11, &t12t, TBL, /**/ TBR, &T20, /**/ &t21, &T22, 1, 1, FLA_BR ); FLA_Repart_2x1_to_3x1( uT, &u01, /* ** */ /* ********* */ &upsilon11, uB, &u21, 1, FLA_BOTTOM ); FLA_Repart_2x1_to_3x1( zT, &z01, /* ** */ /* ****** */ &zeta11, zB, &z21, 1, FLA_BOTTOM ); FLA_Repart_2x1_to_3x1( wT, &w01, /* ** */ /* ******* */ &omega11, wB, &w21, 1, FLA_BOTTOM ); /*------------------------------------------------------------*/ if ( FLA_Obj_length( ATL ) > 0 ) { FLA_Copy( upsilon11, minus_upsilon11 ); FLA_Scal( FLA_MINUS_ONE, minus_upsilon11 ); FLA_Copyt( FLA_CONJ_NO_TRANSPOSE, minus_upsilon11, minus_conj_upsilon11 ); FLA_Copy( zeta11, minus_zeta11 ); FLA_Scal( FLA_MINUS_ONE, minus_zeta11 ); FLA_Copyt( FLA_CONJ_NO_TRANSPOSE, minus_zeta11, minus_conj_zeta11 ); // alpha11 = alpha11 - upsilon11 * conj(zeta11) - zeta11 * conj(upsilon11); FLA_Axpyt( FLA_CONJ_NO_TRANSPOSE, minus_upsilon11, zeta11, alpha11 ); FLA_Axpyt( FLA_CONJ_NO_TRANSPOSE, minus_zeta11, upsilon11, alpha11 ); // a21 = a21 - conj(zeta11) * u21 - conj(upsilon11) * z21; FLA_Axpyt( FLA_NO_TRANSPOSE, minus_conj_zeta11, u21, a21 ); FLA_Axpyt( FLA_NO_TRANSPOSE, minus_conj_upsilon11, z21, a21 ); } if ( FLA_Obj_length( A22 ) > 0 ) { FLA_Part_2x1( a21, &a21_t, &a21_b, 1, FLA_TOP ); // [ x21, tau11, a21 ] = House( a21 ); FLA_Househ2_UT( FLA_LEFT, a21_t, a21_b, tau11 ); // inv_tau11 = 1 / tau11; // minus_inv_tau11 = -1 / tau11; FLA_Obj_set_to_scalar( FLA_ONE, inv_tau11 ); FLA_Inv_scalc( FLA_NO_CONJUGATE, tau11, inv_tau11 ); FLA_Copy( inv_tau11, minus_inv_tau11 ); FLA_Scal( FLA_MINUS_ONE, minus_inv_tau11 ); // Save first element of a21_t and set it to one so we can use a21 as // u21 in subsequent computations. We will restore a21_t later on. FLA_Copy( a21_t, first_elem ); FLA_Obj_set_to_scalar( FLA_ONE, a21_t ); } if ( FLA_Obj_length( ATL ) > 0 ) { // A22 = A22 - u21 * z21' - z21 * u21'; FLA_Her2( FLA_LOWER_TRIANGULAR, FLA_MINUS_ONE, u21, z21, A22 ); } if ( FLA_Obj_length( A22 ) > 0 ) { // w21 = A22 * x21; FLA_Hemv( FLA_LOWER_TRIANGULAR, FLA_ONE, A22, a21, FLA_ZERO, w21 ); // u21 = x21; // z21 = w21; FLA_Copy( a21, u21 ); FLA_Copy( w21, z21 ); // beta = u21' * z21 / 2; FLA_Dotc( FLA_CONJUGATE, a21, z21, beta ); FLA_Inv_scal( FLA_TWO, beta ); // z21 = ( z21 - beta / tau * u21 ) / tau; FLA_Scal( minus_inv_tau11, beta ); FLA_Axpy( beta, a21, z21 ); FLA_Scal( inv_tau11, z21 ); // t01 = U20' * u21; FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, t01 ); // Restore first element of a21. FLA_Copy( first_elem, a21_t ); } // Update A22 if this is the last iteration; this is needed when we're // being called from the blocked routine so A22 is left in a valid state. if ( FLA_Obj_length( ATL ) + 1 == b_alg && FLA_Obj_length( A22 ) > 0 ) { // A22 = A22 - u21 * y21' - z21 * u21'; FLA_Her2( FLA_LOWER_TRIANGULAR, FLA_MINUS_ONE, u21, z21, A22 ); } /*------------------------------------------------------------*/ FLA_Cont_with_3x3_to_2x2( &ATL, /**/ &ATR, A00, a01, /**/ A02, a10t, alpha11, /**/ a12t, /* ************** */ /* ************************ */ &ABL, /**/ &ABR, A20, a21, /**/ A22, FLA_TL ); FLA_Cont_with_3x3_to_2x2( &TTL, /**/ &TTR, T00, t01, /**/ T02, t10t, tau11, /**/ t12t, /* ************** */ /* ********************** */ &TBL, /**/ &TBR, T20, t21, /**/ T22, FLA_TL ); FLA_Cont_with_3x1_to_2x1( &uT, u01, upsilon11, /* ** */ /* ********* */ &uB, u21, FLA_TOP ); FLA_Cont_with_3x1_to_2x1( &zT, z01, zeta11, /* ** */ /* ****** */ &zB, z21, FLA_TOP ); FLA_Cont_with_3x1_to_2x1( &wT, w01, omega11, /* ** */ /* ******* */ &wB, w21, FLA_TOP ); } FLA_Obj_free( &inv_tau11 ); FLA_Obj_free( &minus_inv_tau11 ); FLA_Obj_free( &first_elem ); FLA_Obj_free( &beta ); FLA_Obj_free( &minus_upsilon11 ); FLA_Obj_free( &minus_conj_upsilon11 ); FLA_Obj_free( &minus_zeta11 ); FLA_Obj_free( &minus_conj_zeta11 ); FLA_Obj_free( &u ); FLA_Obj_free( &z ); FLA_Obj_free( &w ); return FLA_SUCCESS; }
References FLA_Axpy(), FLA_Cont_with_3x1_to_2x1(), FLA_Cont_with_3x3_to_2x2(), FLA_Copy(), FLA_Dotc(), FLA_Gemv(), FLA_Gemvc(), FLA_Hemv(), FLA_Househ2_UT(), FLA_Inv_scal(), FLA_Inv_scalc(), FLA_Merge_2x1(), FLA_MINUS_ONE, FLA_Obj_create(), FLA_Obj_datatype(), FLA_Obj_free(), FLA_Obj_length(), FLA_Obj_set_to_scalar(), FLA_ONE, FLA_Part_1x2(), FLA_Part_2x1(), FLA_Part_2x2(), FLA_Repart_2x1_to_3x1(), FLA_Repart_2x2_to_3x3(), FLA_Scal(), FLA_TWO, and FLA_ZERO.
Referenced by FLA_Tridiag_UT_l_unb_var3().
{ FLA_Obj ATL, ATR, A00, a01, A02, ABL, ABR, a10t, alpha11, a12t, A20, a21, A22; FLA_Obj ZTL, ZTR, Z00, z011, Z02, ZBL, ZBR, z10t, zeta11, z12t, Z20, z21, Z22; FLA_Obj TTL, TTR, T00, t01, T02, TBL, TBR, t10t, tau11, t12t, T20, t21, T22; FLA_Obj dT, d01, dB, delta11, d21; FLA_Obj fT, f01, fB, phi11, f21; FLA_Obj d, f; FLA_Obj inv_tau11; FLA_Obj minus_inv_tau11; FLA_Obj beta; FLA_Obj first_elem; FLA_Obj last_elem; FLA_Obj a10t_l, a10t_r; FLA_Obj a21_t, a21_b; FLA_Obj a2; FLA_Datatype datatype_A; dim_t m_A; dim_t b_alg; b_alg = FLA_Obj_length( T ); datatype_A = FLA_Obj_datatype( A ); m_A = FLA_Obj_length( A ); FLA_Obj_create( datatype_A, 1, 1, 0, 0, &inv_tau11 ); FLA_Obj_create( datatype_A, 1, 1, 0, 0, &minus_inv_tau11 ); FLA_Obj_create( datatype_A, 1, 1, 0, 0, &beta ); FLA_Obj_create( datatype_A, 1, 1, 0, 0, &first_elem ); FLA_Obj_create( datatype_A, 1, 1, 0, 0, &last_elem ); FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &d ); FLA_Obj_create( datatype_A, m_A, 1, 0, 0, &f ); FLA_Obj_set_to_scalar( FLA_ZERO, Z ); FLA_Part_2x2( A, &ATL, &ATR, &ABL, &ABR, 0, 0, FLA_TL ); FLA_Part_2x2( Z, &ZTL, &ZTR, &ZBL, &ZBR, 0, 0, FLA_TL ); FLA_Part_2x2( T, &TTL, &TTR, &TBL, &TBR, 0, 0, FLA_TL ); FLA_Part_2x1( d, &dT, &dB, 0, FLA_TOP ); FLA_Part_2x1( f, &fT, &fB, 0, FLA_TOP ); while ( FLA_Obj_length( ATL ) < b_alg ) { FLA_Repart_2x2_to_3x3( ATL, /**/ ATR, &A00, /**/ &a01, &A02, /* ************* */ /* ************************** */ &a10t, /**/ &alpha11, &a12t, ABL, /**/ ABR, &A20, /**/ &a21, &A22, 1, 1, FLA_BR ); FLA_Repart_2x2_to_3x3( ZTL, /**/ ZTR, &Z00, /**/ &z011, &Z02, /* ************* */ /* ************************* */ &z10t, /**/ &zeta11, &z12t, ZBL, /**/ ZBR, &Z20, /**/ &z21, &Z22, 1, 1, FLA_BR ); FLA_Repart_2x2_to_3x3( TTL, /**/ TTR, &T00, /**/ &t01, &T02, /* ************* */ /* ************************ */ &t10t, /**/ &tau11, &t12t, TBL, /**/ TBR, &T20, /**/ &t21, &T22, 1, 1, FLA_BR ); FLA_Repart_2x1_to_3x1( dT, &d01, /* ** */ /* ******* */ &delta11, dB, &d21, 1, FLA_BOTTOM ); FLA_Repart_2x1_to_3x1( fT, &f01, /* ** */ /* ***** */ &phi11, fB, &f21, 1, FLA_BOTTOM ); /*------------------------------------------------------------*/ // Save first element of a10_r and set it to one so we can use a10t as // u10t in subsequent computations. We will restore a10_r later on. if ( FLA_Obj_length( ATL ) > 0 ) { FLA_Part_1x2( a10t, &a10t_l, &a10t_r, 1, FLA_RIGHT ); FLA_Copy( a10t_r, last_elem ); FLA_Obj_set_to_scalar( FLA_ONE, a10t_r ); } FLA_Merge_2x1( alpha11, a21, &a2 ); // alpha11 = alpha11 - u10t * z10t' - z10t * u10t'; // a21 = a21 - U20 * z10t' - Z20 * u10t'; FLA_Gemvc( FLA_NO_TRANSPOSE, FLA_CONJUGATE, FLA_MINUS_ONE, ABL, z10t, FLA_ONE, a2 ); FLA_Gemvc( FLA_NO_TRANSPOSE, FLA_CONJUGATE, FLA_MINUS_ONE, ZBL, a10t, FLA_ONE, a2 ); // Restore last element of a10t. if ( FLA_Obj_length( ATL ) > 0 ) { FLA_Copy( last_elem, a10t_r ); } if ( FLA_Obj_length( A22 ) > 0 ) { FLA_Part_2x1( a21, &a21_t, &a21_b, 1, FLA_TOP ); // [ u21, tau11, a21 ] = House( a21 ); FLA_Househ2_UT( FLA_LEFT, a21_t, a21_b, tau11 ); // inv_tau11 = 1 / tau11; // minus_inv_tau11 = -1 / tau11; FLA_Obj_set_to_scalar( FLA_ONE, inv_tau11 ); FLA_Inv_scalc( FLA_NO_CONJUGATE, tau11, inv_tau11 ); FLA_Copy( inv_tau11, minus_inv_tau11 ); FLA_Scal( FLA_MINUS_ONE, minus_inv_tau11 ); // Save first element of a21_t and set it to one. FLA_Copy( a21_t, first_elem ); FLA_Obj_set_to_scalar( FLA_ONE, a21_t ); // z21 = A22 * u21; FLA_Hemv( FLA_LOWER_TRIANGULAR, FLA_ONE, A22, a21, FLA_ZERO, z21 ); FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, d01 ); FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, Z20, a21, FLA_ZERO, f01 ); // z21 = z21 - U20 * ( Z20' * u21 ) - Z20 * ( U20' * u21 ); FLA_Gemv( FLA_NO_TRANSPOSE, FLA_MINUS_ONE, A20, f01, FLA_ONE, z21 ); FLA_Gemv( FLA_NO_TRANSPOSE, FLA_MINUS_ONE, Z20, d01, FLA_ONE, z21 ); // beta = u21' * z21 / 2; FLA_Dotc( FLA_CONJUGATE, a21, z21, beta ); FLA_Inv_scal( FLA_TWO, beta ); // z21 = ( z21 - beta / tau * u21 ) / tau; FLA_Scal( minus_inv_tau11, beta ); FLA_Axpy( beta, a21, z21 ); FLA_Scal( inv_tau11, z21 ); // t01 = U20' * u21; FLA_Gemv( FLA_CONJ_TRANSPOSE, FLA_ONE, A20, a21, FLA_ZERO, t01 ); // Restore first element of a21. FLA_Copy( first_elem, a21_t ); } /*------------------------------------------------------------*/ FLA_Cont_with_3x3_to_2x2( &ATL, /**/ &ATR, A00, a01, /**/ A02, a10t, alpha11, /**/ a12t, /* ************** */ /* ************************ */ &ABL, /**/ &ABR, A20, a21, /**/ A22, FLA_TL ); FLA_Cont_with_3x3_to_2x2( &ZTL, /**/ &ZTR, Z00, z011, /**/ Z02, z10t, zeta11, /**/ z12t, /* ************** */ /* *********************** */ &ZBL, /**/ &ZBR, Z20, z21, /**/ Z22, FLA_TL ); FLA_Cont_with_3x3_to_2x2( &TTL, /**/ &TTR, T00, t01, /**/ T02, t10t, tau11, /**/ t12t, /* ************** */ /* ********************** */ &TBL, /**/ &TBR, T20, t21, /**/ T22, FLA_TL ); FLA_Cont_with_3x1_to_2x1( &dT, d01, delta11, /* ** */ /* ******* */ &dB, d21, FLA_TOP ); FLA_Cont_with_3x1_to_2x1( &fT, f01, phi11, /* ** */ /* ***** */ &fB, f21, FLA_TOP ); } FLA_Obj_free( &inv_tau11 ); FLA_Obj_free( &minus_inv_tau11 ); FLA_Obj_free( &beta ); FLA_Obj_free( &first_elem ); FLA_Obj_free( &last_elem ); FLA_Obj_free( &d ); FLA_Obj_free( &f ); return FLA_SUCCESS; }
References FLA_Tridiag_UT_l_step_unb_var1().
Referenced by FLA_Tridiag_UT_l().
{ return FLA_Tridiag_UT_l_step_unb_var1( A, T ); }
References FLA_Tridiag_UT_l_step_unb_var2().
Referenced by FLA_Tridiag_UT_l().
{ return FLA_Tridiag_UT_l_step_unb_var2( A, T ); }
References FLA_Obj_create_conf_to(), FLA_Obj_free(), and FLA_Tridiag_UT_l_step_unb_var3().
Referenced by FLA_Tridiag_UT_l().
{ FLA_Error r_val; FLA_Obj Z; //FLA_Obj_create_conf_to( FLA_NO_TRANSPOSE, A, &Y ); FLA_Obj_create_conf_to( FLA_NO_TRANSPOSE, A, &Z ); r_val = FLA_Tridiag_UT_l_step_unb_var3( A, Z, T ); //FLA_Obj_free( &Y ); FLA_Obj_free( &Z ); return r_val; }