LAPACK  3.8.0
LAPACK: Linear Algebra PACKage

◆ LAPACKE_ssbevx_2stage_work()

lapack_int LAPACKE_ssbevx_2stage_work ( int  matrix_layout,
char  jobz,
char  range,
char  uplo,
lapack_int  n,
lapack_int  kd,
float *  ab,
lapack_int  ldab,
float *  q,
lapack_int  ldq,
float  vl,
float  vu,
lapack_int  il,
lapack_int  iu,
float  abstol,
lapack_int m,
float *  w,
float *  z,
lapack_int  ldz,
float *  work,
lapack_int  lwork,
lapack_int iwork,
lapack_int ifail 
)

Definition at line 36 of file lapacke_ssbevx_2stage_work.c.

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