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_cgebal_work( int matrix_order, char job, lapack_int n,
00038 lapack_complex_float* a, lapack_int lda,
00039 lapack_int* ilo, lapack_int* ihi, float* scale )
00040 {
00041 lapack_int info = 0;
00042 if( matrix_order == LAPACK_COL_MAJOR ) {
00043
00044 LAPACK_cgebal( &job, &n, a, &lda, ilo, ihi, scale, &info );
00045 if( info < 0 ) {
00046 info = info - 1;
00047 }
00048 } else if( matrix_order == LAPACK_ROW_MAJOR ) {
00049 lapack_int lda_t = MAX(1,n);
00050 lapack_complex_float* a_t = NULL;
00051
00052 if( lda < n ) {
00053 info = -5;
00054 LAPACKE_xerbla( "LAPACKE_cgebal_work", info );
00055 return info;
00056 }
00057
00058 if( LAPACKE_lsame( job, 'b' ) || LAPACKE_lsame( job, 'p' ) ||
00059 LAPACKE_lsame( job, 's' ) ) {
00060 a_t = (lapack_complex_float*)
00061 LAPACKE_malloc( sizeof(lapack_complex_float) *
00062 lda_t * MAX(1,n) );
00063 if( a_t == NULL ) {
00064 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00065 goto exit_level_0;
00066 }
00067 }
00068
00069 if( LAPACKE_lsame( job, 'b' ) || LAPACKE_lsame( job, 'p' ) ||
00070 LAPACKE_lsame( job, 's' ) ) {
00071 LAPACKE_cge_trans( matrix_order, n, n, a, lda, a_t, lda_t );
00072 }
00073
00074 LAPACK_cgebal( &job, &n, a_t, &lda_t, ilo, ihi, scale, &info );
00075 if( info < 0 ) {
00076 info = info - 1;
00077 }
00078
00079 if( LAPACKE_lsame( job, 'b' ) || LAPACKE_lsame( job, 'p' ) ||
00080 LAPACKE_lsame( job, 's' ) ) {
00081 LAPACKE_cge_trans( LAPACK_COL_MAJOR, n, n, a_t, lda_t, a, lda );
00082 }
00083
00084 if( LAPACKE_lsame( job, 'b' ) || LAPACKE_lsame( job, 'p' ) ||
00085 LAPACKE_lsame( job, 's' ) ) {
00086 LAPACKE_free( a_t );
00087 }
00088 exit_level_0:
00089 if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
00090 LAPACKE_xerbla( "LAPACKE_cgebal_work", info );
00091 }
00092 } else {
00093 info = -1;
00094 LAPACKE_xerbla( "LAPACKE_cgebal_work", info );
00095 }
00096 return info;
00097 }