libflame
12600
|
Functions | |
void | bl1_strmmsx (side1_t side, uplo1_t uplo, trans1_t trans, diag1_t diag, int m, int n, float *alpha, float *a, int a_rs, int a_cs, float *b, int b_rs, int b_cs, float *beta, float *c, int c_rs, int c_cs) |
void | bl1_dtrmmsx (side1_t side, uplo1_t uplo, trans1_t trans, diag1_t diag, int m, int n, double *alpha, double *a, int a_rs, int a_cs, double *b, int b_rs, int b_cs, double *beta, double *c, int c_rs, int c_cs) |
void | bl1_ctrmmsx (side1_t side, uplo1_t uplo, trans1_t trans, diag1_t diag, int m, int n, scomplex *alpha, scomplex *a, int a_rs, int a_cs, scomplex *b, int b_rs, int b_cs, scomplex *beta, scomplex *c, int c_rs, int c_cs) |
void | bl1_ztrmmsx (side1_t side, uplo1_t uplo, trans1_t trans, diag1_t diag, int m, int n, dcomplex *alpha, dcomplex *a, int a_rs, int a_cs, dcomplex *b, int b_rs, int b_cs, dcomplex *beta, dcomplex *c, int c_rs, int c_cs) |
void bl1_ctrmmsx | ( | side1_t | side, |
uplo1_t | uplo, | ||
trans1_t | trans, | ||
diag1_t | diag, | ||
int | m, | ||
int | n, | ||
scomplex * | alpha, | ||
scomplex * | a, | ||
int | a_rs, | ||
int | a_cs, | ||
scomplex * | b, | ||
int | b_rs, | ||
int | b_cs, | ||
scomplex * | beta, | ||
scomplex * | c, | ||
int | c_rs, | ||
int | c_cs | ||
) |
References bl1_c1(), bl1_callocm(), bl1_caxpymt(), bl1_ccopymt(), bl1_ccreate_contigm(), bl1_ccreate_contigmr(), bl1_cfree(), bl1_cfree_contigm(), bl1_cfree_saved_contigm(), bl1_cscalm(), bl1_ctrmm(), bl1_is_col_storage(), bl1_set_dim_with_side(), bl1_zero_dim2(), BLIS1_NO_CONJUGATE, and BLIS1_NO_TRANSPOSE.
Referenced by FLA_Trmmsx_external().
{ int m_save = m; int n_save = n; scomplex* a_save = a; scomplex* b_save = b; scomplex* c_save = c; int a_rs_save = a_rs; int a_cs_save = a_cs; int b_rs_save = b_rs; int b_cs_save = b_cs; int c_rs_save = c_rs; int c_cs_save = c_cs; scomplex one = bl1_c1(); scomplex* b_copy; int dim_a; int b_copy_rs, b_copy_cs; // Return early if possible. if ( bl1_zero_dim2( m, n ) ) return; // If necessary, allocate, initialize, and use a temporary contiguous // copy of each matrix rather than the original matrices. bl1_set_dim_with_side( side, m, n, &dim_a ); bl1_ccreate_contigmr( uplo, dim_a, dim_a, a_save, a_rs_save, a_cs_save, &a, &a_rs, &a_cs ); bl1_ccreate_contigm( m, n, b_save, b_rs_save, b_cs_save, &b, &b_rs, &b_cs ); bl1_ccreate_contigm( m, n, c_save, c_rs_save, c_cs_save, &c, &c_rs, &c_cs ); // Create a copy of B to use in the computation so the original matrix is // left untouched. b_copy = bl1_callocm( m, n ); // Match the strides of B_copy to that of B. if ( bl1_is_col_storage( b_rs, b_cs ) ) { b_copy_rs = 1; b_copy_cs = m; } else // if ( bl1_is_row_storage( b_rs, b_cs ) ) { b_copy_rs = n; b_copy_cs = 1; } // Copy the contents of B to B_copy. bl1_ccopymt( BLIS1_NO_TRANSPOSE, m, n, b, b_rs, b_cs, b_copy, b_copy_rs, b_copy_cs ); // Perform the operation on B_copy. bl1_ctrmm( side, uplo, trans, diag, m, n, alpha, a, a_rs, a_cs, b_copy, b_copy_rs, b_copy_cs ); // Scale C by beta. bl1_cscalm( BLIS1_NO_CONJUGATE, m, n, beta, c, c_rs, c_cs ); // Add B_copy into C. bl1_caxpymt( BLIS1_NO_TRANSPOSE, m, n, &one, b_copy, b_copy_rs, b_copy_cs, c, c_rs, c_cs ); // Free the copy of B. bl1_cfree( b_copy ); // Free any temporary contiguous matrices, copying the result back to // the original matrix. bl1_cfree_contigm( a_save, a_rs_save, a_cs_save, &a, &a_rs, &a_cs ); bl1_cfree_contigm( b_save, b_rs_save, b_cs_save, &b, &b_rs, &b_cs ); bl1_cfree_saved_contigm( m_save, n_save, c_save, c_rs_save, c_cs_save, &c, &c_rs, &c_cs ); }
void bl1_dtrmmsx | ( | side1_t | side, |
uplo1_t | uplo, | ||
trans1_t | trans, | ||
diag1_t | diag, | ||
int | m, | ||
int | n, | ||
double * | alpha, | ||
double * | a, | ||
int | a_rs, | ||
int | a_cs, | ||
double * | b, | ||
int | b_rs, | ||
int | b_cs, | ||
double * | beta, | ||
double * | c, | ||
int | c_rs, | ||
int | c_cs | ||
) |
References bl1_d1(), bl1_dallocm(), bl1_daxpymt(), bl1_dcopymt(), bl1_dcreate_contigm(), bl1_dcreate_contigmr(), bl1_dfree(), bl1_dfree_contigm(), bl1_dfree_saved_contigm(), bl1_dscalm(), bl1_dtrmm(), bl1_is_col_storage(), bl1_set_dim_with_side(), bl1_zero_dim2(), BLIS1_NO_CONJUGATE, and BLIS1_NO_TRANSPOSE.
Referenced by FLA_Trmmsx_external().
{ int m_save = m; int n_save = n; double* a_save = a; double* b_save = b; double* c_save = c; int a_rs_save = a_rs; int a_cs_save = a_cs; int b_rs_save = b_rs; int b_cs_save = b_cs; int c_rs_save = c_rs; int c_cs_save = c_cs; double one = bl1_d1(); double* b_copy; int dim_a; int b_copy_rs, b_copy_cs; // Return early if possible. if ( bl1_zero_dim2( m, n ) ) return; // If necessary, allocate, initialize, and use a temporary contiguous // copy of each matrix rather than the original matrices. bl1_set_dim_with_side( side, m, n, &dim_a ); bl1_dcreate_contigmr( uplo, dim_a, dim_a, a_save, a_rs_save, a_cs_save, &a, &a_rs, &a_cs ); bl1_dcreate_contigm( m, n, b_save, b_rs_save, b_cs_save, &b, &b_rs, &b_cs ); bl1_dcreate_contigm( m, n, c_save, c_rs_save, c_cs_save, &c, &c_rs, &c_cs ); // Create a copy of B to use in the computation so the original matrix is // left untouched. b_copy = bl1_dallocm( m, n ); // Match the strides of B_copy to that of B. if ( bl1_is_col_storage( b_rs, b_cs ) ) { b_copy_rs = 1; b_copy_cs = m; } else // if ( bl1_is_row_storage( b_rs, b_cs ) ) { b_copy_rs = n; b_copy_cs = 1; } // Copy the contents of B to B_copy. bl1_dcopymt( BLIS1_NO_TRANSPOSE, m, n, b, b_rs, b_cs, b_copy, b_copy_rs, b_copy_cs ); // Perform the operation on B_copy. bl1_dtrmm( side, uplo, trans, diag, m, n, alpha, a, a_rs, a_cs, b_copy, b_copy_rs, b_copy_cs ); // Scale C by beta. bl1_dscalm( BLIS1_NO_CONJUGATE, m, n, beta, c, c_rs, c_cs ); // Add B_copy into C. bl1_daxpymt( BLIS1_NO_TRANSPOSE, m, n, &one, b_copy, b_copy_rs, b_copy_cs, c, c_rs, c_cs ); // Free the copy of B. bl1_dfree( b_copy ); // Free any temporary contiguous matrices, copying the result back to // the original matrix. bl1_dfree_contigm( a_save, a_rs_save, a_cs_save, &a, &a_rs, &a_cs ); bl1_dfree_contigm( b_save, b_rs_save, b_cs_save, &b, &b_rs, &b_cs ); bl1_dfree_saved_contigm( m_save, n_save, c_save, c_rs_save, c_cs_save, &c, &c_rs, &c_cs ); }
void bl1_strmmsx | ( | side1_t | side, |
uplo1_t | uplo, | ||
trans1_t | trans, | ||
diag1_t | diag, | ||
int | m, | ||
int | n, | ||
float * | alpha, | ||
float * | a, | ||
int | a_rs, | ||
int | a_cs, | ||
float * | b, | ||
int | b_rs, | ||
int | b_cs, | ||
float * | beta, | ||
float * | c, | ||
int | c_rs, | ||
int | c_cs | ||
) |
References bl1_is_col_storage(), bl1_s1(), bl1_sallocm(), bl1_saxpymt(), bl1_scopymt(), bl1_screate_contigm(), bl1_screate_contigmr(), bl1_set_dim_with_side(), bl1_sfree(), bl1_sfree_contigm(), bl1_sfree_saved_contigm(), bl1_sscalm(), bl1_strmm(), bl1_zero_dim2(), BLIS1_NO_CONJUGATE, and BLIS1_NO_TRANSPOSE.
Referenced by FLA_Trmmsx_external().
{ int m_save = m; int n_save = n; float* a_save = a; float* b_save = b; float* c_save = c; int a_rs_save = a_rs; int a_cs_save = a_cs; int b_rs_save = b_rs; int b_cs_save = b_cs; int c_rs_save = c_rs; int c_cs_save = c_cs; float one = bl1_s1(); float* b_copy; int dim_a; int b_copy_rs, b_copy_cs; // Return early if possible. if ( bl1_zero_dim2( m, n ) ) return; // If necessary, allocate, initialize, and use a temporary contiguous // copy of each matrix rather than the original matrices. bl1_set_dim_with_side( side, m, n, &dim_a ); bl1_screate_contigmr( uplo, dim_a, dim_a, a_save, a_rs_save, a_cs_save, &a, &a_rs, &a_cs ); bl1_screate_contigm( m, n, b_save, b_rs_save, b_cs_save, &b, &b_rs, &b_cs ); bl1_screate_contigm( m, n, c_save, c_rs_save, c_cs_save, &c, &c_rs, &c_cs ); // Create a copy of B to use in the computation so the original matrix is // left untouched. b_copy = bl1_sallocm( m, n ); // Match the strides of B_copy to that of B. if ( bl1_is_col_storage( b_rs, b_cs ) ) { b_copy_rs = 1; b_copy_cs = m; } else // if ( bl1_is_row_storage( b_rs, b_cs ) ) { b_copy_rs = n; b_copy_cs = 1; } // Copy the contents of B to B_copy. bl1_scopymt( BLIS1_NO_TRANSPOSE, m, n, b, b_rs, b_cs, b_copy, b_copy_rs, b_copy_cs ); // Perform the operation on B_copy. bl1_strmm( side, uplo, trans, diag, m, n, alpha, a, a_rs, a_cs, b_copy, b_copy_rs, b_copy_cs ); // Scale C by beta. bl1_sscalm( BLIS1_NO_CONJUGATE, m, n, beta, c, c_rs, c_cs ); // Add B_copy into C. bl1_saxpymt( BLIS1_NO_TRANSPOSE, m, n, &one, b_copy, b_copy_rs, b_copy_cs, c, c_rs, c_cs ); // Free the copy of B. bl1_sfree( b_copy ); // Free any temporary contiguous matrices, copying the result back to // the original matrix. bl1_sfree_contigm( a_save, a_rs_save, a_cs_save, &a, &a_rs, &a_cs ); bl1_sfree_contigm( b_save, b_rs_save, b_cs_save, &b, &b_rs, &b_cs ); bl1_sfree_saved_contigm( m_save, n_save, c_save, c_rs_save, c_cs_save, &c, &c_rs, &c_cs ); }
void bl1_ztrmmsx | ( | side1_t | side, |
uplo1_t | uplo, | ||
trans1_t | trans, | ||
diag1_t | diag, | ||
int | m, | ||
int | n, | ||
dcomplex * | alpha, | ||
dcomplex * | a, | ||
int | a_rs, | ||
int | a_cs, | ||
dcomplex * | b, | ||
int | b_rs, | ||
int | b_cs, | ||
dcomplex * | beta, | ||
dcomplex * | c, | ||
int | c_rs, | ||
int | c_cs | ||
) |
References bl1_is_col_storage(), bl1_set_dim_with_side(), bl1_z1(), bl1_zallocm(), bl1_zaxpymt(), bl1_zcopymt(), bl1_zcreate_contigm(), bl1_zcreate_contigmr(), bl1_zero_dim2(), bl1_zfree(), bl1_zfree_contigm(), bl1_zfree_saved_contigm(), bl1_zscalm(), bl1_ztrmm(), BLIS1_NO_CONJUGATE, and BLIS1_NO_TRANSPOSE.
Referenced by FLA_Trmmsx_external().
{ int m_save = m; int n_save = n; dcomplex* a_save = a; dcomplex* b_save = b; dcomplex* c_save = c; int a_rs_save = a_rs; int a_cs_save = a_cs; int b_rs_save = b_rs; int b_cs_save = b_cs; int c_rs_save = c_rs; int c_cs_save = c_cs; dcomplex one = bl1_z1(); dcomplex* b_copy; int dim_a; int b_copy_rs, b_copy_cs; // Return early if possible. if ( bl1_zero_dim2( m, n ) ) return; // If necessary, allocate, initialize, and use a temporary contiguous // copy of each matrix rather than the original matrices. bl1_set_dim_with_side( side, m, n, &dim_a ); bl1_zcreate_contigmr( uplo, dim_a, dim_a, a_save, a_rs_save, a_cs_save, &a, &a_rs, &a_cs ); bl1_zcreate_contigm( m, n, b_save, b_rs_save, b_cs_save, &b, &b_rs, &b_cs ); bl1_zcreate_contigm( m, n, c_save, c_rs_save, c_cs_save, &c, &c_rs, &c_cs ); // Create a copy of B to use in the computation so the original matrix is // left untouched. b_copy = bl1_zallocm( m, n ); // Match the strides of B_copy to that of B. if ( bl1_is_col_storage( b_rs, b_cs ) ) { b_copy_rs = 1; b_copy_cs = m; } else // if ( bl1_is_row_storage( b_rs, b_cs ) ) { b_copy_rs = n; b_copy_cs = 1; } // Copy the contents of B to B_copy. bl1_zcopymt( BLIS1_NO_TRANSPOSE, m, n, b, b_rs, b_cs, b_copy, b_copy_rs, b_copy_cs ); // Perform the operation on B_copy. bl1_ztrmm( side, uplo, trans, diag, m, n, alpha, a, a_rs, a_cs, b_copy, b_copy_rs, b_copy_cs ); // Scale C by beta. bl1_zscalm( BLIS1_NO_CONJUGATE, m, n, beta, c, c_rs, c_cs ); // Add B_copy into C. bl1_zaxpymt( BLIS1_NO_TRANSPOSE, m, n, &one, b_copy, b_copy_rs, b_copy_cs, c, c_rs, c_cs ); // Free the copy of B. bl1_zfree( b_copy ); // Free any temporary contiguous matrices, copying the result back to // the original matrix. bl1_zfree_contigm( a_save, a_rs_save, a_cs_save, &a, &a_rs, &a_cs ); bl1_zfree_contigm( b_save, b_rs_save, b_cs_save, &b, &b_rs, &b_cs ); bl1_zfree_saved_contigm( m_save, n_save, c_save, c_rs_save, c_cs_save, &c, &c_rs, &c_cs ); }