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_sgees_work( int matrix_order, char jobvs, char sort,
00038 LAPACK_S_SELECT2 select, lapack_int n, float* a,
00039 lapack_int lda, lapack_int* sdim, float* wr,
00040 float* wi, float* vs, lapack_int ldvs,
00041 float* work, lapack_int lwork,
00042 lapack_logical* bwork )
00043 {
00044 lapack_int info = 0;
00045 if( matrix_order == LAPACK_COL_MAJOR ) {
00046
00047 LAPACK_sgees( &jobvs, &sort, select, &n, a, &lda, sdim, wr, wi, vs,
00048 &ldvs, work, &lwork, bwork, &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 ldvs_t = MAX(1,n);
00055 float* a_t = NULL;
00056 float* vs_t = NULL;
00057
00058 if( lda < n ) {
00059 info = -7;
00060 LAPACKE_xerbla( "LAPACKE_sgees_work", info );
00061 return info;
00062 }
00063 if( ldvs < n ) {
00064 info = -12;
00065 LAPACKE_xerbla( "LAPACKE_sgees_work", info );
00066 return info;
00067 }
00068
00069 if( lwork == -1 ) {
00070 LAPACK_sgees( &jobvs, &sort, select, &n, a, &lda_t, sdim, wr, wi,
00071 vs, &ldvs_t, work, &lwork, bwork, &info );
00072 return (info < 0) ? (info - 1) : info;
00073 }
00074
00075 a_t = (float*)LAPACKE_malloc( sizeof(float) * lda_t * MAX(1,n) );
00076 if( a_t == NULL ) {
00077 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00078 goto exit_level_0;
00079 }
00080 if( LAPACKE_lsame( jobvs, 'v' ) ) {
00081 vs_t = (float*)LAPACKE_malloc( sizeof(float) * ldvs_t * MAX(1,n) );
00082 if( vs_t == NULL ) {
00083 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00084 goto exit_level_1;
00085 }
00086 }
00087
00088 LAPACKE_sge_trans( matrix_order, n, n, a, lda, a_t, lda_t );
00089
00090 LAPACK_sgees( &jobvs, &sort, select, &n, a_t, &lda_t, sdim, wr, wi,
00091 vs_t, &ldvs_t, work, &lwork, bwork, &info );
00092 if( info < 0 ) {
00093 info = info - 1;
00094 }
00095
00096 LAPACKE_sge_trans( LAPACK_COL_MAJOR, n, n, a_t, lda_t, a, lda );
00097 if( LAPACKE_lsame( jobvs, 'v' ) ) {
00098 LAPACKE_sge_trans( LAPACK_COL_MAJOR, n, n, vs_t, ldvs_t, vs, ldvs );
00099 }
00100
00101 if( LAPACKE_lsame( jobvs, 'v' ) ) {
00102 LAPACKE_free( vs_t );
00103 }
00104 exit_level_1:
00105 LAPACKE_free( a_t );
00106 exit_level_0:
00107 if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
00108 LAPACKE_xerbla( "LAPACKE_sgees_work", info );
00109 }
00110 } else {
00111 info = -1;
00112 LAPACKE_xerbla( "LAPACKE_sgees_work", info );
00113 }
00114 return info;
00115 }