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_sposvx_work( int matrix_order, char fact, char uplo,
00038 lapack_int n, lapack_int nrhs, float* a,
00039 lapack_int lda, float* af, lapack_int ldaf,
00040 char* equed, float* s, float* b, lapack_int ldb,
00041 float* x, lapack_int ldx, float* rcond,
00042 float* ferr, float* berr, float* work,
00043 lapack_int* iwork )
00044 {
00045 lapack_int info = 0;
00046 if( matrix_order == LAPACK_COL_MAJOR ) {
00047
00048 LAPACK_sposvx( &fact, &uplo, &n, &nrhs, a, &lda, af, &ldaf, equed, s, b,
00049 &ldb, x, &ldx, rcond, ferr, berr, work, iwork, &info );
00050 if( info < 0 ) {
00051 info = info - 1;
00052 }
00053 } else if( matrix_order == LAPACK_ROW_MAJOR ) {
00054 lapack_int lda_t = MAX(1,n);
00055 lapack_int ldaf_t = MAX(1,n);
00056 lapack_int ldb_t = MAX(1,n);
00057 lapack_int ldx_t = MAX(1,n);
00058 float* a_t = NULL;
00059 float* af_t = NULL;
00060 float* b_t = NULL;
00061 float* x_t = NULL;
00062
00063 if( lda < n ) {
00064 info = -7;
00065 LAPACKE_xerbla( "LAPACKE_sposvx_work", info );
00066 return info;
00067 }
00068 if( ldaf < n ) {
00069 info = -9;
00070 LAPACKE_xerbla( "LAPACKE_sposvx_work", info );
00071 return info;
00072 }
00073 if( ldb < nrhs ) {
00074 info = -13;
00075 LAPACKE_xerbla( "LAPACKE_sposvx_work", info );
00076 return info;
00077 }
00078 if( ldx < nrhs ) {
00079 info = -15;
00080 LAPACKE_xerbla( "LAPACKE_sposvx_work", info );
00081 return info;
00082 }
00083
00084 a_t = (float*)LAPACKE_malloc( sizeof(float) * lda_t * MAX(1,n) );
00085 if( a_t == NULL ) {
00086 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00087 goto exit_level_0;
00088 }
00089 af_t = (float*)LAPACKE_malloc( sizeof(float) * ldaf_t * MAX(1,n) );
00090 if( af_t == NULL ) {
00091 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00092 goto exit_level_1;
00093 }
00094 b_t = (float*)LAPACKE_malloc( sizeof(float) * ldb_t * MAX(1,nrhs) );
00095 if( b_t == NULL ) {
00096 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00097 goto exit_level_2;
00098 }
00099 x_t = (float*)LAPACKE_malloc( sizeof(float) * ldx_t * MAX(1,nrhs) );
00100 if( x_t == NULL ) {
00101 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00102 goto exit_level_3;
00103 }
00104
00105 LAPACKE_spo_trans( matrix_order, uplo, n, a, lda, a_t, lda_t );
00106 if( LAPACKE_lsame( fact, 'f' ) ) {
00107 LAPACKE_spo_trans( matrix_order, uplo, n, af, ldaf, af_t, ldaf_t );
00108 }
00109 LAPACKE_sge_trans( matrix_order, n, nrhs, b, ldb, b_t, ldb_t );
00110
00111 LAPACK_sposvx( &fact, &uplo, &n, &nrhs, a_t, &lda_t, af_t, &ldaf_t,
00112 equed, s, b_t, &ldb_t, x_t, &ldx_t, rcond, ferr, berr,
00113 work, iwork, &info );
00114 if( info < 0 ) {
00115 info = info - 1;
00116 }
00117
00118 if( LAPACKE_lsame( fact, 'e' ) && LAPACKE_lsame( *equed, 'y' ) ) {
00119 LAPACKE_spo_trans( LAPACK_COL_MAJOR, uplo, n, a_t, lda_t, a, lda );
00120 }
00121 if( LAPACKE_lsame( fact, 'e' ) || LAPACKE_lsame( fact, 'n' ) ) {
00122 LAPACKE_spo_trans( LAPACK_COL_MAJOR, uplo, n, af_t, ldaf_t, af,
00123 ldaf );
00124 }
00125 LAPACKE_sge_trans( LAPACK_COL_MAJOR, n, nrhs, b_t, ldb_t, b, ldb );
00126 LAPACKE_sge_trans( LAPACK_COL_MAJOR, n, nrhs, x_t, ldx_t, x, ldx );
00127
00128 LAPACKE_free( x_t );
00129 exit_level_3:
00130 LAPACKE_free( b_t );
00131 exit_level_2:
00132 LAPACKE_free( af_t );
00133 exit_level_1:
00134 LAPACKE_free( a_t );
00135 exit_level_0:
00136 if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
00137 LAPACKE_xerbla( "LAPACKE_sposvx_work", info );
00138 }
00139 } else {
00140 info = -1;
00141 LAPACKE_xerbla( "LAPACKE_sposvx_work", info );
00142 }
00143 return info;
00144 }