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_zptrfs( int matrix_order, char uplo, lapack_int n,
00038 lapack_int nrhs, const double* d,
00039 const lapack_complex_double* e, const double* df,
00040 const lapack_complex_double* ef,
00041 const lapack_complex_double* b, lapack_int ldb,
00042 lapack_complex_double* x, lapack_int ldx,
00043 double* ferr, double* berr )
00044 {
00045 lapack_int info = 0;
00046 double* rwork = NULL;
00047 lapack_complex_double* work = NULL;
00048 if( matrix_order != LAPACK_COL_MAJOR && matrix_order != LAPACK_ROW_MAJOR ) {
00049 LAPACKE_xerbla( "LAPACKE_zptrfs", -1 );
00050 return -1;
00051 }
00052 #ifndef LAPACK_DISABLE_NAN_CHECK
00053
00054 if( LAPACKE_zge_nancheck( matrix_order, n, nrhs, b, ldb ) ) {
00055 return -9;
00056 }
00057 if( LAPACKE_d_nancheck( n, d, 1 ) ) {
00058 return -5;
00059 }
00060 if( LAPACKE_d_nancheck( n, df, 1 ) ) {
00061 return -7;
00062 }
00063 if( LAPACKE_z_nancheck( n-1, e, 1 ) ) {
00064 return -6;
00065 }
00066 if( LAPACKE_z_nancheck( n-1, ef, 1 ) ) {
00067 return -8;
00068 }
00069 if( LAPACKE_zge_nancheck( matrix_order, n, nrhs, x, ldx ) ) {
00070 return -11;
00071 }
00072 #endif
00073
00074 rwork = (double*)LAPACKE_malloc( sizeof(double) * MAX(1,n) );
00075 if( rwork == NULL ) {
00076 info = LAPACK_WORK_MEMORY_ERROR;
00077 goto exit_level_0;
00078 }
00079 work = (lapack_complex_double*)
00080 LAPACKE_malloc( sizeof(lapack_complex_double) * MAX(1,n) );
00081 if( work == NULL ) {
00082 info = LAPACK_WORK_MEMORY_ERROR;
00083 goto exit_level_1;
00084 }
00085
00086 info = LAPACKE_zptrfs_work( matrix_order, uplo, n, nrhs, d, e, df, ef, b,
00087 ldb, x, ldx, ferr, berr, work, rwork );
00088
00089 LAPACKE_free( work );
00090 exit_level_1:
00091 LAPACKE_free( rwork );
00092 exit_level_0:
00093 if( info == LAPACK_WORK_MEMORY_ERROR ) {
00094 LAPACKE_xerbla( "LAPACKE_zptrfs", info );
00095 }
00096 return info;
00097 }