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_chbevx_work( int matrix_order, char jobz, char range,
00038 char uplo, lapack_int n, lapack_int kd,
00039 lapack_complex_float* ab, lapack_int ldab,
00040 lapack_complex_float* q, lapack_int ldq,
00041 float vl, float vu, lapack_int il,
00042 lapack_int iu, float abstol, lapack_int* m,
00043 float* w, lapack_complex_float* z,
00044 lapack_int ldz, lapack_complex_float* work,
00045 float* rwork, lapack_int* iwork,
00046 lapack_int* ifail )
00047 {
00048 lapack_int info = 0;
00049 if( matrix_order == LAPACK_COL_MAJOR ) {
00050
00051 LAPACK_chbevx( &jobz, &range, &uplo, &n, &kd, ab, &ldab, q, &ldq, &vl,
00052 &vu, &il, &iu, &abstol, m, w, z, &ldz, work, rwork,
00053 iwork, ifail, &info );
00054 if( info < 0 ) {
00055 info = info - 1;
00056 }
00057 } else if( matrix_order == LAPACK_ROW_MAJOR ) {
00058 lapack_int ncols_z = ( LAPACKE_lsame( range, 'a' ) ||
00059 LAPACKE_lsame( range, 'v' ) ) ? n :
00060 ( LAPACKE_lsame( range, 'i' ) ? (iu-il+1) : 1);
00061 lapack_int ldab_t = MAX(1,kd+1);
00062 lapack_int ldq_t = MAX(1,n);
00063 lapack_int ldz_t = MAX(1,n);
00064 lapack_complex_float* ab_t = NULL;
00065 lapack_complex_float* q_t = NULL;
00066 lapack_complex_float* z_t = NULL;
00067
00068 if( ldab < n ) {
00069 info = -8;
00070 LAPACKE_xerbla( "LAPACKE_chbevx_work", info );
00071 return info;
00072 }
00073 if( ldq < n ) {
00074 info = -10;
00075 LAPACKE_xerbla( "LAPACKE_chbevx_work", info );
00076 return info;
00077 }
00078 if( ldz < ncols_z ) {
00079 info = -19;
00080 LAPACKE_xerbla( "LAPACKE_chbevx_work", info );
00081 return info;
00082 }
00083
00084 ab_t = (lapack_complex_float*)
00085 LAPACKE_malloc( sizeof(lapack_complex_float) * ldab_t * MAX(1,n) );
00086 if( ab_t == NULL ) {
00087 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00088 goto exit_level_0;
00089 }
00090 if( LAPACKE_lsame( jobz, 'v' ) ) {
00091 q_t = (lapack_complex_float*)
00092 LAPACKE_malloc( sizeof(lapack_complex_float) *
00093 ldq_t * MAX(1,n) );
00094 if( q_t == NULL ) {
00095 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00096 goto exit_level_1;
00097 }
00098 }
00099 if( LAPACKE_lsame( jobz, 'v' ) ) {
00100 z_t = (lapack_complex_float*)
00101 LAPACKE_malloc( sizeof(lapack_complex_float) *
00102 ldz_t * MAX(1,ncols_z) );
00103 if( z_t == NULL ) {
00104 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00105 goto exit_level_2;
00106 }
00107 }
00108
00109 LAPACKE_chb_trans( matrix_order, uplo, n, kd, ab, ldab, ab_t, ldab_t );
00110
00111 LAPACK_chbevx( &jobz, &range, &uplo, &n, &kd, ab_t, &ldab_t, q_t,
00112 &ldq_t, &vl, &vu, &il, &iu, &abstol, m, w, z_t, &ldz_t,
00113 work, rwork, iwork, ifail, &info );
00114 if( info < 0 ) {
00115 info = info - 1;
00116 }
00117
00118 LAPACKE_chb_trans( LAPACK_COL_MAJOR, uplo, n, kd, ab_t, ldab_t, ab,
00119 ldab );
00120 if( LAPACKE_lsame( jobz, 'v' ) ) {
00121 LAPACKE_cge_trans( LAPACK_COL_MAJOR, n, n, q_t, ldq_t, q, ldq );
00122 }
00123 if( LAPACKE_lsame( jobz, 'v' ) ) {
00124 LAPACKE_cge_trans( LAPACK_COL_MAJOR, n, ncols_z, z_t, ldz_t, z,
00125 ldz );
00126 }
00127
00128 if( LAPACKE_lsame( jobz, 'v' ) ) {
00129 LAPACKE_free( z_t );
00130 }
00131 exit_level_2:
00132 if( LAPACKE_lsame( jobz, 'v' ) ) {
00133 LAPACKE_free( q_t );
00134 }
00135 exit_level_1:
00136 LAPACKE_free( ab_t );
00137 exit_level_0:
00138 if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
00139 LAPACKE_xerbla( "LAPACKE_chbevx_work", info );
00140 }
00141 } else {
00142 info = -1;
00143 LAPACKE_xerbla( "LAPACKE_chbevx_work", info );
00144 }
00145 return info;
00146 }