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