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_cgtrfs( int matrix_order, char trans, lapack_int n,
00038 lapack_int nrhs, const lapack_complex_float* dl,
00039 const lapack_complex_float* d,
00040 const lapack_complex_float* du,
00041 const lapack_complex_float* dlf,
00042 const lapack_complex_float* df,
00043 const lapack_complex_float* duf,
00044 const lapack_complex_float* du2,
00045 const lapack_int* ipiv,
00046 const lapack_complex_float* b, lapack_int ldb,
00047 lapack_complex_float* x, lapack_int ldx, float* ferr,
00048 float* berr )
00049 {
00050 lapack_int info = 0;
00051 float* rwork = NULL;
00052 lapack_complex_float* work = NULL;
00053 if( matrix_order != LAPACK_COL_MAJOR && matrix_order != LAPACK_ROW_MAJOR ) {
00054 LAPACKE_xerbla( "LAPACKE_cgtrfs", -1 );
00055 return -1;
00056 }
00057 #ifndef LAPACK_DISABLE_NAN_CHECK
00058
00059 if( LAPACKE_cge_nancheck( matrix_order, n, nrhs, b, ldb ) ) {
00060 return -13;
00061 }
00062 if( LAPACKE_c_nancheck( n, d, 1 ) ) {
00063 return -6;
00064 }
00065 if( LAPACKE_c_nancheck( n, df, 1 ) ) {
00066 return -9;
00067 }
00068 if( LAPACKE_c_nancheck( n-1, dl, 1 ) ) {
00069 return -5;
00070 }
00071 if( LAPACKE_c_nancheck( n-1, dlf, 1 ) ) {
00072 return -8;
00073 }
00074 if( LAPACKE_c_nancheck( n-1, du, 1 ) ) {
00075 return -7;
00076 }
00077 if( LAPACKE_c_nancheck( n-2, du2, 1 ) ) {
00078 return -11;
00079 }
00080 if( LAPACKE_c_nancheck( n-1, duf, 1 ) ) {
00081 return -10;
00082 }
00083 if( LAPACKE_cge_nancheck( matrix_order, n, nrhs, x, ldx ) ) {
00084 return -15;
00085 }
00086 #endif
00087
00088 rwork = (float*)LAPACKE_malloc( sizeof(float) * MAX(1,n) );
00089 if( rwork == NULL ) {
00090 info = LAPACK_WORK_MEMORY_ERROR;
00091 goto exit_level_0;
00092 }
00093 work = (lapack_complex_float*)
00094 LAPACKE_malloc( sizeof(lapack_complex_float) * MAX(1,2*n) );
00095 if( work == NULL ) {
00096 info = LAPACK_WORK_MEMORY_ERROR;
00097 goto exit_level_1;
00098 }
00099
00100 info = LAPACKE_cgtrfs_work( matrix_order, trans, n, nrhs, dl, d, du, dlf,
00101 df, duf, du2, ipiv, b, ldb, x, ldx, ferr, berr,
00102 work, rwork );
00103
00104 LAPACKE_free( work );
00105 exit_level_1:
00106 LAPACKE_free( rwork );
00107 exit_level_0:
00108 if( info == LAPACK_WORK_MEMORY_ERROR ) {
00109 LAPACKE_xerbla( "LAPACKE_cgtrfs", info );
00110 }
00111 return info;
00112 }