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_zhegst_work( int matrix_order, lapack_int itype, char uplo,
00038 lapack_int n, lapack_complex_double* a,
00039 lapack_int lda, const lapack_complex_double* b,
00040 lapack_int ldb )
00041 {
00042 lapack_int info = 0;
00043 if( matrix_order == LAPACK_COL_MAJOR ) {
00044
00045 LAPACK_zhegst( &itype, &uplo, &n, a, &lda, b, &ldb, &info );
00046 if( info < 0 ) {
00047 info = info - 1;
00048 }
00049 } else if( matrix_order == LAPACK_ROW_MAJOR ) {
00050 lapack_int lda_t = MAX(1,n);
00051 lapack_int ldb_t = MAX(1,n);
00052 lapack_complex_double* a_t = NULL;
00053 lapack_complex_double* b_t = NULL;
00054
00055 if( lda < n ) {
00056 info = -6;
00057 LAPACKE_xerbla( "LAPACKE_zhegst_work", info );
00058 return info;
00059 }
00060 if( ldb < n ) {
00061 info = -8;
00062 LAPACKE_xerbla( "LAPACKE_zhegst_work", info );
00063 return info;
00064 }
00065
00066 a_t = (lapack_complex_double*)
00067 LAPACKE_malloc( sizeof(lapack_complex_double) * lda_t * MAX(1,n) );
00068 if( a_t == NULL ) {
00069 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00070 goto exit_level_0;
00071 }
00072 b_t = (lapack_complex_double*)
00073 LAPACKE_malloc( sizeof(lapack_complex_double) * ldb_t * MAX(1,n) );
00074 if( b_t == NULL ) {
00075 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00076 goto exit_level_1;
00077 }
00078
00079 LAPACKE_zhe_trans( matrix_order, uplo, n, a, lda, a_t, lda_t );
00080 LAPACKE_zge_trans( matrix_order, n, n, b, ldb, b_t, ldb_t );
00081
00082 LAPACK_zhegst( &itype, &uplo, &n, a_t, &lda_t, b_t, &ldb_t, &info );
00083 if( info < 0 ) {
00084 info = info - 1;
00085 }
00086
00087 LAPACKE_zhe_trans( LAPACK_COL_MAJOR, uplo, n, a_t, lda_t, a, lda );
00088
00089 LAPACKE_free( b_t );
00090 exit_level_1:
00091 LAPACKE_free( a_t );
00092 exit_level_0:
00093 if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
00094 LAPACKE_xerbla( "LAPACKE_zhegst_work", info );
00095 }
00096 } else {
00097 info = -1;
00098 LAPACKE_xerbla( "LAPACKE_zhegst_work", info );
00099 }
00100 return info;
00101 }