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_cpbsv_work( int matrix_order, char uplo, lapack_int n,
00038 lapack_int kd, lapack_int nrhs,
00039 lapack_complex_float* ab, lapack_int ldab,
00040 lapack_complex_float* b, lapack_int ldb )
00041 {
00042 lapack_int info = 0;
00043 if( matrix_order == LAPACK_COL_MAJOR ) {
00044
00045 LAPACK_cpbsv( &uplo, &n, &kd, &nrhs, ab, &ldab, b, &ldb, &info );
00046 if( info < 0 ) {
00047 info = info - 1;
00048 }
00049 } else if( matrix_order == LAPACK_ROW_MAJOR ) {
00050 lapack_int ldab_t = MAX(1,kd+1);
00051 lapack_int ldb_t = MAX(1,n);
00052 lapack_complex_float* ab_t = NULL;
00053 lapack_complex_float* b_t = NULL;
00054
00055 if( ldab < n ) {
00056 info = -7;
00057 LAPACKE_xerbla( "LAPACKE_cpbsv_work", info );
00058 return info;
00059 }
00060 if( ldb < nrhs ) {
00061 info = -9;
00062 LAPACKE_xerbla( "LAPACKE_cpbsv_work", info );
00063 return info;
00064 }
00065
00066 ab_t = (lapack_complex_float*)
00067 LAPACKE_malloc( sizeof(lapack_complex_float) * ldab_t * MAX(1,n) );
00068 if( ab_t == NULL ) {
00069 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00070 goto exit_level_0;
00071 }
00072 b_t = (lapack_complex_float*)
00073 LAPACKE_malloc( sizeof(lapack_complex_float) *
00074 ldb_t * MAX(1,nrhs) );
00075 if( b_t == NULL ) {
00076 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00077 goto exit_level_1;
00078 }
00079
00080 LAPACKE_cpb_trans( matrix_order, uplo, n, kd, ab, ldab, ab_t, ldab_t );
00081 LAPACKE_cge_trans( matrix_order, n, nrhs, b, ldb, b_t, ldb_t );
00082
00083 LAPACK_cpbsv( &uplo, &n, &kd, &nrhs, ab_t, &ldab_t, b_t, &ldb_t,
00084 &info );
00085 if( info < 0 ) {
00086 info = info - 1;
00087 }
00088
00089 LAPACKE_cpb_trans( LAPACK_COL_MAJOR, uplo, n, kd, ab_t, ldab_t, ab,
00090 ldab );
00091 LAPACKE_cge_trans( LAPACK_COL_MAJOR, n, nrhs, b_t, ldb_t, b, ldb );
00092
00093 LAPACKE_free( b_t );
00094 exit_level_1:
00095 LAPACKE_free( ab_t );
00096 exit_level_0:
00097 if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
00098 LAPACKE_xerbla( "LAPACKE_cpbsv_work", info );
00099 }
00100 } else {
00101 info = -1;
00102 LAPACKE_xerbla( "LAPACKE_cpbsv_work", info );
00103 }
00104 return info;
00105 }