LAPACK  3.7.0
LAPACK: Linear Algebra PACKage
lapack_int LAPACKE_chbevx_2stage_work ( int  matrix_layout,
char  jobz,
char  range,
char  uplo,
lapack_int  n,
lapack_int  kd,
lapack_complex_float *  ab,
lapack_int  ldab,
lapack_complex_float *  q,
lapack_int  ldq,
float  vl,
float  vu,
lapack_int  il,
lapack_int  iu,
float  abstol,
lapack_int *  m,
float *  w,
lapack_complex_float *  z,
lapack_int  ldz,
lapack_complex_float *  work,
lapack_int  lwork,
float *  rwork,
lapack_int *  iwork,
lapack_int *  ifail 
)

Definition at line 36 of file lapacke_chbevx_2stage_work.c.

46 {
47  lapack_int info = 0;
48  if( matrix_layout == LAPACK_COL_MAJOR ) {
49  /* Call LAPACK function and adjust info */
50  LAPACK_chbevx_2stage( &jobz, &range, &uplo, &n, &kd, ab, &ldab, q, &ldq, &vl,
51  &vu, &il, &iu, &abstol, m, w, z, &ldz, work, &lwork, rwork,
52  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 ldab_t = MAX(1,kd+1);
61  lapack_int ldq_t = MAX(1,n);
62  lapack_int ldz_t = MAX(1,n);
63  lapack_complex_float* ab_t = NULL;
64  lapack_complex_float* q_t = NULL;
65  lapack_complex_float* z_t = NULL;
66  /* Check leading dimension(s) */
67  if( ldab < n ) {
68  info = -8;
69  LAPACKE_xerbla( "LAPACKE_chbevx_2stage_work", info );
70  return info;
71  }
72  if( ldq < n ) {
73  info = -10;
74  LAPACKE_xerbla( "LAPACKE_chbevx_2stage_work", info );
75  return info;
76  }
77  if( ldz < ncols_z ) {
78  info = -19;
79  LAPACKE_xerbla( "LAPACKE_chbevx_2stage_work", info );
80  return info;
81  }
82  /* Allocate memory for temporary array(s) */
83  ab_t = (lapack_complex_float*)
84  LAPACKE_malloc( sizeof(lapack_complex_float) * ldab_t * MAX(1,n) );
85  if( ab_t == NULL ) {
86  info = LAPACK_TRANSPOSE_MEMORY_ERROR;
87  goto exit_level_0;
88  }
89  if( LAPACKE_lsame( jobz, 'v' ) ) {
90  q_t = (lapack_complex_float*)
91  LAPACKE_malloc( sizeof(lapack_complex_float) *
92  ldq_t * MAX(1,n) );
93  if( q_t == NULL ) {
94  info = LAPACK_TRANSPOSE_MEMORY_ERROR;
95  goto exit_level_1;
96  }
97  }
98  if( LAPACKE_lsame( jobz, 'v' ) ) {
99  z_t = (lapack_complex_float*)
100  LAPACKE_malloc( sizeof(lapack_complex_float) *
101  ldz_t * MAX(1,ncols_z) );
102  if( z_t == NULL ) {
103  info = LAPACK_TRANSPOSE_MEMORY_ERROR;
104  goto exit_level_2;
105  }
106  }
107  /* Query optimal working array(s) size if requested */
108  if( lwork == -1 ) {
109  LAPACK_chbevx_2stage( &jobz, &range, &uplo, &n, &kd, ab_t, &ldab_t, q_t,
110  &ldq_t, &vl, &vu, &il, &iu, &abstol, m, w, z_t, &ldz_t,
111  work, &lwork, rwork, iwork, ifail, &info );
112  return (info < 0) ? (info - 1) : info;
113  }
114  /* Transpose input matrices */
115  LAPACKE_chb_trans( matrix_layout, uplo, n, kd, ab, ldab, ab_t, ldab_t );
116  /* Call LAPACK function and adjust info */
117  LAPACK_chbevx_2stage( &jobz, &range, &uplo, &n, &kd, ab_t, &ldab_t, q_t,
118  &ldq_t, &vl, &vu, &il, &iu, &abstol, m, w, z_t, &ldz_t,
119  work, &lwork, rwork, iwork, ifail, &info );
120  if( info < 0 ) {
121  info = info - 1;
122  }
123  /* Transpose output matrices */
124  LAPACKE_chb_trans( LAPACK_COL_MAJOR, uplo, n, kd, ab_t, ldab_t, ab,
125  ldab );
126  if( LAPACKE_lsame( jobz, 'v' ) ) {
127  LAPACKE_cge_trans( LAPACK_COL_MAJOR, n, n, q_t, ldq_t, q, ldq );
128  }
129  if( LAPACKE_lsame( jobz, 'v' ) ) {
130  LAPACKE_cge_trans( LAPACK_COL_MAJOR, n, ncols_z, z_t, ldz_t, z,
131  ldz );
132  }
133  /* Release memory and exit */
134  if( LAPACKE_lsame( jobz, 'v' ) ) {
135  LAPACKE_free( z_t );
136  }
137 exit_level_2:
138  if( LAPACKE_lsame( jobz, 'v' ) ) {
139  LAPACKE_free( q_t );
140  }
141 exit_level_1:
142  LAPACKE_free( ab_t );
143 exit_level_0:
144  if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
145  LAPACKE_xerbla( "LAPACKE_chbevx_2stage_work", info );
146  }
147  } else {
148  info = -1;
149  LAPACKE_xerbla( "LAPACKE_chbevx_2stage_work", info );
150  }
151  return info;
152 }
LAPACKE_cge_trans
void LAPACKE_cge_trans(int matrix_layout, lapack_int m, lapack_int n, const lapack_complex_float *in, lapack_int ldin, lapack_complex_float *out, lapack_int ldout)
Definition: lapacke_cge_trans.c:40
LAPACK_ROW_MAJOR
#define LAPACK_ROW_MAJOR
Definition: lapacke.h:119
lapack_complex_float
#define lapack_complex_float
Definition: lapacke.h:74
LAPACK_chbevx_2stage
void LAPACK_chbevx_2stage(char *jobz, char *range, char *uplo, lapack_int *n, lapack_int *kd, lapack_complex_float *ab, lapack_int *ldab, lapack_complex_float *q, lapack_int *ldq, float *vl, float *vu, lapack_int *il, lapack_int *iu, float *abstol, lapack_int *m, float *w, lapack_complex_float *z, lapack_int *ldz, lapack_complex_float *work, lapack_int *lwork, float *rwork, lapack_int *iwork, lapack_int *ifail, lapack_int *info)
MAX
#define MAX(x, y)
Definition: lapacke_utils.h:47
LAPACKE_free
#define LAPACKE_free(p)
Definition: lapacke.h:113
LAPACKE_malloc
#define LAPACKE_malloc(size)
Definition: lapacke.h:110
LAPACKE_lsame
lapack_logical LAPACKE_lsame(char ca, char cb)
Definition: lapacke_lsame.c:36
LAPACK_COL_MAJOR
#define LAPACK_COL_MAJOR
Definition: lapacke.h:120
LAPACKE_xerbla
void LAPACKE_xerbla(const char *name, lapack_int info)
Definition: lapacke_xerbla.c:37
lapack_int
#define lapack_int
Definition: lapacke.h:47
LAPACK_TRANSPOSE_MEMORY_ERROR
#define LAPACK_TRANSPOSE_MEMORY_ERROR
Definition: lapacke.h:123
LAPACKE_chb_trans
void LAPACKE_chb_trans(int matrix_layout, char uplo, lapack_int n, lapack_int kd, const lapack_complex_float *in, lapack_int ldin, lapack_complex_float *out, lapack_int ldout)
Definition: lapacke_chb_trans.c:40

Here is the call graph for this function:

Here is the caller graph for this function: