LAPACK  3.10.0
LAPACK: Linear Algebra PACKage

◆ LAPACKE_chegv_2stage()

lapack_int LAPACKE_chegv_2stage ( int  matrix_layout,
lapack_int  itype,
char  jobz,
char  uplo,
lapack_int  n,
lapack_complex_float a,
lapack_int  lda,
lapack_complex_float b,
lapack_int  ldb,
float *  w 
)

Definition at line 35 of file lapacke_chegv_2stage.c.

39 {
40  lapack_int info = 0;
41  lapack_int lwork = -1;
42  float* rwork = NULL;
43  lapack_complex_float* work = NULL;
44  lapack_complex_float work_query;
45  if( matrix_layout != LAPACK_COL_MAJOR && matrix_layout != LAPACK_ROW_MAJOR ) {
46  LAPACKE_xerbla( "LAPACKE_chegv_2stage", -1 );
47  return -1;
48  }
49 #ifndef LAPACK_DISABLE_NAN_CHECK
50  if( LAPACKE_get_nancheck() ) {
51  /* Optionally check input matrices for NaNs */
52  if( LAPACKE_che_nancheck( matrix_layout, uplo, n, a, lda ) ) {
53  return -6;
54  }
55  if( LAPACKE_che_nancheck( matrix_layout, uplo, n, b, ldb ) ) {
56  return -8;
57  }
58  }
59 #endif
60  /* Allocate memory for working array(s) */
61  rwork = (float*)LAPACKE_malloc( sizeof(float) * MAX(1,3*n-2) );
62  if( rwork == NULL ) {
64  goto exit_level_0;
65  }
66  /* Query optimal working array(s) size */
67  info = LAPACKE_chegv_2stage_work( matrix_layout, itype, jobz, uplo, n, a, lda, b,
68  ldb, w, &work_query, lwork, rwork );
69  if( info != 0 ) {
70  goto exit_level_1;
71  }
72  lwork = LAPACK_C2INT( work_query );
73  /* Allocate memory for work arrays */
74  work = (lapack_complex_float*)
75  LAPACKE_malloc( sizeof(lapack_complex_float) * lwork );
76  if( work == NULL ) {
78  goto exit_level_1;
79  }
80  /* Call middle-level interface */
81  info = LAPACKE_chegv_2stage_work( matrix_layout, itype, jobz, uplo, n, a, lda, b,
82  ldb, w, work, lwork, rwork );
83  /* Release memory and exit */
84  LAPACKE_free( work );
85 exit_level_1:
86  LAPACKE_free( rwork );
87 exit_level_0:
88  if( info == LAPACK_WORK_MEMORY_ERROR ) {
89  LAPACKE_xerbla( "LAPACKE_chegv_2stage", info );
90  }
91  return info;
92 }
#define lapack_int
Definition: lapack.h:83
#define lapack_complex_float
Definition: lapack.h:45
#define LAPACK_WORK_MEMORY_ERROR
Definition: lapacke.h:55
#define LAPACK_COL_MAJOR
Definition: lapacke.h:53
#define LAPACK_C2INT(x)
Definition: lapacke.h:49
#define LAPACKE_free(p)
Definition: lapacke.h:46
#define LAPACK_ROW_MAJOR
Definition: lapacke.h:52
int LAPACKE_get_nancheck(void)
lapack_int LAPACKE_chegv_2stage_work(int matrix_layout, lapack_int itype, char jobz, char uplo, lapack_int n, lapack_complex_float *a, lapack_int lda, lapack_complex_float *b, lapack_int ldb, float *w, lapack_complex_float *work, lapack_int lwork, float *rwork)
#define LAPACKE_malloc(size)
Definition: lapacke.h:43
void LAPACKE_xerbla(const char *name, lapack_int info)
lapack_logical LAPACKE_che_nancheck(int matrix_layout, char uplo, lapack_int n, const lapack_complex_float *a, lapack_int lda)
#define MAX(x, y)
Definition: lapacke_utils.h:46
Here is the call graph for this function: