LAPACK  3.8.0
LAPACK: Linear Algebra PACKage

◆ LAPACKE_dgesvd_work()

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

Definition at line 36 of file lapacke_dgesvd_work.c.

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