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_dggbal_work( int matrix_order, char job, lapack_int n,
00038 double* a, lapack_int lda, double* b,
00039 lapack_int ldb, lapack_int* ilo,
00040 lapack_int* ihi, double* lscale, double* rscale,
00041 double* work )
00042 {
00043 lapack_int info = 0;
00044 if( matrix_order == LAPACK_COL_MAJOR ) {
00045
00046 LAPACK_dggbal( &job, &n, a, &lda, b, &ldb, ilo, ihi, lscale, rscale,
00047 work, &info );
00048 if( info < 0 ) {
00049 info = info - 1;
00050 }
00051 } else if( matrix_order == LAPACK_ROW_MAJOR ) {
00052 lapack_int lda_t = MAX(1,n);
00053 lapack_int ldb_t = MAX(1,n);
00054 double* a_t = NULL;
00055 double* b_t = NULL;
00056
00057 if( lda < n ) {
00058 info = -5;
00059 LAPACKE_xerbla( "LAPACKE_dggbal_work", info );
00060 return info;
00061 }
00062 if( ldb < n ) {
00063 info = -7;
00064 LAPACKE_xerbla( "LAPACKE_dggbal_work", info );
00065 return info;
00066 }
00067
00068 if( LAPACKE_lsame( job, 'p' ) || LAPACKE_lsame( job, 's' ) ||
00069 LAPACKE_lsame( job, 'b' ) ) {
00070 a_t = (double*)LAPACKE_malloc( sizeof(double) * lda_t * MAX(1,n) );
00071 if( a_t == NULL ) {
00072 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00073 goto exit_level_0;
00074 }
00075 }
00076 if( LAPACKE_lsame( job, 'p' ) || LAPACKE_lsame( job, 's' ) ||
00077 LAPACKE_lsame( job, 'b' ) ) {
00078 b_t = (double*)LAPACKE_malloc( sizeof(double) * ldb_t * MAX(1,n) );
00079 if( b_t == NULL ) {
00080 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00081 goto exit_level_1;
00082 }
00083 }
00084
00085 if( LAPACKE_lsame( job, 'p' ) || LAPACKE_lsame( job, 's' ) ||
00086 LAPACKE_lsame( job, 'b' ) ) {
00087 LAPACKE_dge_trans( matrix_order, n, n, a, lda, a_t, lda_t );
00088 }
00089 if( LAPACKE_lsame( job, 'p' ) || LAPACKE_lsame( job, 's' ) ||
00090 LAPACKE_lsame( job, 'b' ) ) {
00091 LAPACKE_dge_trans( matrix_order, n, n, b, ldb, b_t, ldb_t );
00092 }
00093
00094 LAPACK_dggbal( &job, &n, a_t, &lda_t, b_t, &ldb_t, ilo, ihi, lscale,
00095 rscale, work, &info );
00096 if( info < 0 ) {
00097 info = info - 1;
00098 }
00099
00100 if( LAPACKE_lsame( job, 'p' ) || LAPACKE_lsame( job, 's' ) ||
00101 LAPACKE_lsame( job, 'b' ) ) {
00102 LAPACKE_dge_trans( LAPACK_COL_MAJOR, n, n, a_t, lda_t, a, lda );
00103 }
00104 if( LAPACKE_lsame( job, 'p' ) || LAPACKE_lsame( job, 's' ) ||
00105 LAPACKE_lsame( job, 'b' ) ) {
00106 LAPACKE_dge_trans( LAPACK_COL_MAJOR, n, n, b_t, ldb_t, b, ldb );
00107 }
00108
00109 if( LAPACKE_lsame( job, 'p' ) || LAPACKE_lsame( job, 's' ) ||
00110 LAPACKE_lsame( job, 'b' ) ) {
00111 LAPACKE_free( b_t );
00112 }
00113 exit_level_1:
00114 if( LAPACKE_lsame( job, 'p' ) || LAPACKE_lsame( job, 's' ) ||
00115 LAPACKE_lsame( job, 'b' ) ) {
00116 LAPACKE_free( a_t );
00117 }
00118 exit_level_0:
00119 if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
00120 LAPACKE_xerbla( "LAPACKE_dggbal_work", info );
00121 }
00122 } else {
00123 info = -1;
00124 LAPACKE_xerbla( "LAPACKE_dggbal_work", info );
00125 }
00126 return info;
00127 }