00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034 #include "lapacke.h"
00035 #include "lapacke_utils.h"
00036
00037 lapack_int LAPACKE_dbdsqr_work( int matrix_order, char uplo, lapack_int n,
00038 lapack_int ncvt, lapack_int nru, lapack_int ncc,
00039 double* d, double* e, double* vt,
00040 lapack_int ldvt, double* u, lapack_int ldu,
00041 double* c, lapack_int ldc, double* work )
00042 {
00043 lapack_int info = 0;
00044 if( matrix_order == LAPACK_COL_MAJOR ) {
00045
00046 LAPACK_dbdsqr( &uplo, &n, &ncvt, &nru, &ncc, d, e, vt, &ldvt, u, &ldu,
00047 c, &ldc, work, &info );
00048 if( info < 0 ) {
00049 info = info - 1;
00050 }
00051 } else if( matrix_order == LAPACK_ROW_MAJOR ) {
00052 lapack_int ldc_t = MAX(1,n);
00053 lapack_int ldu_t = MAX(1,nru);
00054 lapack_int ldvt_t = MAX(1,n);
00055 double* vt_t = NULL;
00056 double* u_t = NULL;
00057 double* c_t = NULL;
00058
00059 if( ldc < ncc ) {
00060 info = -14;
00061 LAPACKE_xerbla( "LAPACKE_dbdsqr_work", info );
00062 return info;
00063 }
00064 if( ldu < n ) {
00065 info = -12;
00066 LAPACKE_xerbla( "LAPACKE_dbdsqr_work", info );
00067 return info;
00068 }
00069 if( ldvt < ncvt ) {
00070 info = -10;
00071 LAPACKE_xerbla( "LAPACKE_dbdsqr_work", info );
00072 return info;
00073 }
00074
00075 if( ncvt != 0 ) {
00076 vt_t = (double*)
00077 LAPACKE_malloc( sizeof(double) * ldvt_t * MAX(1,ncvt) );
00078 if( vt_t == NULL ) {
00079 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00080 goto exit_level_0;
00081 }
00082 }
00083 if( nru != 0 ) {
00084 u_t = (double*)LAPACKE_malloc( sizeof(double) * ldu_t * MAX(1,n) );
00085 if( u_t == NULL ) {
00086 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00087 goto exit_level_1;
00088 }
00089 }
00090 if( ncc != 0 ) {
00091 c_t = (double*)
00092 LAPACKE_malloc( sizeof(double) * ldc_t * MAX(1,ncc) );
00093 if( c_t == NULL ) {
00094 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00095 goto exit_level_2;
00096 }
00097 }
00098
00099 if( ncvt != 0 ) {
00100 LAPACKE_dge_trans( matrix_order, n, ncvt, vt, ldvt, vt_t, ldvt_t );
00101 }
00102 if( nru != 0 ) {
00103 LAPACKE_dge_trans( matrix_order, nru, n, u, ldu, u_t, ldu_t );
00104 }
00105 if( ncc != 0 ) {
00106 LAPACKE_dge_trans( matrix_order, n, ncc, c, ldc, c_t, ldc_t );
00107 }
00108
00109 LAPACK_dbdsqr( &uplo, &n, &ncvt, &nru, &ncc, d, e, vt_t, &ldvt_t, u_t,
00110 &ldu_t, c_t, &ldc_t, work, &info );
00111 if( info < 0 ) {
00112 info = info - 1;
00113 }
00114
00115 if( ncvt != 0 ) {
00116 LAPACKE_dge_trans( LAPACK_COL_MAJOR, n, ncvt, vt_t, ldvt_t, vt,
00117 ldvt );
00118 }
00119 if( nru != 0 ) {
00120 LAPACKE_dge_trans( LAPACK_COL_MAJOR, nru, n, u_t, ldu_t, u, ldu );
00121 }
00122 if( ncc != 0 ) {
00123 LAPACKE_dge_trans( LAPACK_COL_MAJOR, n, ncc, c_t, ldc_t, c, ldc );
00124 }
00125
00126 if( ncc != 0 ) {
00127 LAPACKE_free( c_t );
00128 }
00129 exit_level_2:
00130 if( nru != 0 ) {
00131 LAPACKE_free( u_t );
00132 }
00133 exit_level_1:
00134 if( ncvt != 0 ) {
00135 LAPACKE_free( vt_t );
00136 }
00137 exit_level_0:
00138 if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
00139 LAPACKE_xerbla( "LAPACKE_dbdsqr_work", info );
00140 }
00141 } else {
00142 info = -1;
00143 LAPACKE_xerbla( "LAPACKE_dbdsqr_work", info );
00144 }
00145 return info;
00146 }