LAPACK  3.8.0
LAPACK: Linear Algebra PACKage

◆ LAPACKE_sgesdd_work()

lapack_int LAPACKE_sgesdd_work ( int  matrix_layout,
char  jobz,
lapack_int  m,
lapack_int  n,
float *  a,
lapack_int  lda,
float *  s,
float *  u,
lapack_int  ldu,
float *  vt,
lapack_int  ldvt,
float *  work,
lapack_int  lwork,
lapack_int iwork 
)

Definition at line 36 of file lapacke_sgesdd_work.c.

41 {
42  lapack_int info = 0;
43  if( matrix_layout == LAPACK_COL_MAJOR ) {
44  /* Call LAPACK function and adjust info */
45  LAPACK_sgesdd( &jobz, &m, &n, a, &lda, s, u, &ldu, vt, &ldvt, work,
46  &lwork, iwork, &info );
47  if( info < 0 ) {
48  info = info - 1;
49  }
50  } else if( matrix_layout == LAPACK_ROW_MAJOR ) {
51  lapack_int nrows_u = ( LAPACKE_lsame( jobz, 'a' ) ||
52  LAPACKE_lsame( jobz, 's' ) ||
53  ( LAPACKE_lsame( jobz, 'o' ) && m<n) ) ? m : 1;
54  lapack_int ncols_u = ( LAPACKE_lsame( jobz, 'a' ) ||
55  ( LAPACKE_lsame( jobz, 'o' ) && m<n) ) ? m :
56  ( LAPACKE_lsame( jobz, 's' ) ? MIN(m,n) : 1);
57  lapack_int nrows_vt = ( LAPACKE_lsame( jobz, 'a' ) ||
58  ( LAPACKE_lsame( jobz, 'o' ) && m>=n) ) ? n :
59  ( LAPACKE_lsame( jobz, 's' ) ? MIN(m,n) : 1);
60  lapack_int lda_t = MAX(1,m);
61  lapack_int ldu_t = MAX(1,nrows_u);
62  lapack_int ldvt_t = MAX(1,nrows_vt);
63  float* a_t = NULL;
64  float* u_t = NULL;
65  float* vt_t = NULL;
66  /* Check leading dimension(s) */
67  if( lda < n ) {
68  info = -6;
69  LAPACKE_xerbla( "LAPACKE_sgesdd_work", info );
70  return info;
71  }
72  if( ldu < ncols_u ) {
73  info = -9;
74  LAPACKE_xerbla( "LAPACKE_sgesdd_work", info );
75  return info;
76  }
77  if( ldvt < n ) {
78  info = -11;
79  LAPACKE_xerbla( "LAPACKE_sgesdd_work", info );
80  return info;
81  }
82  /* Query optimal working array(s) size if requested */
83  if( lwork == -1 ) {
84  LAPACK_sgesdd( &jobz, &m, &n, a, &lda_t, s, u, &ldu_t, vt, &ldvt_t,
85  work, &lwork, iwork, &info );
86  return (info < 0) ? (info - 1) : info;
87  }
88  /* Allocate memory for temporary array(s) */
89  a_t = (float*)LAPACKE_malloc( sizeof(float) * lda_t * MAX(1,n) );
90  if( a_t == NULL ) {
92  goto exit_level_0;
93  }
94  if( LAPACKE_lsame( jobz, 'a' ) || LAPACKE_lsame( jobz, 's' ) ||
95  ( LAPACKE_lsame( jobz, 'o' ) && (m<n) ) ) {
96  u_t = (float*)
97  LAPACKE_malloc( sizeof(float) * ldu_t * MAX(1,ncols_u) );
98  if( u_t == NULL ) {
100  goto exit_level_1;
101  }
102  }
103  if( LAPACKE_lsame( jobz, 'a' ) || LAPACKE_lsame( jobz, 's' ) ||
104  ( LAPACKE_lsame( jobz, 'o' ) && (m>=n) ) ) {
105  vt_t = (float*)LAPACKE_malloc( sizeof(float) * ldvt_t * MAX(1,n) );
106  if( vt_t == NULL ) {
108  goto exit_level_2;
109  }
110  }
111  /* Transpose input matrices */
112  LAPACKE_sge_trans( matrix_layout, m, n, a, lda, a_t, lda_t );
113  /* Call LAPACK function and adjust info */
114  LAPACK_sgesdd( &jobz, &m, &n, a_t, &lda_t, s, u_t, &ldu_t, vt_t,
115  &ldvt_t, work, &lwork, iwork, &info );
116  if( info < 0 ) {
117  info = info - 1;
118  }
119  /* Transpose output matrices */
120  LAPACKE_sge_trans( LAPACK_COL_MAJOR, m, n, a_t, lda_t, a, lda );
121  if( LAPACKE_lsame( jobz, 'a' ) || LAPACKE_lsame( jobz, 's' ) ||
122  ( LAPACKE_lsame( jobz, 'o' ) && (m<n) ) ) {
123  LAPACKE_sge_trans( LAPACK_COL_MAJOR, nrows_u, ncols_u, u_t, ldu_t,
124  u, ldu );
125  }
126  if( LAPACKE_lsame( jobz, 'a' ) || LAPACKE_lsame( jobz, 's' ) ||
127  ( LAPACKE_lsame( jobz, 'o' ) && (m>=n) ) ) {
128  LAPACKE_sge_trans( LAPACK_COL_MAJOR, nrows_vt, n, vt_t, ldvt_t, vt,
129  ldvt );
130  }
131  /* Release memory and exit */
132  if( LAPACKE_lsame( jobz, 'a' ) || LAPACKE_lsame( jobz, 's' ) ||
133  ( LAPACKE_lsame( jobz, 'o' ) && (m>=n) ) ) {
134  LAPACKE_free( vt_t );
135  }
136 exit_level_2:
137  if( LAPACKE_lsame( jobz, 'a' ) || LAPACKE_lsame( jobz, 's' ) ||
138  ( LAPACKE_lsame( jobz, 'o' ) && (m<n) ) ) {
139  LAPACKE_free( u_t );
140  }
141 exit_level_1:
142  LAPACKE_free( a_t );
143 exit_level_0:
144  if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
145  LAPACKE_xerbla( "LAPACKE_sgesdd_work", info );
146  }
147  } else {
148  info = -1;
149  LAPACKE_xerbla( "LAPACKE_sgesdd_work", info );
150  }
151  return info;
152 }
#define LAPACK_ROW_MAJOR
Definition: lapacke.h:119
#define LAPACKE_free(p)
Definition: lapacke.h:113
#define MIN(x, y)
Definition: lapacke_utils.h:50
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
void LAPACK_sgesdd(char *jobz, lapack_int *m, lapack_int *n, float *a, lapack_int *lda, float *s, float *u, lapack_int *ldu, float *vt, lapack_int *ldvt, float *work, lapack_int *lwork, lapack_int *iwork, lapack_int *info)
Here is the call graph for this function:
Here is the caller graph for this function: