lapack_int LAPACKE_csptri_work( int matrix_order, char uplo, lapack_int n,
lapack_complex_float* ap,
const lapack_int* ipiv,
lapack_complex_float* work )
{
lapack_int info = 0;
if( matrix_order == LAPACK_COL_MAJOR ) {
/* Call LAPACK function and adjust info */
LAPACK_csptri( &uplo, &n, ap, ipiv, work, &info );
if( info < 0 ) {
info = info - 1;
}
} else if( matrix_order == LAPACK_ROW_MAJOR ) {
lapack_complex_float* ap_t = NULL;
/* Allocate memory for temporary array(s) */
ap_t = (lapack_complex_float*)
LAPACKE_malloc( sizeof(lapack_complex_float) *
( MAX(1,n) * MAX(2,n+1) ) / 2 );
if( ap_t == NULL ) {
info = LAPACK_TRANSPOSE_MEMORY_ERROR;
goto exit_level_0;
}
/* Transpose input matrices */
LAPACKE_csp_trans( matrix_order, uplo, n, ap, ap_t );
/* Call LAPACK function and adjust info */
LAPACK_csptri( &uplo, &n, ap_t, ipiv, work, &info );
if( info < 0 ) {
info = info - 1;
}
/* Transpose output matrices */
LAPACKE_csp_trans( LAPACK_COL_MAJOR, uplo, n, ap_t, ap );
/* Release memory and exit */
LAPACKE_free( ap_t );
exit_level_0:
if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
LAPACKE_xerbla( "LAPACKE_csptri_work", info );
}
} else {
info = -1;
LAPACKE_xerbla( "LAPACKE_csptri_work", info );
}
return info;
}
lapack_int LAPACKE_csptrs_work( int matrix_layout, char uplo, lapack_int n,
lapack_int nrhs, const lapack_complex_float* ap,
const lapack_int* ipiv, lapack_complex_float* b,
lapack_int ldb )
{
lapack_int info = 0;
if( matrix_layout == LAPACK_COL_MAJOR ) {
/* Call LAPACK function and adjust info */
LAPACK_csptrs( &uplo, &n, &nrhs, ap, ipiv, b, &ldb, &info );
if( info < 0 ) {
info = info - 1;
}
} else if( matrix_layout == LAPACK_ROW_MAJOR ) {
lapack_int ldb_t = MAX(1,n);
lapack_complex_float* b_t = NULL;
lapack_complex_float* ap_t = NULL;
/* Check leading dimension(s) */
if( ldb < nrhs ) {
info = -8;
LAPACKE_xerbla( "LAPACKE_csptrs_work", info );
return info;
}
/* Allocate memory for temporary array(s) */
b_t = (lapack_complex_float*)
LAPACKE_malloc( sizeof(lapack_complex_float) *
ldb_t * MAX(1,nrhs) );
if( b_t == NULL ) {
info = LAPACK_TRANSPOSE_MEMORY_ERROR;
goto exit_level_0;
}
ap_t = (lapack_complex_float*)
LAPACKE_malloc( sizeof(lapack_complex_float) *
( MAX(1,n) * MAX(2,n+1) ) / 2 );
if( ap_t == NULL ) {
info = LAPACK_TRANSPOSE_MEMORY_ERROR;
goto exit_level_1;
}
/* Transpose input matrices */
LAPACKE_cge_trans( matrix_layout, n, nrhs, b, ldb, b_t, ldb_t );
LAPACKE_csp_trans( matrix_layout, uplo, n, ap, ap_t );
/* Call LAPACK function and adjust info */
LAPACK_csptrs( &uplo, &n, &nrhs, ap_t, ipiv, b_t, &ldb_t, &info );
if( info < 0 ) {
info = info - 1;
}
/* Transpose output matrices */
LAPACKE_cge_trans( LAPACK_COL_MAJOR, n, nrhs, b_t, ldb_t, b, ldb );
/* Release memory and exit */
LAPACKE_free( ap_t );
exit_level_1:
LAPACKE_free( b_t );
exit_level_0:
if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
LAPACKE_xerbla( "LAPACKE_csptrs_work", info );
}
} else {
info = -1;
LAPACKE_xerbla( "LAPACKE_csptrs_work", info );
}
return info;
}
lapack_int LAPACKE_cspsvx_work( int matrix_order, char fact, char uplo,
lapack_int n, lapack_int nrhs,
const lapack_complex_float* ap,
lapack_complex_float* afp, lapack_int* ipiv,
const lapack_complex_float* b, lapack_int ldb,
lapack_complex_float* x, lapack_int ldx,
float* rcond, float* ferr, float* berr,
lapack_complex_float* work, float* rwork )
{
lapack_int info = 0;
if( matrix_order == LAPACK_COL_MAJOR ) {
/* Call LAPACK function and adjust info */
LAPACK_cspsvx( &fact, &uplo, &n, &nrhs, ap, afp, ipiv, b, &ldb, x, &ldx,
rcond, ferr, berr, work, rwork, &info );
if( info < 0 ) {
info = info - 1;
}
} else if( matrix_order == LAPACK_ROW_MAJOR ) {
lapack_int ldb_t = MAX(1,n);
lapack_int ldx_t = MAX(1,n);
lapack_complex_float* b_t = NULL;
lapack_complex_float* x_t = NULL;
lapack_complex_float* ap_t = NULL;
lapack_complex_float* afp_t = NULL;
/* Check leading dimension(s) */
if( ldb < nrhs ) {
info = -10;
LAPACKE_xerbla( "LAPACKE_cspsvx_work", info );
return info;
}
if( ldx < nrhs ) {
info = -12;
LAPACKE_xerbla( "LAPACKE_cspsvx_work", info );
return info;
}
/* Allocate memory for temporary array(s) */
b_t = (lapack_complex_float*)
LAPACKE_malloc( sizeof(lapack_complex_float) *
ldb_t * MAX(1,nrhs) );
if( b_t == NULL ) {
info = LAPACK_TRANSPOSE_MEMORY_ERROR;
goto exit_level_0;
}
x_t = (lapack_complex_float*)
LAPACKE_malloc( sizeof(lapack_complex_float) *
ldx_t * MAX(1,nrhs) );
if( x_t == NULL ) {
info = LAPACK_TRANSPOSE_MEMORY_ERROR;
goto exit_level_1;
}
ap_t = (lapack_complex_float*)
LAPACKE_malloc( sizeof(lapack_complex_float) *
( MAX(1,n) * MAX(2,n+1) ) / 2 );
if( ap_t == NULL ) {
info = LAPACK_TRANSPOSE_MEMORY_ERROR;
goto exit_level_2;
}
afp_t = (lapack_complex_float*)
LAPACKE_malloc( sizeof(lapack_complex_float) *
( MAX(1,n) * MAX(2,n+1) ) / 2 );
if( afp_t == NULL ) {
info = LAPACK_TRANSPOSE_MEMORY_ERROR;
goto exit_level_3;
}
/* Transpose input matrices */
LAPACKE_cge_trans( matrix_order, n, nrhs, b, ldb, b_t, ldb_t );
LAPACKE_csp_trans( matrix_order, uplo, n, ap, ap_t );
if( LAPACKE_lsame( fact, 'f' ) ) {
LAPACKE_csp_trans( matrix_order, uplo, n, afp, afp_t );
}
/* Call LAPACK function and adjust info */
LAPACK_cspsvx( &fact, &uplo, &n, &nrhs, ap_t, afp_t, ipiv, b_t, &ldb_t,
x_t, &ldx_t, rcond, ferr, berr, work, rwork, &info );
if( info < 0 ) {
info = info - 1;
}
/* Transpose output matrices */
LAPACKE_cge_trans( LAPACK_COL_MAJOR, n, nrhs, x_t, ldx_t, x, ldx );
if( LAPACKE_lsame( fact, 'n' ) ) {
LAPACKE_csp_trans( LAPACK_COL_MAJOR, uplo, n, afp_t, afp );
}
/* Release memory and exit */
LAPACKE_free( afp_t );
exit_level_3:
LAPACKE_free( ap_t );
exit_level_2:
LAPACKE_free( x_t );
exit_level_1:
LAPACKE_free( b_t );
exit_level_0:
if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
LAPACKE_xerbla( "LAPACKE_cspsvx_work", info );
}
} else {
info = -1;
LAPACKE_xerbla( "LAPACKE_cspsvx_work", info );
}
return info;
}
