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