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_ssyevx_work( int matrix_order, char jobz, char range,
00038 char uplo, lapack_int n, float* a,
00039 lapack_int lda, float vl, float vu,
00040 lapack_int il, lapack_int iu, float abstol,
00041 lapack_int* m, float* w, float* z,
00042 lapack_int ldz, float* work, lapack_int lwork,
00043 lapack_int* iwork, lapack_int* ifail )
00044 {
00045 lapack_int info = 0;
00046 if( matrix_order == LAPACK_COL_MAJOR ) {
00047
00048 LAPACK_ssyevx( &jobz, &range, &uplo, &n, a, &lda, &vl, &vu, &il, &iu,
00049 &abstol, m, w, z, &ldz, work, &lwork, iwork, ifail,
00050 &info );
00051 if( info < 0 ) {
00052 info = info - 1;
00053 }
00054 } else if( matrix_order == LAPACK_ROW_MAJOR ) {
00055 lapack_int ncols_z = ( LAPACKE_lsame( range, 'a' ) ||
00056 LAPACKE_lsame( range, 'v' ) ) ? n :
00057 ( LAPACKE_lsame( range, 'i' ) ? (iu-il+1) : 1);
00058 lapack_int lda_t = MAX(1,n);
00059 lapack_int ldz_t = MAX(1,n);
00060 float* a_t = NULL;
00061 float* z_t = NULL;
00062
00063 if( lda < n ) {
00064 info = -7;
00065 LAPACKE_xerbla( "LAPACKE_ssyevx_work", info );
00066 return info;
00067 }
00068 if( ldz < ncols_z ) {
00069 info = -16;
00070 LAPACKE_xerbla( "LAPACKE_ssyevx_work", info );
00071 return info;
00072 }
00073
00074 if( lwork == -1 ) {
00075 LAPACK_ssyevx( &jobz, &range, &uplo, &n, a, &lda_t, &vl, &vu, &il,
00076 &iu, &abstol, m, w, z, &ldz_t, work, &lwork, iwork,
00077 ifail, &info );
00078 return (info < 0) ? (info - 1) : info;
00079 }
00080
00081 a_t = (float*)LAPACKE_malloc( sizeof(float) * lda_t * MAX(1,n) );
00082 if( a_t == NULL ) {
00083 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00084 goto exit_level_0;
00085 }
00086 if( LAPACKE_lsame( jobz, 'v' ) ) {
00087 z_t = (float*)
00088 LAPACKE_malloc( sizeof(float) * ldz_t * MAX(1,ncols_z) );
00089 if( z_t == NULL ) {
00090 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00091 goto exit_level_1;
00092 }
00093 }
00094
00095 LAPACKE_ssy_trans( matrix_order, uplo, n, a, lda, a_t, lda_t );
00096
00097 LAPACK_ssyevx( &jobz, &range, &uplo, &n, a_t, &lda_t, &vl, &vu, &il,
00098 &iu, &abstol, m, w, z_t, &ldz_t, work, &lwork, iwork,
00099 ifail, &info );
00100 if( info < 0 ) {
00101 info = info - 1;
00102 }
00103
00104 LAPACKE_ssy_trans( LAPACK_COL_MAJOR, uplo, n, a_t, lda_t, a, lda );
00105 if( LAPACKE_lsame( jobz, 'v' ) ) {
00106 LAPACKE_sge_trans( LAPACK_COL_MAJOR, n, ncols_z, z_t, ldz_t, z,
00107 ldz );
00108 }
00109
00110 if( LAPACKE_lsame( jobz, 'v' ) ) {
00111 LAPACKE_free( z_t );
00112 }
00113 exit_level_1:
00114 LAPACKE_free( a_t );
00115 exit_level_0:
00116 if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
00117 LAPACKE_xerbla( "LAPACKE_ssyevx_work", info );
00118 }
00119 } else {
00120 info = -1;
00121 LAPACKE_xerbla( "LAPACKE_ssyevx_work", info );
00122 }
00123 return info;
00124 }