LAPACK  3.8.0
LAPACK: Linear Algebra PACKage

◆ LAPACKE_zhegvx_work()

lapack_int LAPACKE_zhegvx_work ( int  matrix_layout,
lapack_int  itype,
char  jobz,
char  range,
char  uplo,
lapack_int  n,
lapack_complex_double a,
lapack_int  lda,
lapack_complex_double b,
lapack_int  ldb,
double  vl,
double  vu,
lapack_int  il,
lapack_int  iu,
double  abstol,
lapack_int m,
double *  w,
lapack_complex_double z,
lapack_int  ldz,
lapack_complex_double work,
lapack_int  lwork,
double *  rwork,
lapack_int iwork,
lapack_int ifail 
)

Definition at line 36 of file lapacke_zhegvx_work.c.

46 {
47  lapack_int info = 0;
48  if( matrix_layout == LAPACK_COL_MAJOR ) {
49  /* Call LAPACK function and adjust info */
50  LAPACK_zhegvx( &itype, &jobz, &range, &uplo, &n, a, &lda, b, &ldb, &vl,
51  &vu, &il, &iu, &abstol, m, w, z, &ldz, work, &lwork,
52  rwork, iwork, ifail, &info );
53  if( info < 0 ) {
54  info = info - 1;
55  }
56  } else if( matrix_layout == LAPACK_ROW_MAJOR ) {
57  lapack_int ncols_z = ( LAPACKE_lsame( range, 'a' ) ||
58  LAPACKE_lsame( range, 'v' ) ) ? n :
59  ( LAPACKE_lsame( range, 'i' ) ? (iu-il+1) : 1);
60  lapack_int lda_t = MAX(1,n);
61  lapack_int ldb_t = MAX(1,n);
62  lapack_int ldz_t = MAX(1,n);
63  lapack_complex_double* a_t = NULL;
64  lapack_complex_double* b_t = NULL;
65  lapack_complex_double* z_t = NULL;
66  /* Check leading dimension(s) */
67  if( lda < n ) {
68  info = -8;
69  LAPACKE_xerbla( "LAPACKE_zhegvx_work", info );
70  return info;
71  }
72  if( ldb < n ) {
73  info = -10;
74  LAPACKE_xerbla( "LAPACKE_zhegvx_work", info );
75  return info;
76  }
77  if( ldz < ncols_z ) {
78  info = -19;
79  LAPACKE_xerbla( "LAPACKE_zhegvx_work", info );
80  return info;
81  }
82  /* Query optimal working array(s) size if requested */
83  if( lwork == -1 ) {
84  LAPACK_zhegvx( &itype, &jobz, &range, &uplo, &n, a, &lda_t, b,
85  &ldb_t, &vl, &vu, &il, &iu, &abstol, m, w, z, &ldz_t,
86  work, &lwork, rwork, iwork, ifail, &info );
87  return (info < 0) ? (info - 1) : info;
88  }
89  /* Allocate memory for temporary array(s) */
90  a_t = (lapack_complex_double*)
91  LAPACKE_malloc( sizeof(lapack_complex_double) * lda_t * MAX(1,n) );
92  if( a_t == NULL ) {
94  goto exit_level_0;
95  }
96  b_t = (lapack_complex_double*)
97  LAPACKE_malloc( sizeof(lapack_complex_double) * ldb_t * MAX(1,n) );
98  if( b_t == NULL ) {
100  goto exit_level_1;
101  }
102  if( LAPACKE_lsame( jobz, 'v' ) ) {
103  z_t = (lapack_complex_double*)
105  ldz_t * MAX(1,ncols_z) );
106  if( z_t == NULL ) {
108  goto exit_level_2;
109  }
110  }
111  /* Transpose input matrices */
112  LAPACKE_zhe_trans( matrix_layout, uplo, n, a, lda, a_t, lda_t );
113  LAPACKE_zhe_trans( matrix_layout, uplo, n, b, ldb, b_t, ldb_t );
114  /* Call LAPACK function and adjust info */
115  LAPACK_zhegvx( &itype, &jobz, &range, &uplo, &n, a_t, &lda_t, b_t,
116  &ldb_t, &vl, &vu, &il, &iu, &abstol, m, w, z_t, &ldz_t,
117  work, &lwork, rwork, iwork, ifail, &info );
118  if( info < 0 ) {
119  info = info - 1;
120  }
121  /* Transpose output matrices */
122  LAPACKE_zhe_trans( LAPACK_COL_MAJOR, uplo, n, a_t, lda_t, a, lda );
123  LAPACKE_zhe_trans( LAPACK_COL_MAJOR, uplo, n, b_t, ldb_t, b, ldb );
124  if( LAPACKE_lsame( jobz, 'v' ) ) {
125  LAPACKE_zge_trans( LAPACK_COL_MAJOR, n, ncols_z, z_t, ldz_t, z,
126  ldz );
127  }
128  /* Release memory and exit */
129  if( LAPACKE_lsame( jobz, 'v' ) ) {
130  LAPACKE_free( z_t );
131  }
132 exit_level_2:
133  LAPACKE_free( b_t );
134 exit_level_1:
135  LAPACKE_free( a_t );
136 exit_level_0:
137  if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
138  LAPACKE_xerbla( "LAPACKE_zhegvx_work", info );
139  }
140  } else {
141  info = -1;
142  LAPACKE_xerbla( "LAPACKE_zhegvx_work", info );
143  }
144  return info;
145 }
#define LAPACK_ROW_MAJOR
Definition: lapacke.h:119
#define LAPACKE_free(p)
Definition: lapacke.h:113
#define lapack_complex_double
Definition: lapacke.h:90
void LAPACK_zhegvx(lapack_int *itype, char *jobz, char *range, char *uplo, lapack_int *n, lapack_complex_double *a, lapack_int *lda, lapack_complex_double *b, lapack_int *ldb, double *vl, double *vu, lapack_int *il, lapack_int *iu, double *abstol, lapack_int *m, double *w, lapack_complex_double *z, lapack_int *ldz, lapack_complex_double *work, lapack_int *lwork, double *rwork, lapack_int *iwork, lapack_int *ifail, lapack_int *info)
#define MAX(x, y)
Definition: lapacke_utils.h:47
#define LAPACKE_malloc(size)
Definition: lapacke.h:110
lapack_logical LAPACKE_lsame(char ca, char cb)
Definition: lapacke_lsame.c:36
#define LAPACK_COL_MAJOR
Definition: lapacke.h:120
void LAPACKE_xerbla(const char *name, lapack_int info)
void LAPACKE_zhe_trans(int matrix_layout, char uplo, lapack_int n, const lapack_complex_double *in, lapack_int ldin, lapack_complex_double *out, lapack_int ldout)
#define lapack_int
Definition: lapacke.h:47
#define LAPACK_TRANSPOSE_MEMORY_ERROR
Definition: lapacke.h:123
void LAPACKE_zge_trans(int matrix_layout, lapack_int m, lapack_int n, const lapack_complex_double *in, lapack_int ldin, lapack_complex_double *out, lapack_int ldout)
Here is the call graph for this function:
Here is the caller graph for this function: