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_zgtsvx_work( int matrix_order, char fact, char trans,
00038 lapack_int n, lapack_int nrhs,
00039 const lapack_complex_double* dl,
00040 const lapack_complex_double* d,
00041 const lapack_complex_double* du,
00042 lapack_complex_double* dlf,
00043 lapack_complex_double* df,
00044 lapack_complex_double* duf,
00045 lapack_complex_double* du2, lapack_int* ipiv,
00046 const lapack_complex_double* b, lapack_int ldb,
00047 lapack_complex_double* x, lapack_int ldx,
00048 double* rcond, double* ferr, double* berr,
00049 lapack_complex_double* work, double* rwork )
00050 {
00051 lapack_int info = 0;
00052 if( matrix_order == LAPACK_COL_MAJOR ) {
00053
00054 LAPACK_zgtsvx( &fact, &trans, &n, &nrhs, dl, d, du, dlf, df, duf, du2,
00055 ipiv, b, &ldb, x, &ldx, rcond, ferr, berr, work, rwork,
00056 &info );
00057 if( info < 0 ) {
00058 info = info - 1;
00059 }
00060 } else if( matrix_order == LAPACK_ROW_MAJOR ) {
00061 lapack_int ldb_t = MAX(1,n);
00062 lapack_int ldx_t = MAX(1,n);
00063 lapack_complex_double* b_t = NULL;
00064 lapack_complex_double* x_t = NULL;
00065
00066 if( ldb < nrhs ) {
00067 info = -15;
00068 LAPACKE_xerbla( "LAPACKE_zgtsvx_work", info );
00069 return info;
00070 }
00071 if( ldx < nrhs ) {
00072 info = -17;
00073 LAPACKE_xerbla( "LAPACKE_zgtsvx_work", info );
00074 return info;
00075 }
00076
00077 b_t = (lapack_complex_double*)
00078 LAPACKE_malloc( sizeof(lapack_complex_double) *
00079 ldb_t * MAX(1,nrhs) );
00080 if( b_t == NULL ) {
00081 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00082 goto exit_level_0;
00083 }
00084 x_t = (lapack_complex_double*)
00085 LAPACKE_malloc( sizeof(lapack_complex_double) *
00086 ldx_t * MAX(1,nrhs) );
00087 if( x_t == NULL ) {
00088 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00089 goto exit_level_1;
00090 }
00091
00092 LAPACKE_zge_trans( matrix_order, n, nrhs, b, ldb, b_t, ldb_t );
00093
00094 LAPACK_zgtsvx( &fact, &trans, &n, &nrhs, dl, d, du, dlf, df, duf, du2,
00095 ipiv, b_t, &ldb_t, x_t, &ldx_t, rcond, ferr, berr, work,
00096 rwork, &info );
00097 if( info < 0 ) {
00098 info = info - 1;
00099 }
00100
00101 LAPACKE_zge_trans( LAPACK_COL_MAJOR, n, nrhs, x_t, ldx_t, x, ldx );
00102
00103 LAPACKE_free( x_t );
00104 exit_level_1:
00105 LAPACKE_free( b_t );
00106 exit_level_0:
00107 if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
00108 LAPACKE_xerbla( "LAPACKE_zgtsvx_work", info );
00109 }
00110 } else {
00111 info = -1;
00112 LAPACKE_xerbla( "LAPACKE_zgtsvx_work", info );
00113 }
00114 return info;
00115 }