LAPACK  3.8.0
LAPACK: Linear Algebra PACKage

◆ LAPACKE_spteqr_work()

lapack_int LAPACKE_spteqr_work ( int  matrix_layout,
char  compz,
lapack_int  n,
float *  d,
float *  e,
float *  z,
lapack_int  ldz,
float *  work 
)

Definition at line 36 of file lapacke_spteqr_work.c.

39 {
40  lapack_int info = 0;
41  if( matrix_layout == LAPACK_COL_MAJOR ) {
42  /* Call LAPACK function and adjust info */
43  LAPACK_spteqr( &compz, &n, d, e, z, &ldz, work, &info );
44  if( info < 0 ) {
45  info = info - 1;
46  }
47  } else if( matrix_layout == LAPACK_ROW_MAJOR ) {
48  lapack_int ldz_t = MAX(1,n);
49  float* z_t = NULL;
50  /* Check leading dimension(s) */
51  if( ldz < n ) {
52  info = -7;
53  LAPACKE_xerbla( "LAPACKE_spteqr_work", info );
54  return info;
55  }
56  /* Allocate memory for temporary array(s) */
57  z_t = (float*)LAPACKE_malloc( sizeof(float) * ldz_t * MAX(1,n) );
58  if( z_t == NULL ) {
60  goto exit_level_0;
61  }
62  /* Transpose input matrices */
63  if( LAPACKE_lsame( compz, 'v' ) ) {
64  LAPACKE_sge_trans( matrix_layout, n, n, z, ldz, z_t, ldz_t );
65  }
66  /* Call LAPACK function and adjust info */
67  LAPACK_spteqr( &compz, &n, d, e, z_t, &ldz_t, work, &info );
68  if( info < 0 ) {
69  info = info - 1;
70  }
71  /* Transpose output matrices */
72  LAPACKE_sge_trans( LAPACK_COL_MAJOR, n, n, z_t, ldz_t, z, ldz );
73  /* Release memory and exit */
74  LAPACKE_free( z_t );
75 exit_level_0:
76  if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
77  LAPACKE_xerbla( "LAPACKE_spteqr_work", info );
78  }
79  } else {
80  info = -1;
81  LAPACKE_xerbla( "LAPACKE_spteqr_work", info );
82  }
83  return info;
84 }
#define LAPACK_ROW_MAJOR
Definition: lapacke.h:119
void LAPACK_spteqr(char *compz, lapack_int *n, float *d, float *e, float *z, lapack_int *ldz, float *work, lapack_int *info)
#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
#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: