LAPACK  3.8.0
LAPACK: Linear Algebra PACKage

◆ LAPACKE_zgesvd_work()

lapack_int LAPACKE_zgesvd_work ( int  matrix_layout,
char  jobu,
char  jobvt,
lapack_int  m,
lapack_int  n,
lapack_complex_double a,
lapack_int  lda,
double *  s,
lapack_complex_double u,
lapack_int  ldu,
lapack_complex_double vt,
lapack_int  ldvt,
lapack_complex_double work,
lapack_int  lwork,
double *  rwork 
)

Definition at line 36 of file lapacke_zgesvd_work.c.

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