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