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