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_dgbrfsx_work( int matrix_order, char trans, char equed,
00038 lapack_int n, lapack_int kl, lapack_int ku,
00039 lapack_int nrhs, const double* ab,
00040 lapack_int ldab, const double* afb,
00041 lapack_int ldafb, const lapack_int* ipiv,
00042 const double* r, const double* c,
00043 const double* b, lapack_int ldb, double* x,
00044 lapack_int ldx, double* rcond, double* berr,
00045 lapack_int n_err_bnds, double* err_bnds_norm,
00046 double* err_bnds_comp, lapack_int nparams,
00047 double* params, double* work,
00048 lapack_int* iwork )
00049 {
00050 lapack_int info = 0;
00051 if( matrix_order == LAPACK_COL_MAJOR ) {
00052
00053 LAPACK_dgbrfsx( &trans, &equed, &n, &kl, &ku, &nrhs, ab, &ldab, afb,
00054 &ldafb, ipiv, r, c, b, &ldb, x, &ldx, rcond, berr,
00055 &n_err_bnds, err_bnds_norm, err_bnds_comp, &nparams,
00056 params, work, iwork, &info );
00057 if( info < 0 ) {
00058 info = info - 1;
00059 }
00060 } else if( matrix_order == LAPACK_ROW_MAJOR ) {
00061 lapack_int ldab_t = MAX(1,kl+ku+1);
00062 lapack_int ldafb_t = MAX(1,2*kl+ku+1);
00063 lapack_int ldb_t = MAX(1,n);
00064 lapack_int ldx_t = MAX(1,n);
00065 double* ab_t = NULL;
00066 double* afb_t = NULL;
00067 double* b_t = NULL;
00068 double* x_t = NULL;
00069 double* err_bnds_norm_t = NULL;
00070 double* err_bnds_comp_t = NULL;
00071
00072 if( ldab < n ) {
00073 info = -9;
00074 LAPACKE_xerbla( "LAPACKE_dgbrfsx_work", info );
00075 return info;
00076 }
00077 if( ldafb < n ) {
00078 info = -11;
00079 LAPACKE_xerbla( "LAPACKE_dgbrfsx_work", info );
00080 return info;
00081 }
00082 if( ldb < nrhs ) {
00083 info = -16;
00084 LAPACKE_xerbla( "LAPACKE_dgbrfsx_work", info );
00085 return info;
00086 }
00087 if( ldx < nrhs ) {
00088 info = -18;
00089 LAPACKE_xerbla( "LAPACKE_dgbrfsx_work", info );
00090 return info;
00091 }
00092
00093 ab_t = (double*)LAPACKE_malloc( sizeof(double) * ldab_t * MAX(1,n) );
00094 if( ab_t == NULL ) {
00095 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00096 goto exit_level_0;
00097 }
00098 afb_t = (double*)LAPACKE_malloc( sizeof(double) * ldafb_t * MAX(1,n) );
00099 if( afb_t == NULL ) {
00100 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00101 goto exit_level_1;
00102 }
00103 b_t = (double*)LAPACKE_malloc( sizeof(double) * ldb_t * MAX(1,nrhs) );
00104 if( b_t == NULL ) {
00105 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00106 goto exit_level_2;
00107 }
00108 x_t = (double*)LAPACKE_malloc( sizeof(double) * ldx_t * MAX(1,nrhs) );
00109 if( x_t == NULL ) {
00110 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00111 goto exit_level_3;
00112 }
00113 err_bnds_norm_t = (double*)
00114 LAPACKE_malloc( sizeof(double) * nrhs * MAX(1,n_err_bnds) );
00115 if( err_bnds_norm_t == NULL ) {
00116 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00117 goto exit_level_4;
00118 }
00119 err_bnds_comp_t = (double*)
00120 LAPACKE_malloc( sizeof(double) * nrhs * MAX(1,n_err_bnds) );
00121 if( err_bnds_comp_t == NULL ) {
00122 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00123 goto exit_level_5;
00124 }
00125
00126 LAPACKE_dgb_trans( matrix_order, n, n, kl, ku, ab, ldab, ab_t, ldab_t );
00127 LAPACKE_dgb_trans( matrix_order, n, n, kl, kl+ku, afb, ldafb, afb_t,
00128 ldafb_t );
00129 LAPACKE_dge_trans( matrix_order, n, nrhs, b, ldb, b_t, ldb_t );
00130 LAPACKE_dge_trans( matrix_order, n, nrhs, x, ldx, x_t, ldx_t );
00131
00132 LAPACK_dgbrfsx( &trans, &equed, &n, &kl, &ku, &nrhs, ab_t, &ldab_t,
00133 afb_t, &ldafb_t, ipiv, r, c, b_t, &ldb_t, x_t, &ldx_t,
00134 rcond, berr, &n_err_bnds, err_bnds_norm_t,
00135 err_bnds_comp_t, &nparams, params, work, iwork, &info );
00136 if( info < 0 ) {
00137 info = info - 1;
00138 }
00139
00140 LAPACKE_dge_trans( LAPACK_COL_MAJOR, n, nrhs, x_t, ldx_t, x, ldx );
00141 LAPACKE_dge_trans( LAPACK_COL_MAJOR, nrhs, n_err_bnds, err_bnds_norm_t,
00142 nrhs, err_bnds_norm, nrhs );
00143 LAPACKE_dge_trans( LAPACK_COL_MAJOR, nrhs, n_err_bnds, err_bnds_comp_t,
00144 nrhs, err_bnds_comp, nrhs );
00145
00146 LAPACKE_free( err_bnds_comp_t );
00147 exit_level_5:
00148 LAPACKE_free( err_bnds_norm_t );
00149 exit_level_4:
00150 LAPACKE_free( x_t );
00151 exit_level_3:
00152 LAPACKE_free( b_t );
00153 exit_level_2:
00154 LAPACKE_free( afb_t );
00155 exit_level_1:
00156 LAPACKE_free( ab_t );
00157 exit_level_0:
00158 if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
00159 LAPACKE_xerbla( "LAPACKE_dgbrfsx_work", info );
00160 }
00161 } else {
00162 info = -1;
00163 LAPACKE_xerbla( "LAPACKE_dgbrfsx_work", info );
00164 }
00165 return info;
00166 }