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