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