00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034 #include "lapacke.h"
00035 #include "lapacke_utils.h"
00036
00037 lapack_int LAPACKE_sgtsvx_work( int matrix_order, char fact, char trans,
00038 lapack_int n, lapack_int nrhs, const float* dl,
00039 const float* d, const float* du, float* dlf,
00040 float* df, float* duf, float* du2,
00041 lapack_int* ipiv, const float* b,
00042 lapack_int ldb, float* x, lapack_int ldx,
00043 float* rcond, float* ferr, float* berr,
00044 float* work, lapack_int* iwork )
00045 {
00046 lapack_int info = 0;
00047 if( matrix_order == LAPACK_COL_MAJOR ) {
00048
00049 LAPACK_sgtsvx( &fact, &trans, &n, &nrhs, dl, d, du, dlf, df, duf, du2,
00050 ipiv, b, &ldb, x, &ldx, rcond, ferr, berr, work, iwork,
00051 &info );
00052 if( info < 0 ) {
00053 info = info - 1;
00054 }
00055 } else if( matrix_order == LAPACK_ROW_MAJOR ) {
00056 lapack_int ldb_t = MAX(1,n);
00057 lapack_int ldx_t = MAX(1,n);
00058 float* b_t = NULL;
00059 float* x_t = NULL;
00060
00061 if( ldb < nrhs ) {
00062 info = -15;
00063 LAPACKE_xerbla( "LAPACKE_sgtsvx_work", info );
00064 return info;
00065 }
00066 if( ldx < nrhs ) {
00067 info = -17;
00068 LAPACKE_xerbla( "LAPACKE_sgtsvx_work", info );
00069 return info;
00070 }
00071
00072 b_t = (float*)LAPACKE_malloc( sizeof(float) * ldb_t * MAX(1,nrhs) );
00073 if( b_t == NULL ) {
00074 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00075 goto exit_level_0;
00076 }
00077 x_t = (float*)LAPACKE_malloc( sizeof(float) * ldx_t * MAX(1,nrhs) );
00078 if( x_t == NULL ) {
00079 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00080 goto exit_level_1;
00081 }
00082
00083 LAPACKE_sge_trans( matrix_order, n, nrhs, b, ldb, b_t, ldb_t );
00084
00085 LAPACK_sgtsvx( &fact, &trans, &n, &nrhs, dl, d, du, dlf, df, duf, du2,
00086 ipiv, b_t, &ldb_t, x_t, &ldx_t, rcond, ferr, berr, work,
00087 iwork, &info );
00088 if( info < 0 ) {
00089 info = info - 1;
00090 }
00091
00092 LAPACKE_sge_trans( LAPACK_COL_MAJOR, n, nrhs, x_t, ldx_t, x, ldx );
00093
00094 LAPACKE_free( x_t );
00095 exit_level_1:
00096 LAPACKE_free( b_t );
00097 exit_level_0:
00098 if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
00099 LAPACKE_xerbla( "LAPACKE_sgtsvx_work", info );
00100 }
00101 } else {
00102 info = -1;
00103 LAPACKE_xerbla( "LAPACKE_sgtsvx_work", info );
00104 }
00105 return info;
00106 }