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