|
libflame
revision_anchor
|
Go to the source code of this file.
Functions | |
| FLA_Error | FLA_Tridiag_UT_l_blk_var1 (FLA_Obj A, FLA_Obj T) |
| FLA_Error | FLA_Tridiag_UT_l_unb_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_blk_var2 (FLA_Obj A, FLA_Obj T) |
| FLA_Error | FLA_Tridiag_UT_l_blf_var2 (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_step_unb_var2 (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_blf_var3 (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_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_l_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_l_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_l_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_l_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_l_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 t, 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_t, int inc_t, 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_t, int inc_t, 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_t, int inc_t, 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_t, int inc_t, dcomplex *buff_u, int inc_u, dcomplex *buff_w, int inc_w) |
| FLA_Error FLA_Fused_Her2_Ax_l_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(), BLIS_CONJUGATE, F77_caxpy(), FLA_ZERO, scomplex::imag, and scomplex::real.
Referenced by FLA_Fused_Her2_Ax_l_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_l_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_daxpyv2bdotaxpy(), bli_dsetv(), and FLA_ZERO.
Referenced by FLA_Fused_Her2_Ax_l_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_dmult3( alpha11, chi1, &temp );
// bli_dadd3( &temp, omega1, omega1 );
*omega1 += *alpha11 * *chi1;
bli_daxpyv2bdotaxpy( m_ahead,
&minus_conj_zeta1,
u2, inc_u,
&minus_conj_upsilon1,
z2, inc_z,
a21, rs_A,
x2, inc_x,
chi1,
&temp,
w2, inc_w );
// bli_dadd3( &temp, omega1, omega1 );
*omega1 += temp;
/*------------------------------------------------------------*/
}
return FLA_SUCCESS;
}
| FLA_Error FLA_Fused_Her2_Ax_l_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_l_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_l_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_l_opc_var1(), FLA_Fused_Her2_Ax_l_opd_var1(), FLA_Fused_Her2_Ax_l_ops_var1(), FLA_Fused_Her2_Ax_l_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_l_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_l_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_l_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_l_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_l_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_z0(), bli_zaxpyv2b(), bli_zdotaxpy(), bli_zsetv(), dcomplex::imag, and dcomplex::real.
Referenced by FLA_Fused_Her2_Ax_l_opt_var1(), and FLA_Tridiag_UT_l_step_ofz_var2().
{
dcomplex zero = bli_z0();
int i;
bli_zsetv( m_A,
&zero,
buff_w, inc_w );
for ( i = 0; i < m_A; ++i )
{
dcomplex* restrict alpha11 = buff_A + (i )*cs_A + (i )*rs_A;
dcomplex* restrict a21 = buff_A + (i )*cs_A + (i+1)*rs_A;
dcomplex* restrict upsilon1 = buff_u + (i )*inc_u;
dcomplex* restrict u2 = buff_u + (i+1)*inc_u;
dcomplex* restrict zeta1 = buff_z + (i )*inc_z;
dcomplex* restrict z2 = buff_z + (i+1)*inc_z;
dcomplex* restrict chi1 = buff_x + (i )*inc_x;
dcomplex* restrict x2 = buff_x + (i+1)*inc_x;
dcomplex* restrict omega1 = buff_w + (i )*inc_w;
dcomplex* restrict w2 = buff_w + (i+1)*inc_w;
//dcomplex* restrict beta = buff_beta;
dcomplex minus_conj_upsilon1;
dcomplex minus_conj_zeta1;
dcomplex temp;
dcomplex ze1;
dcomplex up1;
dcomplex a11;
dcomplex om1;
dcomplex ch1;
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;
ze1 = *zeta1;
up1 = *upsilon1;
a11 = *alpha11;
om1 = *omega1;
ch1 = *chi1;
//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;
a11.real -= ze1.real * up1.real - -ze1.imag * up1.imag +
up1.real * ze1.real - -up1.imag * ze1.imag;
a11.imag -= ze1.real * up1.imag + -ze1.imag * up1.real +
up1.real * ze1.imag + -up1.imag * ze1.real;
// 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;
om1.real += a11.real * ch1.real - a11.imag * ch1.imag;
om1.imag += a11.imag * ch1.real + a11.real * ch1.imag;
*alpha11 = a11;
*omega1 = om1;
/*
bli_zaxpyv2bdotaxpy( m_ahead,
&minus_conj_zeta1,
u2, inc_u,
&minus_conj_upsilon1,
z2, inc_z,
a21, rs_A,
x2, inc_x,
chi1,
&temp,
w2, inc_w );
*/
bli_zaxpyv2b( m_ahead,
&minus_conj_zeta1,
&minus_conj_upsilon1,
u2, inc_u,
z2, inc_z,
a21, rs_A );
bli_zdotaxpy( m_ahead,
a21, rs_A,
x2, inc_x,
chi1,
&temp,
w2, inc_w );
// bli_zadd3( &temp, omega1, omega1 );
omega1->real += temp.real;
omega1->imag += temp.imag;
/*------------------------------------------------------------*/
}
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_t, | ||
| int | inc_t, | ||
| scomplex * | buff_u, | ||
| int | inc_u, | ||
| scomplex * | buff_w, | ||
| int | inc_w | ||
| ) |
References bli_cdot(), BLIS_CONJUGATE, 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* tau1 = buff_t + (i )*inc_t;
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 );
*tau1 = 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_t, | ||
| int | inc_t, | ||
| double * | buff_u, | ||
| int | inc_u, | ||
| double * | buff_w, | ||
| int | inc_w | ||
| ) |
References bli_d0(), bli_daxpyv(), bli_daxpyv2b(), bli_ddot(), bli_ddotaxpy(), bli_ddotsv2(), bli_ddotv2axpyv2b(), BLIS_CONJUGATE, and BLIS_NO_CONJUGATE.
Referenced by FLA_Fused_UZhu_ZUhu_opt_var1(), and FLA_Tridiag_UT_l_step_ofd_var3().
{
double zero = bli_d0();
int n_run = n_U / 2;
int n_left = n_U % 2;
int step_u = 2*cs_U;
int step_z = 2*cs_Z;
int step_tau = 2*inc_t;
int i;
double* u = buff_u;
double* w = buff_w;
//double* delta = buff_delta;
double* u1;
double* u2;
double* u3;
double* z1;
double* z2;
double* z3;
double* tau1;
double* tau2;
double* tau3;
u1 = buff_U;
u2 = buff_U + cs_U;
u3 = buff_U + 2*cs_U;
z1 = buff_Z;
z2 = buff_Z + cs_Z;
z3 = buff_Z + 2*cs_Z;
tau1 = buff_t;
tau2 = buff_t + inc_t;
tau3 = buff_t + 2*inc_t;
for ( i = 0; i < n_run; ++i )
{
double rho_z1u;
double rho_z2u;
//double rho_z3u;
double rho_u1u;
double rho_u2u;
//double rho_u3u;
/*------------------------------------------------------------*/
/*
bli_ddotsv3( BLIS_CONJUGATE,
m_U,
z1, rs_Z,
z2, rs_Z,
z3, rs_Z,
u, inc_u,
&zero,
&rho_z1u,
&rho_z2u,
&rho_z3u );
bli_dneg1( &rho_z1u );
bli_dneg1( &rho_z2u );
bli_dneg1( &rho_z3u );
bli_ddotv2axpyv2b( m_U,
u1, rs_U,
u2, rs_U,
u, inc_u,
&rho_z1u,
&rho_z2u,
&rho_u1u,
&rho_u2u,
w, inc_w );
bli_ddotaxpy( m_U,
u3, rs_U,
u, inc_u,
&rho_z3u,
&rho_u3u,
w, inc_w );
*tau1 = rho_u1u;
*tau2 = rho_u2u;
*tau3 = rho_u3u;
bli_dneg1( &rho_u1u );
bli_dneg1( &rho_u2u );
bli_dneg1( &rho_u3u );
bli_daxpyv3b( m_U,
&rho_u1u,
&rho_u2u,
&rho_u3u,
z1, rs_Z,
z2, rs_Z,
z3, rs_Z,
w, inc_w );
*/
bli_ddotsv2( BLIS_CONJUGATE,
m_U,
z1, rs_Z,
z2, rs_Z,
u, inc_u,
&zero,
&rho_z1u,
&rho_z2u );
bli_dneg1( &rho_z1u );
bli_dneg1( &rho_z2u );
bli_ddotv2axpyv2b( m_U,
u1, rs_U,
u2, rs_U,
u, inc_u,
&rho_z1u,
&rho_z2u,
&rho_u1u,
&rho_u2u,
w, inc_w );
*tau1 = rho_u1u;
*tau2 = rho_u2u;
bli_dneg1( &rho_u1u );
bli_dneg1( &rho_u2u );
bli_daxpyv2b( m_U,
&rho_u1u,
&rho_u2u,
z1, rs_Z,
z2, rs_Z,
w, inc_w );
/*------------------------------------------------------------*/
u1 += step_u;
u2 += step_u;
u3 += step_u;
z1 += step_z;
z2 += step_z;
z3 += step_z;
tau1 += step_tau;
tau2 += step_tau;
tau3 += step_tau;
}
if ( n_left > 0 )
{
for ( i = 0; i < n_left; ++i )
{
double rho_z1u;
double rho_u1u;
bli_ddot( BLIS_CONJUGATE,
m_U,
z1, rs_Z,
u, inc_u,
&rho_z1u );
bli_dneg1( &rho_z1u );
bli_ddotaxpy( m_U,
u1, rs_U,
u, inc_u,
&rho_z1u,
&rho_u1u,
w, inc_w );
*tau1 = rho_u1u;
bli_dneg1( &rho_u1u );
bli_daxpyv( BLIS_NO_CONJUGATE,
m_U,
&rho_u1u,
z1, rs_Z,
w, inc_w );
u1 += cs_U;
z1 += cs_Z;
tau1 += inc_t;
}
}
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_t, | ||
| int | inc_t, | ||
| 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* tau1 = buff_t + (i )*inc_t;
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 );
*tau1 = beta;
// 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 | t, | ||
| 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' u ) );
t = U' u;
*/
FLA_Datatype datatype;
int m_U, n_U;
int rs_U, cs_U;
int rs_Z, cs_Z;
int inc_u, inc_w, inc_t;
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 );
inc_t = FLA_Obj_vector_inc( t );
switch ( datatype )
{
case FLA_FLOAT:
{
float* buff_U = FLA_FLOAT_PTR( U );
float* buff_Z = FLA_FLOAT_PTR( Z );
float* buff_t = FLA_FLOAT_PTR( t );
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_t, inc_t,
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_t = FLA_DOUBLE_PTR( t );
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_t, inc_t,
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_t = FLA_COMPLEX_PTR( t );
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_t, inc_t,
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_t = FLA_DOUBLE_COMPLEX_PTR( t );
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_t, inc_t,
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_t, | ||
| int | inc_t, | ||
| dcomplex * | buff_u, | ||
| int | inc_u, | ||
| dcomplex * | buff_w, | ||
| int | inc_w | ||
| ) |
References bli_zaxpyv(), bli_zdot(), bli_zdotaxpy(), BLIS_CONJUGATE, and BLIS_NO_CONJUGATE.
Referenced by FLA_Fused_UZhu_ZUhu_opt_var1(), and FLA_Tridiag_UT_l_step_ofz_var3().
{
//dcomplex zero = bli_z0();
int n_run = n_U / 1;
int n_left = n_U % 1;
int step_u = 1*cs_U;
int step_z = 1*cs_Z;
int step_tau = 1*inc_t;
int i;
dcomplex* u = buff_u;
dcomplex* w = buff_w;
//dcomplex* delta = buff_delta;
dcomplex* u1;
dcomplex* u2;
dcomplex* z1;
dcomplex* z2;
dcomplex* tau1;
dcomplex* tau2;
u1 = buff_U;
u2 = buff_U + cs_U;
z1 = buff_Z;
z2 = buff_Z + cs_Z;
tau1 = buff_t;
tau2 = buff_t + inc_t;
for ( i = 0; i < n_run; ++i )
{
dcomplex rho_z1u;
//dcomplex rho_z2u;
dcomplex rho_u1u;
//dcomplex rho_u2u;
/*------------------------------------------------------------*/
/*
Effective computation:
w = w + delta * ( U ( Z' u ) + Z ( U' u ) );
*/
/*
bli_zdotsv2( BLIS_CONJUGATE,
m_U,
z1, rs_Z,
u1, rs_U,
u, inc_u,
&zero,
&rho_z1u,
&rho_u1u );
*tau1 = rho_u1u;
//bli_zscals( delta, &rho_z1u );
//bli_zscals( delta, &rho_u1u );
bli_zneg1( &rho_z1u );
bli_zneg1( &rho_u1u );
bli_zaxpyv2b( m_U,
&rho_z1u,
&rho_u1u,
u1, rs_U,
z1, rs_Z,
w, inc_w );
*/
/*
bli_zdotsv2( BLIS_CONJUGATE,
m_U,
z1, rs_Z,
z2, rs_Z,
u, inc_u,
&zero,
&rho_z1u,
&rho_z2u );
bli_zneg1( &rho_z1u );
bli_zneg1( &rho_z2u );
bli_zdotv2axpyv2b( m_U,
u1, rs_U,
u2, rs_U,
u, inc_u,
&rho_z1u,
&rho_z2u,
&rho_u1u,
&rho_u2u,
w, inc_w );
*tau1 = rho_u1u;
*tau2 = rho_u2u;
bli_zneg1( &rho_u1u );
bli_zneg1( &rho_u2u );
bli_zaxpyv2b( m_U,
&rho_u1u,
&rho_u2u,
z1, rs_Z,
z2, rs_Z,
w, inc_w );
*/
bli_zdot( BLIS_CONJUGATE,
m_U,
z1, rs_Z,
u, inc_u,
&rho_z1u );
bli_zneg1( &rho_z1u );
bli_zdotaxpy( m_U,
u1, rs_U,
u, inc_u,
&rho_z1u,
&rho_u1u,
w, inc_w );
*tau1 = rho_u1u;
bli_zneg1( &rho_u1u );
bli_zaxpyv( BLIS_NO_CONJUGATE,
m_U,
&rho_u1u,
z1, rs_Z,
w, inc_w );
/*------------------------------------------------------------*/
u1 += step_u;
u2 += step_u;
z1 += step_z;
z2 += step_z;
tau1 += step_tau;
tau2 += step_tau;
}
if ( n_left == 1 )
{
dcomplex rho_z1u;
dcomplex rho_u1u;
bli_zdot( BLIS_CONJUGATE,
m_U,
z1, rs_Z,
u, inc_u,
&rho_z1u );
bli_zneg1( &rho_z1u );
bli_zdotaxpy( m_U,
u1, rs_U,
u, inc_u,
&rho_z1u,
&rho_u1u,
w, inc_w );
*tau1 = rho_u1u;
bli_zneg1( &rho_u1u );
bli_zaxpyv( BLIS_NO_CONJUGATE,
m_U,
&rho_u1u,
z1, rs_Z,
w, inc_w );
}
return FLA_SUCCESS;
}
| FLA_Error FLA_Tridiag_UT_l_blf_var2 | ( | FLA_Obj | A, |
| FLA_Obj | T | ||
| ) |
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_ofu_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;
}
| FLA_Error FLA_Tridiag_UT_l_blf_var3 | ( | FLA_Obj | A, |
| FLA_Obj | T | ||
| ) |
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_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_Set(), FLA_Triangularize(), FLA_Tridiag_UT_l_step_ofu_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_Set( 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;
}
| FLA_Error FLA_Tridiag_UT_l_blk_var1 | ( | FLA_Obj | A, |
| FLA_Obj | T | ||
| ) |
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;
}
| FLA_Error FLA_Tridiag_UT_l_blk_var2 | ( | FLA_Obj | A, |
| FLA_Obj | T | ||
| ) |
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;
}
| FLA_Error FLA_Tridiag_UT_l_blk_var3 | ( | FLA_Obj | A, |
| FLA_Obj | T | ||
| ) |
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_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_Set(), 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_Set( 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;
}
| FLA_Error FLA_Tridiag_UT_l_ofu_var1 | ( | FLA_Obj | A, |
| FLA_Obj | T | ||
| ) |
| FLA_Error FLA_Tridiag_UT_l_ofu_var2 | ( | FLA_Obj | A, |
| FLA_Obj | T | ||
| ) |
References FLA_Tridiag_UT_l_step_ofu_var2().
{
return FLA_Tridiag_UT_l_step_ofu_var2( A, T );
}
| FLA_Error FLA_Tridiag_UT_l_ofu_var3 | ( | FLA_Obj | A, |
| FLA_Obj | 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;
}
| FLA_Error FLA_Tridiag_UT_l_opt_var1 | ( | FLA_Obj | A, |
| FLA_Obj | T | ||
| ) |
References FLA_Tridiag_UT_l_step_opt_var1().
Referenced by FLA_Tridiag_UT_l().
{
return FLA_Tridiag_UT_l_step_opt_var1( A, T );
}
| FLA_Error FLA_Tridiag_UT_l_opt_var2 | ( | FLA_Obj | A, |
| FLA_Obj | 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 );
}
| FLA_Error FLA_Tridiag_UT_l_opt_var3 | ( | FLA_Obj | A, |
| FLA_Obj | 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(), BLIS_CONJ_TRANSPOSE, BLIS_CONJUGATE, BLIS_LOWER_TRIANGULAR, BLIS_NO_CONJUGATE, BLIS_NO_TRANSPOSE, FLA_free(), FLA_Fused_Her2_Ax_l_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_Set( 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_Set( 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_l_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(), BLIS_CONJUGATE, BLIS_LOWER_TRIANGULAR, BLIS_NO_CONJUGATE, BLIS_NO_TRANSPOSE, 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_Set( 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_Set( 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_Set( 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_Set( 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_Copy( d01, t01 );
FLA_Fused_UZhu_ZUhu_opc_var1( m_ahead,
n_behind,
buff_m1,
A20, rs_A, cs_A,
Z20, rs_Z, cs_Z,
t01, rs_T,
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_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(), BLIS_CONJ_TRANSPOSE, BLIS_CONJUGATE, BLIS_LOWER_TRIANGULAR, BLIS_NO_CONJUGATE, FLA_free(), FLA_Fused_Her2_Ax_l_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_Set( 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_Set( 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_l_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(), BLIS_CONJUGATE, BLIS_LOWER_TRIANGULAR, BLIS_NO_CONJUGATE, BLIS_NO_TRANSPOSE, 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_Set( 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_Set( 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_Set( 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_Set( 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_Copy( d01, t01 );
FLA_Fused_UZhu_ZUhu_opd_var1( m_ahead,
n_behind,
buff_m1,
A20, rs_A, cs_A,
Z20, rs_Z, cs_Z,
t01, rs_T,
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_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(), BLIS_CONJ_TRANSPOSE, BLIS_CONJUGATE, BLIS_LOWER_TRIANGULAR, BLIS_NO_CONJUGATE, FLA_free(), FLA_Fused_Her2_Ax_l_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_Set( 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_Set( 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_l_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(), BLIS_CONJUGATE, BLIS_LOWER_TRIANGULAR, BLIS_NO_CONJUGATE, BLIS_NO_TRANSPOSE, 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_Set( 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_Set( 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_Set( 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_Set( 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_Copy( d01, t01 );
FLA_Fused_UZhu_ZUhu_ops_var1( m_ahead,
n_behind,
buff_m1,
A20, rs_A, cs_A,
Z20, rs_Z, cs_Z,
t01, rs_T,
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_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_blf_var2(), and 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;
}
| FLA_Error FLA_Tridiag_UT_l_step_ofu_var3 | ( | FLA_Obj | A, |
| FLA_Obj | Z, | ||
| FLA_Obj | T | ||
| ) |
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_blf_var3(), and 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(), BLIS_CONJ_TRANSPOSE, BLIS_CONJUGATE, BLIS_LOWER_TRIANGULAR, BLIS_NO_CONJUGATE, BLIS_NO_TRANSPOSE, FLA_free(), FLA_Fused_Her2_Ax_l_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_Set( 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_Set( 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_l_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(), BLIS_CONJUGATE, BLIS_LOWER_TRIANGULAR, BLIS_NO_CONJUGATE, BLIS_NO_TRANSPOSE, 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_Set( 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_Set( 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_Set( 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_Set( 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_Copy( d01, t01 );
FLA_Fused_UZhu_ZUhu_opz_var1( m_ahead,
n_behind,
buff_m1,
A20, rs_A, cs_A,
Z20, rs_Z, cs_Z,
t01, rs_T,
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_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(), BLIS_CONJ_TRANSPOSE, BLIS_CONJUGATE, BLIS_LOWER_TRIANGULAR, BLIS_NO_CONJUGATE, BLIS_NO_TRANSPOSE, 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_Set( 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_Set( 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(), BLIS_CONJ_TRANSPOSE, BLIS_CONJUGATE, BLIS_LOWER_TRIANGULAR, BLIS_NO_CONJUGATE, BLIS_NO_TRANSPOSE, 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_Set( 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_Set( 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_ccopyv(), bli_cdot(), bli_cgemv(), bli_chemv(), bli_cscalv(), bli_csetm(), BLIS_CONJ_TRANSPOSE, BLIS_CONJUGATE, BLIS_LOWER_TRIANGULAR, BLIS_NO_CONJUGATE, BLIS_NO_TRANSPOSE, 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_Set( 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_Set( 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_Set( 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_Set( 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_Copy( d01, t01 );
bli_ccopyv( BLIS_NO_CONJUGATE,
n_behind,
d01, inc_d,
t01, rs_T );
// 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_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(), BLIS_CONJ_TRANSPOSE, BLIS_CONJUGATE, BLIS_LOWER_TRIANGULAR, BLIS_NO_CONJUGATE, 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_Set( 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_Set( 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(), BLIS_CONJ_TRANSPOSE, BLIS_CONJUGATE, BLIS_LOWER_TRIANGULAR, BLIS_NO_CONJUGATE, 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_Set( 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_Set( 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_dcopyv(), bli_ddot(), bli_dgemv(), bli_dscalv(), bli_dsetm(), bli_dsymv(), BLIS_CONJ_TRANSPOSE, BLIS_CONJUGATE, BLIS_LOWER_TRIANGULAR, BLIS_NO_CONJUGATE, BLIS_NO_TRANSPOSE, 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_Set( 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_Set( 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_Set( 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_Set( 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_Copy( d01, t01 );
bli_dcopyv( BLIS_NO_CONJUGATE,
n_behind,
d01, inc_d,
t01, rs_T );
// 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_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(), BLIS_CONJ_TRANSPOSE, BLIS_CONJUGATE, BLIS_LOWER_TRIANGULAR, BLIS_NO_CONJUGATE, 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_Set( 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_Set( 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(), BLIS_CONJ_TRANSPOSE, BLIS_CONJUGATE, BLIS_LOWER_TRIANGULAR, BLIS_NO_CONJUGATE, 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_Set( 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_Set( 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_scopyv(), bli_sdot(), bli_sgemv(), bli_sscalv(), bli_ssetm(), bli_ssymv(), BLIS_CONJ_TRANSPOSE, BLIS_CONJUGATE, BLIS_LOWER_TRIANGULAR, BLIS_NO_CONJUGATE, BLIS_NO_TRANSPOSE, 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_Set( 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_Set( 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_Set( 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_Set( 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_Copy( d01, t01 );
bli_scopyv( BLIS_NO_CONJUGATE,
n_behind,
d01, inc_d,
t01, rs_T );
// 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_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;
}
| FLA_Error FLA_Tridiag_UT_l_step_opt_var3 | ( | FLA_Obj | A, |
| FLA_Obj | Z, | ||
| FLA_Obj | T | ||
| ) |
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(), BLIS_CONJ_TRANSPOSE, BLIS_CONJUGATE, BLIS_LOWER_TRIANGULAR, BLIS_NO_CONJUGATE, BLIS_NO_TRANSPOSE, 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_Set( 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_Set( 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(), BLIS_CONJ_TRANSPOSE, BLIS_CONJUGATE, BLIS_LOWER_TRIANGULAR, BLIS_NO_CONJUGATE, BLIS_NO_TRANSPOSE, 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_Set( 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_Set( 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_zcopyv(), bli_zdot(), bli_zgemv(), bli_zhemv(), bli_zscalv(), bli_zsetm(), BLIS_CONJ_TRANSPOSE, BLIS_CONJUGATE, BLIS_LOWER_TRIANGULAR, BLIS_NO_CONJUGATE, BLIS_NO_TRANSPOSE, 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_Set( 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_Set( 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_Set( 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_Set( 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_Copy( d01, t01 );
bli_zcopyv( BLIS_NO_CONJUGATE,
n_behind,
d01, inc_d,
t01, rs_T );
// 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_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_ONE, FLA_Part_2x1(), FLA_Part_2x2(), FLA_Repart_2x1_to_3x1(), FLA_Repart_2x2_to_3x3(), FLA_Scal(), FLA_Set(), 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_Set( 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_Set( 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_ONE, FLA_Part_2x1(), FLA_Part_2x2(), FLA_Repart_2x1_to_3x1(), FLA_Repart_2x2_to_3x3(), FLA_Scal(), FLA_Set(), 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_Set( 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_Set( 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;
}
| FLA_Error FLA_Tridiag_UT_l_step_unb_var3 | ( | FLA_Obj | A, |
| FLA_Obj | Z, | ||
| FLA_Obj | T | ||
| ) |
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_ONE, FLA_Part_1x2(), FLA_Part_2x1(), FLA_Part_2x2(), FLA_Repart_2x1_to_3x1(), FLA_Repart_2x2_to_3x3(), FLA_Scal(), FLA_Set(), 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_Set( 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_Set( 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_Set( 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_Set( FLA_ONE, a21_t );
// z21 = A22 * u21;
FLA_Hemv( FLA_LOWER_TRIANGULAR, FLA_ONE, A22, a21, FLA_ZERO, z21 );
// z21 = z21 - U20 * ( Z20' * u21 ) - Z20 * ( U20' * u21 );
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 );
// t01 = U20' * u21;
FLA_Copy( d01, t01 );
// 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 );
// 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;
}
| FLA_Error FLA_Tridiag_UT_l_unb_var1 | ( | FLA_Obj | A, |
| FLA_Obj | T | ||
| ) |
References FLA_Tridiag_UT_l_step_unb_var1().
Referenced by FLA_Tridiag_UT_l().
{
return FLA_Tridiag_UT_l_step_unb_var1( A, T );
}
| FLA_Error FLA_Tridiag_UT_l_unb_var2 | ( | FLA_Obj | A, |
| FLA_Obj | 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 );
}
| FLA_Error FLA_Tridiag_UT_l_unb_var3 | ( | FLA_Obj | A, |
| FLA_Obj | 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;
}
1.7.6.1