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_zhbev_work( int matrix_order, char jobz, char uplo,
00038 lapack_int n, lapack_int kd,
00039 lapack_complex_double* ab, lapack_int ldab,
00040 double* w, lapack_complex_double* z,
00041 lapack_int ldz, lapack_complex_double* work,
00042 double* rwork )
00043 {
00044 lapack_int info = 0;
00045 if( matrix_order == LAPACK_COL_MAJOR ) {
00046
00047 LAPACK_zhbev( &jobz, &uplo, &n, &kd, ab, &ldab, w, z, &ldz, work, rwork,
00048 &info );
00049 if( info < 0 ) {
00050 info = info - 1;
00051 }
00052 } else if( matrix_order == LAPACK_ROW_MAJOR ) {
00053 lapack_int ldab_t = MAX(1,kd+1);
00054 lapack_int ldz_t = MAX(1,n);
00055 lapack_complex_double* ab_t = NULL;
00056 lapack_complex_double* z_t = NULL;
00057
00058 if( ldab < n ) {
00059 info = -7;
00060 LAPACKE_xerbla( "LAPACKE_zhbev_work", info );
00061 return info;
00062 }
00063 if( ldz < n ) {
00064 info = -10;
00065 LAPACKE_xerbla( "LAPACKE_zhbev_work", info );
00066 return info;
00067 }
00068
00069 ab_t = (lapack_complex_double*)
00070 LAPACKE_malloc( sizeof(lapack_complex_double) * ldab_t * MAX(1,n) );
00071 if( ab_t == NULL ) {
00072 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00073 goto exit_level_0;
00074 }
00075 if( LAPACKE_lsame( jobz, 'v' ) ) {
00076 z_t = (lapack_complex_double*)
00077 LAPACKE_malloc( sizeof(lapack_complex_double) *
00078 ldz_t * MAX(1,n) );
00079 if( z_t == NULL ) {
00080 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00081 goto exit_level_1;
00082 }
00083 }
00084
00085 LAPACKE_zhb_trans( matrix_order, uplo, n, kd, ab, ldab, ab_t, ldab_t );
00086
00087 LAPACK_zhbev( &jobz, &uplo, &n, &kd, ab_t, &ldab_t, w, z_t, &ldz_t,
00088 work, rwork, &info );
00089 if( info < 0 ) {
00090 info = info - 1;
00091 }
00092
00093 LAPACKE_zhb_trans( LAPACK_COL_MAJOR, uplo, n, kd, ab_t, ldab_t, ab,
00094 ldab );
00095 if( LAPACKE_lsame( jobz, 'v' ) ) {
00096 LAPACKE_zge_trans( LAPACK_COL_MAJOR, n, n, z_t, ldz_t, z, ldz );
00097 }
00098
00099 if( LAPACKE_lsame( jobz, 'v' ) ) {
00100 LAPACKE_free( z_t );
00101 }
00102 exit_level_1:
00103 LAPACKE_free( ab_t );
00104 exit_level_0:
00105 if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
00106 LAPACKE_xerbla( "LAPACKE_zhbev_work", info );
00107 }
00108 } else {
00109 info = -1;
00110 LAPACKE_xerbla( "LAPACKE_zhbev_work", info );
00111 }
00112 return info;
00113 }