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_dtgsen_work( int matrix_order, lapack_int ijob,
00038 lapack_logical wantq, lapack_logical wantz,
00039 const lapack_logical* select, lapack_int n,
00040 double* a, lapack_int lda, double* b,
00041 lapack_int ldb, double* alphar, double* alphai,
00042 double* beta, double* q, lapack_int ldq,
00043 double* z, lapack_int ldz, lapack_int* m,
00044 double* pl, double* pr, double* dif,
00045 double* work, lapack_int lwork,
00046 lapack_int* iwork, lapack_int liwork )
00047 {
00048 lapack_int info = 0;
00049 if( matrix_order == LAPACK_COL_MAJOR ) {
00050
00051 LAPACK_dtgsen( &ijob, &wantq, &wantz, select, &n, a, &lda, b, &ldb,
00052 alphar, alphai, beta, q, &ldq, z, &ldz, m, pl, pr, dif,
00053 work, &lwork, iwork, &liwork, &info );
00054 if( info < 0 ) {
00055 info = info - 1;
00056 }
00057 } else if( matrix_order == LAPACK_ROW_MAJOR ) {
00058 lapack_int lda_t = MAX(1,n);
00059 lapack_int ldb_t = MAX(1,n);
00060 lapack_int ldq_t = MAX(1,n);
00061 lapack_int ldz_t = MAX(1,n);
00062 double* a_t = NULL;
00063 double* b_t = NULL;
00064 double* q_t = NULL;
00065 double* z_t = NULL;
00066
00067 if( lda < n ) {
00068 info = -8;
00069 LAPACKE_xerbla( "LAPACKE_dtgsen_work", info );
00070 return info;
00071 }
00072 if( ldb < n ) {
00073 info = -10;
00074 LAPACKE_xerbla( "LAPACKE_dtgsen_work", info );
00075 return info;
00076 }
00077 if( ldq < n ) {
00078 info = -15;
00079 LAPACKE_xerbla( "LAPACKE_dtgsen_work", info );
00080 return info;
00081 }
00082 if( ldz < n ) {
00083 info = -17;
00084 LAPACKE_xerbla( "LAPACKE_dtgsen_work", info );
00085 return info;
00086 }
00087
00088 if( liwork == -1 || lwork == -1 ) {
00089 LAPACK_dtgsen( &ijob, &wantq, &wantz, select, &n, a, &lda_t, b,
00090 &ldb_t, alphar, alphai, beta, q, &ldq_t, z, &ldz_t,
00091 m, pl, pr, dif, work, &lwork, iwork, &liwork,
00092 &info );
00093 return (info < 0) ? (info - 1) : info;
00094 }
00095
00096 a_t = (double*)LAPACKE_malloc( sizeof(double) * lda_t * MAX(1,n) );
00097 if( a_t == NULL ) {
00098 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00099 goto exit_level_0;
00100 }
00101 b_t = (double*)LAPACKE_malloc( sizeof(double) * ldb_t * MAX(1,n) );
00102 if( b_t == NULL ) {
00103 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00104 goto exit_level_1;
00105 }
00106 if( wantq ) {
00107 q_t = (double*)LAPACKE_malloc( sizeof(double) * ldq_t * MAX(1,n) );
00108 if( q_t == NULL ) {
00109 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00110 goto exit_level_2;
00111 }
00112 }
00113 if( wantz ) {
00114 z_t = (double*)LAPACKE_malloc( sizeof(double) * ldz_t * MAX(1,n) );
00115 if( z_t == NULL ) {
00116 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00117 goto exit_level_3;
00118 }
00119 }
00120
00121 LAPACKE_dge_trans( matrix_order, n, n, a, lda, a_t, lda_t );
00122 LAPACKE_dge_trans( matrix_order, n, n, b, ldb, b_t, ldb_t );
00123 if( wantq ) {
00124 LAPACKE_dge_trans( matrix_order, n, n, q, ldq, q_t, ldq_t );
00125 }
00126 if( wantz ) {
00127 LAPACKE_dge_trans( matrix_order, n, n, z, ldz, z_t, ldz_t );
00128 }
00129
00130 LAPACK_dtgsen( &ijob, &wantq, &wantz, select, &n, a_t, &lda_t, b_t,
00131 &ldb_t, alphar, alphai, beta, q_t, &ldq_t, z_t, &ldz_t,
00132 m, pl, pr, dif, work, &lwork, iwork, &liwork, &info );
00133 if( info < 0 ) {
00134 info = info - 1;
00135 }
00136
00137 LAPACKE_dge_trans( LAPACK_COL_MAJOR, n, n, a_t, lda_t, a, lda );
00138 LAPACKE_dge_trans( LAPACK_COL_MAJOR, n, n, b_t, ldb_t, b, ldb );
00139 if( wantq ) {
00140 LAPACKE_dge_trans( LAPACK_COL_MAJOR, n, n, q_t, ldq_t, q, ldq );
00141 }
00142 if( wantz ) {
00143 LAPACKE_dge_trans( LAPACK_COL_MAJOR, n, n, z_t, ldz_t, z, ldz );
00144 }
00145
00146 if( wantz ) {
00147 LAPACKE_free( z_t );
00148 }
00149 exit_level_3:
00150 if( wantq ) {
00151 LAPACKE_free( q_t );
00152 }
00153 exit_level_2:
00154 LAPACKE_free( b_t );
00155 exit_level_1:
00156 LAPACKE_free( a_t );
00157 exit_level_0:
00158 if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
00159 LAPACKE_xerbla( "LAPACKE_dtgsen_work", info );
00160 }
00161 } else {
00162 info = -1;
00163 LAPACKE_xerbla( "LAPACKE_dtgsen_work", info );
00164 }
00165 return info;
00166 }