LAPACK  3.8.0
LAPACK: Linear Algebra PACKage

◆ LAPACKE_ssbev_2stage_work()

lapack_int LAPACKE_ssbev_2stage_work ( int  matrix_layout,
char  jobz,
char  uplo,
lapack_int  n,
lapack_int  kd,
float *  ab,
lapack_int  ldab,
float *  w,
float *  z,
lapack_int  ldz,
float *  work,
lapack_int  lwork 
)

Definition at line 36 of file lapacke_ssbev_2stage_work.c.

40 {
41  lapack_int info = 0;
42  if( matrix_layout == LAPACK_COL_MAJOR ) {
43  /* Call LAPACK function and adjust info */
44  LAPACK_ssbev_2stage( &jobz, &uplo, &n, &kd, ab, &ldab, w, z, &ldz, work,
45  &lwork, &info );
46  if( info < 0 ) {
47  info = info - 1;
48  }
49  } else if( matrix_layout == LAPACK_ROW_MAJOR ) {
50  lapack_int ldab_t = MAX(1,kd+1);
51  lapack_int ldz_t = MAX(1,n);
52  float* ab_t = NULL;
53  float* z_t = NULL;
54  /* Check leading dimension(s) */
55  if( ldab < n ) {
56  info = -7;
57  LAPACKE_xerbla( "LAPACKE_ssbev_2stage_work", info );
58  return info;
59  }
60  if( ldz < n ) {
61  info = -10;
62  LAPACKE_xerbla( "LAPACKE_ssbev_2stage_work", info );
63  return info;
64  }
65  /* Query optimal working array(s) size if requested */
66  if( lwork == -1 ) {
67  LAPACK_ssbev_2stage( &jobz, &uplo, &n, &kd, ab_t, &ldab_t, w, z_t, &ldz_t,
68  work, &lwork, &info );
69  return (info < 0) ? (info - 1) : info;
70  }
71  /* Allocate memory for temporary array(s) */
72  ab_t = (float*)LAPACKE_malloc( sizeof(float) * ldab_t * MAX(1,n) );
73  if( ab_t == NULL ) {
75  goto exit_level_0;
76  }
77  if( LAPACKE_lsame( jobz, 'v' ) ) {
78  z_t = (float*)LAPACKE_malloc( sizeof(float) * ldz_t * MAX(1,n) );
79  if( z_t == NULL ) {
81  goto exit_level_1;
82  }
83  }
84  /* Transpose input matrices */
85  LAPACKE_ssb_trans( matrix_layout, uplo, n, kd, ab, ldab, ab_t, ldab_t );
86  /* Call LAPACK function and adjust info */
87  LAPACK_ssbev_2stage( &jobz, &uplo, &n, &kd, ab_t, &ldab_t, w, z_t, &ldz_t,
88  work, &lwork, &info );
89  if( info < 0 ) {
90  info = info - 1;
91  }
92  /* Transpose output matrices */
93  LAPACKE_ssb_trans( LAPACK_COL_MAJOR, uplo, n, kd, ab_t, ldab_t, ab,
94  ldab );
95  if( LAPACKE_lsame( jobz, 'v' ) ) {
96  LAPACKE_sge_trans( LAPACK_COL_MAJOR, n, n, z_t, ldz_t, z, ldz );
97  }
98  /* Release memory and exit */
99  if( LAPACKE_lsame( jobz, 'v' ) ) {
100  LAPACKE_free( z_t );
101  }
102 exit_level_1:
103  LAPACKE_free( ab_t );
104 exit_level_0:
105  if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
106  LAPACKE_xerbla( "LAPACKE_ssbev_2stage_work", info );
107  }
108  } else {
109  info = -1;
110  LAPACKE_xerbla( "LAPACKE_ssbev_2stage_work", info );
111  }
112  return info;
113 }
#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)
void LAPACK_ssbev_2stage(char *jobz, char *uplo, lapack_int *n, lapack_int *kd, float *ab, lapack_int *ldab, float *w, float *z, lapack_int *ldz, float *work, lapack_int *lwork, 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: