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_zgghrd_work( int matrix_order, char compq, char compz,
00038 lapack_int n, lapack_int ilo, lapack_int ihi,
00039 lapack_complex_double* a, lapack_int lda,
00040 lapack_complex_double* b, lapack_int ldb,
00041 lapack_complex_double* q, lapack_int ldq,
00042 lapack_complex_double* z, lapack_int ldz )
00043 {
00044 lapack_int info = 0;
00045 if( matrix_order == LAPACK_COL_MAJOR ) {
00046
00047 LAPACK_zgghrd( &compq, &compz, &n, &ilo, &ihi, a, &lda, b, &ldb, q,
00048 &ldq, z, &ldz, &info );
00049 if( info < 0 ) {
00050 info = info - 1;
00051 }
00052 } else if( matrix_order == LAPACK_ROW_MAJOR ) {
00053 lapack_int lda_t = MAX(1,n);
00054 lapack_int ldb_t = MAX(1,n);
00055 lapack_int ldq_t = MAX(1,n);
00056 lapack_int ldz_t = MAX(1,n);
00057 lapack_complex_double* a_t = NULL;
00058 lapack_complex_double* b_t = NULL;
00059 lapack_complex_double* q_t = NULL;
00060 lapack_complex_double* z_t = NULL;
00061
00062 if( lda < n ) {
00063 info = -8;
00064 LAPACKE_xerbla( "LAPACKE_zgghrd_work", info );
00065 return info;
00066 }
00067 if( ldb < n ) {
00068 info = -10;
00069 LAPACKE_xerbla( "LAPACKE_zgghrd_work", info );
00070 return info;
00071 }
00072 if( ldq < n ) {
00073 info = -12;
00074 LAPACKE_xerbla( "LAPACKE_zgghrd_work", info );
00075 return info;
00076 }
00077 if( ldz < n ) {
00078 info = -14;
00079 LAPACKE_xerbla( "LAPACKE_zgghrd_work", info );
00080 return info;
00081 }
00082
00083 a_t = (lapack_complex_double*)
00084 LAPACKE_malloc( sizeof(lapack_complex_double) * lda_t * MAX(1,n) );
00085 if( a_t == NULL ) {
00086 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00087 goto exit_level_0;
00088 }
00089 b_t = (lapack_complex_double*)
00090 LAPACKE_malloc( sizeof(lapack_complex_double) * ldb_t * MAX(1,n) );
00091 if( b_t == NULL ) {
00092 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00093 goto exit_level_1;
00094 }
00095 if( LAPACKE_lsame( compq, 'i' ) || LAPACKE_lsame( compq, 'v' ) ) {
00096 q_t = (lapack_complex_double*)
00097 LAPACKE_malloc( sizeof(lapack_complex_double) *
00098 ldq_t * MAX(1,n) );
00099 if( q_t == NULL ) {
00100 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00101 goto exit_level_2;
00102 }
00103 }
00104 if( LAPACKE_lsame( compz, 'i' ) || LAPACKE_lsame( compz, 'v' ) ) {
00105 z_t = (lapack_complex_double*)
00106 LAPACKE_malloc( sizeof(lapack_complex_double) *
00107 ldz_t * MAX(1,n) );
00108 if( z_t == NULL ) {
00109 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00110 goto exit_level_3;
00111 }
00112 }
00113
00114 LAPACKE_zge_trans( matrix_order, n, n, a, lda, a_t, lda_t );
00115 LAPACKE_zge_trans( matrix_order, n, n, b, ldb, b_t, ldb_t );
00116 if( LAPACKE_lsame( compq, 'v' ) ) {
00117 LAPACKE_zge_trans( matrix_order, n, n, q, ldq, q_t, ldq_t );
00118 }
00119 if( LAPACKE_lsame( compz, 'v' ) ) {
00120 LAPACKE_zge_trans( matrix_order, n, n, z, ldz, z_t, ldz_t );
00121 }
00122
00123 LAPACK_zgghrd( &compq, &compz, &n, &ilo, &ihi, a_t, &lda_t, b_t, &ldb_t,
00124 q_t, &ldq_t, z_t, &ldz_t, &info );
00125 if( info < 0 ) {
00126 info = info - 1;
00127 }
00128
00129 LAPACKE_zge_trans( LAPACK_COL_MAJOR, n, n, a_t, lda_t, a, lda );
00130 LAPACKE_zge_trans( LAPACK_COL_MAJOR, n, n, b_t, ldb_t, b, ldb );
00131 if( LAPACKE_lsame( compq, 'i' ) || LAPACKE_lsame( compq, 'v' ) ) {
00132 LAPACKE_zge_trans( LAPACK_COL_MAJOR, n, n, q_t, ldq_t, q, ldq );
00133 }
00134 if( LAPACKE_lsame( compz, 'i' ) || LAPACKE_lsame( compz, 'v' ) ) {
00135 LAPACKE_zge_trans( LAPACK_COL_MAJOR, n, n, z_t, ldz_t, z, ldz );
00136 }
00137
00138 if( LAPACKE_lsame( compz, 'i' ) || LAPACKE_lsame( compz, 'v' ) ) {
00139 LAPACKE_free( z_t );
00140 }
00141 exit_level_3:
00142 if( LAPACKE_lsame( compq, 'i' ) || LAPACKE_lsame( compq, 'v' ) ) {
00143 LAPACKE_free( q_t );
00144 }
00145 exit_level_2:
00146 LAPACKE_free( b_t );
00147 exit_level_1:
00148 LAPACKE_free( a_t );
00149 exit_level_0:
00150 if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
00151 LAPACKE_xerbla( "LAPACKE_zgghrd_work", info );
00152 }
00153 } else {
00154 info = -1;
00155 LAPACKE_xerbla( "LAPACKE_zgghrd_work", info );
00156 }
00157 return info;
00158 }