|
libflame
revision_anchor
|
Functions | |
| FLA_Error | FLA_LU_nopiv_opt_var2 (FLA_Obj A) |
| FLA_Error | FLA_LU_nopiv_ops_var2 (int m_A, int n_A, float *buff_A, int rs_A, int cs_A) |
| FLA_Error | FLA_LU_nopiv_opd_var2 (int m_A, int n_A, double *buff_A, int rs_A, int cs_A) |
| FLA_Error | FLA_LU_nopiv_opc_var2 (int m_A, int n_A, scomplex *buff_A, int rs_A, int cs_A) |
| FLA_Error | FLA_LU_nopiv_opz_var2 (int m_A, int n_A, dcomplex *buff_A, int rs_A, int cs_A) |
| FLA_Error FLA_LU_nopiv_opc_var2 | ( | int | m_A, |
| int | n_A, | ||
| scomplex * | buff_A, | ||
| int | rs_A, | ||
| int | cs_A | ||
| ) |
References bli_cdots(), bli_cgemv(), bli_ctrsm(), bli_ctrsv(), BLIS_NO_CONJUGATE, BLIS_NO_TRANSPOSE, BLIS_NONUNIT_DIAG, BLIS_RIGHT, BLIS_TRANSPOSE, BLIS_UPPER_TRIANGULAR, FLA_MINUS_ONE, and FLA_ONE.
Referenced by FLA_LU_nopiv_opt_var2().
{
scomplex* buff_1 = FLA_COMPLEX_PTR( FLA_ONE );
scomplex* buff_m1 = FLA_COMPLEX_PTR( FLA_MINUS_ONE );
int min_m_n = min( m_A, n_A );
int i;
for ( i = 0; i < min_m_n; ++i )
{
scomplex* A00 = buff_A + (0 )*cs_A + (0 )*rs_A;
scomplex* a10t = buff_A + (0 )*cs_A + (i )*rs_A;
scomplex* a01 = buff_A + (i )*cs_A + (0 )*rs_A;
scomplex* alpha11 = buff_A + (i )*cs_A + (i )*rs_A;
scomplex* A02 = buff_A + (i+1)*cs_A + (0 )*rs_A;
scomplex* a12t = buff_A + (i+1)*cs_A + (i )*rs_A;
int n_ahead = n_A - i - 1;
int mn_behind = i;
/*------------------------------------------------------------*/
// FLA_Trsv_external( FLA_UPPER_TRIANGULAR, FLA_TRANSPOSE, FLA_NONUNIT_DIAG, A00, a10t );
bli_ctrsv( BLIS_UPPER_TRIANGULAR,
BLIS_TRANSPOSE,
BLIS_NONUNIT_DIAG,
mn_behind,
A00, rs_A, cs_A,
a10t, cs_A );
// FLA_Dots_external( FLA_MINUS_ONE, a10t, a01, FLA_ONE, alpha11 );
bli_cdots( BLIS_NO_CONJUGATE,
mn_behind,
buff_m1,
a10t, cs_A,
a01, rs_A,
buff_1,
alpha11 );
// FLA_Gemv_external( FLA_TRANSPOSE, FLA_MINUS_ONE, A02, a10t, FLA_ONE, a12t );
bli_cgemv( BLIS_TRANSPOSE,
BLIS_NO_CONJUGATE,
mn_behind,
n_ahead,
buff_m1,
A02, rs_A, cs_A,
a10t, cs_A,
buff_1,
a12t, cs_A );
/*------------------------------------------------------------*/
}
if ( m_A > n_A )
{
scomplex* ATL = buff_A;
scomplex* ABL = buff_A + n_A*rs_A;
// FLA_Trsm_external( FLA_RIGHT, FLA_UPPER_TRIANGULAR,
// FLA_NO_TRANSPOSE, FLA_NONUNIT_DIAG,
// FLA_ONE, ATL, ABL );
bli_ctrsm( BLIS_RIGHT,
BLIS_UPPER_TRIANGULAR,
BLIS_NO_TRANSPOSE,
BLIS_NONUNIT_DIAG,
m_A - n_A,
n_A,
buff_1,
ATL, rs_A, cs_A,
ABL, rs_A, cs_A );
}
return FLA_SUCCESS;
}
| FLA_Error FLA_LU_nopiv_opd_var2 | ( | int | m_A, |
| int | n_A, | ||
| double * | buff_A, | ||
| int | rs_A, | ||
| int | cs_A | ||
| ) |
References bli_ddots(), bli_dgemv(), bli_dtrsm(), bli_dtrsv(), BLIS_NO_CONJUGATE, BLIS_NO_TRANSPOSE, BLIS_NONUNIT_DIAG, BLIS_RIGHT, BLIS_TRANSPOSE, BLIS_UPPER_TRIANGULAR, FLA_MINUS_ONE, and FLA_ONE.
Referenced by FLA_LU_nopiv_opt_var2().
{
double* buff_1 = FLA_DOUBLE_PTR( FLA_ONE );
double* buff_m1 = FLA_DOUBLE_PTR( FLA_MINUS_ONE );
int min_m_n = min( m_A, n_A );
int i;
for ( i = 0; i < min_m_n; ++i )
{
double* A00 = buff_A + (0 )*cs_A + (0 )*rs_A;
double* a10t = buff_A + (0 )*cs_A + (i )*rs_A;
double* a01 = buff_A + (i )*cs_A + (0 )*rs_A;
double* alpha11 = buff_A + (i )*cs_A + (i )*rs_A;
double* A02 = buff_A + (i+1)*cs_A + (0 )*rs_A;
double* a12t = buff_A + (i+1)*cs_A + (i )*rs_A;
int n_ahead = n_A - i - 1;
int mn_behind = i;
/*------------------------------------------------------------*/
// FLA_Trsv_external( FLA_UPPER_TRIANGULAR, FLA_TRANSPOSE, FLA_NONUNIT_DIAG, A00, a10t );
bli_dtrsv( BLIS_UPPER_TRIANGULAR,
BLIS_TRANSPOSE,
BLIS_NONUNIT_DIAG,
mn_behind,
A00, rs_A, cs_A,
a10t, cs_A );
// FLA_Dots_external( FLA_MINUS_ONE, a10t, a01, FLA_ONE, alpha11 );
bli_ddots( BLIS_NO_CONJUGATE,
mn_behind,
buff_m1,
a10t, cs_A,
a01, rs_A,
buff_1,
alpha11 );
// FLA_Gemv_external( FLA_TRANSPOSE, FLA_MINUS_ONE, A02, a10t, FLA_ONE, a12t );
bli_dgemv( BLIS_TRANSPOSE,
BLIS_NO_CONJUGATE,
mn_behind,
n_ahead,
buff_m1,
A02, rs_A, cs_A,
a10t, cs_A,
buff_1,
a12t, cs_A );
/*------------------------------------------------------------*/
}
if ( m_A > n_A )
{
double* ATL = buff_A;
double* ABL = buff_A + n_A*rs_A;
// FLA_Trsm_external( FLA_RIGHT, FLA_UPPER_TRIANGULAR,
// FLA_NO_TRANSPOSE, FLA_NONUNIT_DIAG,
// FLA_ONE, ATL, ABL );
bli_dtrsm( BLIS_RIGHT,
BLIS_UPPER_TRIANGULAR,
BLIS_NO_TRANSPOSE,
BLIS_NONUNIT_DIAG,
m_A - n_A,
n_A,
buff_1,
ATL, rs_A, cs_A,
ABL, rs_A, cs_A );
}
return FLA_SUCCESS;
}
| FLA_Error FLA_LU_nopiv_ops_var2 | ( | int | m_A, |
| int | n_A, | ||
| float * | buff_A, | ||
| int | rs_A, | ||
| int | cs_A | ||
| ) |
References bli_sdots(), bli_sgemv(), bli_strsm(), bli_strsv(), BLIS_NO_CONJUGATE, BLIS_NO_TRANSPOSE, BLIS_NONUNIT_DIAG, BLIS_RIGHT, BLIS_TRANSPOSE, BLIS_UPPER_TRIANGULAR, FLA_MINUS_ONE, and FLA_ONE.
Referenced by FLA_LU_nopiv_opt_var2().
{
float* buff_1 = FLA_FLOAT_PTR( FLA_ONE );
float* buff_m1 = FLA_FLOAT_PTR( FLA_MINUS_ONE );
int min_m_n = min( m_A, n_A );
int i;
for ( i = 0; i < min_m_n; ++i )
{
float* A00 = buff_A + (0 )*cs_A + (0 )*rs_A;
float* a10t = buff_A + (0 )*cs_A + (i )*rs_A;
float* a01 = buff_A + (i )*cs_A + (0 )*rs_A;
float* alpha11 = buff_A + (i )*cs_A + (i )*rs_A;
float* A02 = buff_A + (i+1)*cs_A + (0 )*rs_A;
float* a12t = buff_A + (i+1)*cs_A + (i )*rs_A;
int n_ahead = n_A - i - 1;
int mn_behind = i;
/*------------------------------------------------------------*/
// FLA_Trsv_external( FLA_UPPER_TRIANGULAR, FLA_TRANSPOSE, FLA_NONUNIT_DIAG, A00, a10t );
bli_strsv( BLIS_UPPER_TRIANGULAR,
BLIS_TRANSPOSE,
BLIS_NONUNIT_DIAG,
mn_behind,
A00, rs_A, cs_A,
a10t, cs_A );
// FLA_Dots_external( FLA_MINUS_ONE, a10t, a01, FLA_ONE, alpha11 );
bli_sdots( BLIS_NO_CONJUGATE,
mn_behind,
buff_m1,
a10t, cs_A,
a01, rs_A,
buff_1,
alpha11 );
// FLA_Gemv_external( FLA_TRANSPOSE, FLA_MINUS_ONE, A02, a10t, FLA_ONE, a12t );
bli_sgemv( BLIS_TRANSPOSE,
BLIS_NO_CONJUGATE,
mn_behind,
n_ahead,
buff_m1,
A02, rs_A, cs_A,
a10t, cs_A,
buff_1,
a12t, cs_A );
/*------------------------------------------------------------*/
}
if ( m_A > n_A )
{
float* ATL = buff_A;
float* ABL = buff_A + n_A*rs_A;
// FLA_Trsm_external( FLA_RIGHT, FLA_UPPER_TRIANGULAR,
// FLA_NO_TRANSPOSE, FLA_NONUNIT_DIAG,
// FLA_ONE, ATL, ABL );
bli_strsm( BLIS_RIGHT,
BLIS_UPPER_TRIANGULAR,
BLIS_NO_TRANSPOSE,
BLIS_NONUNIT_DIAG,
m_A - n_A,
n_A,
buff_1,
ATL, rs_A, cs_A,
ABL, rs_A, cs_A );
}
return FLA_SUCCESS;
}
References FLA_LU_nopiv_opc_var2(), FLA_LU_nopiv_opd_var2(), FLA_LU_nopiv_ops_var2(), FLA_LU_nopiv_opz_var2(), FLA_Obj_col_stride(), FLA_Obj_datatype(), FLA_Obj_length(), FLA_Obj_row_stride(), and FLA_Obj_width().
Referenced by FLA_LU_nopiv_internal().
{
FLA_Datatype datatype;
int m_A, n_A;
int rs_A, cs_A;
datatype = FLA_Obj_datatype( A );
m_A = FLA_Obj_length( A );
n_A = FLA_Obj_width( A );
rs_A = FLA_Obj_row_stride( A );
cs_A = FLA_Obj_col_stride( A );
switch ( datatype )
{
case FLA_FLOAT:
{
float* buff_A = FLA_FLOAT_PTR( A );
FLA_LU_nopiv_ops_var2( m_A,
n_A,
buff_A, rs_A, cs_A );
break;
}
case FLA_DOUBLE:
{
double* buff_A = FLA_DOUBLE_PTR( A );
FLA_LU_nopiv_opd_var2( m_A,
n_A,
buff_A, rs_A, cs_A );
break;
}
case FLA_COMPLEX:
{
scomplex* buff_A = FLA_COMPLEX_PTR( A );
FLA_LU_nopiv_opc_var2( m_A,
n_A,
buff_A, rs_A, cs_A );
break;
}
case FLA_DOUBLE_COMPLEX:
{
dcomplex* buff_A = FLA_DOUBLE_COMPLEX_PTR( A );
FLA_LU_nopiv_opz_var2( m_A,
n_A,
buff_A, rs_A, cs_A );
break;
}
}
return FLA_SUCCESS;
}
| FLA_Error FLA_LU_nopiv_opz_var2 | ( | int | m_A, |
| int | n_A, | ||
| dcomplex * | buff_A, | ||
| int | rs_A, | ||
| int | cs_A | ||
| ) |
References bli_zdots(), bli_zgemv(), bli_ztrsm(), bli_ztrsv(), BLIS_NO_CONJUGATE, BLIS_NO_TRANSPOSE, BLIS_NONUNIT_DIAG, BLIS_RIGHT, BLIS_TRANSPOSE, BLIS_UPPER_TRIANGULAR, FLA_MINUS_ONE, and FLA_ONE.
Referenced by FLA_LU_nopiv_opt_var2().
{
dcomplex* buff_1 = FLA_DOUBLE_COMPLEX_PTR( FLA_ONE );
dcomplex* buff_m1 = FLA_DOUBLE_COMPLEX_PTR( FLA_MINUS_ONE );
int min_m_n = min( m_A, n_A );
int i;
for ( i = 0; i < min_m_n; ++i )
{
dcomplex* A00 = buff_A + (0 )*cs_A + (0 )*rs_A;
dcomplex* a10t = buff_A + (0 )*cs_A + (i )*rs_A;
dcomplex* a01 = buff_A + (i )*cs_A + (0 )*rs_A;
dcomplex* alpha11 = buff_A + (i )*cs_A + (i )*rs_A;
dcomplex* A02 = buff_A + (i+1)*cs_A + (0 )*rs_A;
dcomplex* a12t = buff_A + (i+1)*cs_A + (i )*rs_A;
int n_ahead = n_A - i - 1;
int mn_behind = i;
/*------------------------------------------------------------*/
// FLA_Trsv_external( FLA_UPPER_TRIANGULAR, FLA_TRANSPOSE, FLA_NONUNIT_DIAG, A00, a10t );
bli_ztrsv( BLIS_UPPER_TRIANGULAR,
BLIS_TRANSPOSE,
BLIS_NONUNIT_DIAG,
mn_behind,
A00, rs_A, cs_A,
a10t, cs_A );
// FLA_Dots_external( FLA_MINUS_ONE, a10t, a01, FLA_ONE, alpha11 );
bli_zdots( BLIS_NO_CONJUGATE,
mn_behind,
buff_m1,
a10t, cs_A,
a01, rs_A,
buff_1,
alpha11 );
// FLA_Gemv_external( FLA_TRANSPOSE, FLA_MINUS_ONE, A02, a10t, FLA_ONE, a12t );
bli_zgemv( BLIS_TRANSPOSE,
BLIS_NO_CONJUGATE,
mn_behind,
n_ahead,
buff_m1,
A02, rs_A, cs_A,
a10t, cs_A,
buff_1,
a12t, cs_A );
/*------------------------------------------------------------*/
}
if ( m_A > n_A )
{
dcomplex* ATL = buff_A;
dcomplex* ABL = buff_A + n_A*rs_A;
// FLA_Trsm_external( FLA_RIGHT, FLA_UPPER_TRIANGULAR,
// FLA_NO_TRANSPOSE, FLA_NONUNIT_DIAG,
// FLA_ONE, ATL, ABL );
bli_ztrsm( BLIS_RIGHT,
BLIS_UPPER_TRIANGULAR,
BLIS_NO_TRANSPOSE,
BLIS_NONUNIT_DIAG,
m_A - n_A,
n_A,
buff_1,
ATL, rs_A, cs_A,
ABL, rs_A, cs_A );
}
return FLA_SUCCESS;
}
1.7.6.1