floraachy
/
OpenBLAS
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #1175 from martin-frbg/lapack_143 

Fix workspace computation in LAPACKE ?tpmqrt
This commit is contained in:
Martin Kroeker 2017-05-05 12:00:04 +02:00 committed by GitHub
parent 5649b2c53a d5ea8fd823
commit bba6676803
4 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
lapack-netlib/LAPACKE/src/lapacke_ctpmqrt.c
View File

@ -43,6 +43,7 @@ lapack_int LAPACKE_ctpmqrt( int matrix_layout, char side, char trans,
{ {
lapack_int ncols_a, nrows_a; lapack_int ncols_a, nrows_a;
lapack_int nrows_v; lapack_int nrows_v;
lapack_int lwork;
lapack_int info = 0; lapack_int info = 0;
lapack_complex_float* work = NULL; lapack_complex_float* work = NULL;
if( matrix_layout != LAPACK_COL_MAJOR && matrix_layout != LAPACK_ROW_MAJOR ) { if( matrix_layout != LAPACK_COL_MAJOR && matrix_layout != LAPACK_ROW_MAJOR ) {
@ -71,8 +72,10 @@ lapack_int LAPACKE_ctpmqrt( int matrix_layout, char side, char trans,
} }
#endif #endif
/* Allocate memory for working array(s) */ /* Allocate memory for working array(s) */
lwork = LAPACKE_lsame( side, 'L' ) ? MAX(1,nb) * MAX(1,n) :
( LAPACKE_lsame( side, 'R' ) ? MAX(1,m) * MAX(1,nb) : 0 );
work = (lapack_complex_float*) work = (lapack_complex_float*)
LAPACKE_malloc( sizeof(lapack_complex_float) * MAX(1,m) * MAX(1,nb) ); LAPACKE_malloc( sizeof(lapack_complex_float) * lwork );
if( work == NULL ) { if( work == NULL ) {
info = LAPACK_WORK_MEMORY_ERROR; info = LAPACK_WORK_MEMORY_ERROR;
goto exit_level_0; goto exit_level_0;

5
lapack-netlib/LAPACKE/src/lapacke_dtpmqrt.c
View File

@ -42,6 +42,7 @@ lapack_int LAPACKE_dtpmqrt( int matrix_layout, char side, char trans,
{ {
lapack_int ncols_a, nrows_a; lapack_int ncols_a, nrows_a;
lapack_int nrows_v; lapack_int nrows_v;
lapack_int lwork;
lapack_int info = 0; lapack_int info = 0;
double* work = NULL; double* work = NULL;
if( matrix_layout != LAPACK_COL_MAJOR && matrix_layout != LAPACK_ROW_MAJOR ) { if( matrix_layout != LAPACK_COL_MAJOR && matrix_layout != LAPACK_ROW_MAJOR ) {
@ -70,7 +71,9 @@ lapack_int LAPACKE_dtpmqrt( int matrix_layout, char side, char trans,
} }
#endif #endif
/* Allocate memory for working array(s) */ /* Allocate memory for working array(s) */
work = (double*)LAPACKE_malloc( sizeof(double) * MAX(1,m) * MAX(1,nb) ); lwork = LAPACKE_lsame( side, 'L' ) ? MAX(1,nb) * MAX(1,n) :
( LAPACKE_lsame( side, 'R' ) ? MAX(1,m) * MAX(1,nb) : 0 );
work = (double*)LAPACKE_malloc( sizeof(double) * lwork );
if( work == NULL ) { if( work == NULL ) {
info = LAPACK_WORK_MEMORY_ERROR; info = LAPACK_WORK_MEMORY_ERROR;
goto exit_level_0; goto exit_level_0;

5
lapack-netlib/LAPACKE/src/lapacke_stpmqrt.c
View File

@ -41,6 +41,7 @@ lapack_int LAPACKE_stpmqrt( int matrix_layout, char side, char trans,
{ {
lapack_int ncols_a, nrows_a; lapack_int ncols_a, nrows_a;
lapack_int nrows_v; lapack_int nrows_v;
lapack_int lwork;
lapack_int info = 0; lapack_int info = 0;
float* work = NULL; float* work = NULL;
if( matrix_layout != LAPACK_COL_MAJOR && matrix_layout != LAPACK_ROW_MAJOR ) { if( matrix_layout != LAPACK_COL_MAJOR && matrix_layout != LAPACK_ROW_MAJOR ) {
@ -69,7 +70,9 @@ lapack_int LAPACKE_stpmqrt( int matrix_layout, char side, char trans,
} }
#endif #endif
/* Allocate memory for working array(s) */ /* Allocate memory for working array(s) */
work = (float*)LAPACKE_malloc( sizeof(float) * MAX(1,m) * MAX(1,nb) ); lwork = LAPACKE_lsame( side, 'L' ) ? MAX(1,nb) * MAX(1,n) :
( LAPACKE_lsame( side, 'R' ) ? MAX(1,m) * MAX(1,nb) : 0 );
work = (float*)LAPACKE_malloc( sizeof(float) * lwork );
if( work == NULL ) { if( work == NULL ) {
info = LAPACK_WORK_MEMORY_ERROR; info = LAPACK_WORK_MEMORY_ERROR;
goto exit_level_0; goto exit_level_0;

5
lapack-netlib/LAPACKE/src/lapacke_ztpmqrt.c
View File

@ -43,6 +43,7 @@ lapack_int LAPACKE_ztpmqrt( int matrix_layout, char side, char trans,
{ {
lapack_int ncols_a, nrows_a; lapack_int ncols_a, nrows_a;
lapack_int nrows_v; lapack_int nrows_v;
lapack_int lwork;
lapack_int info = 0; lapack_int info = 0;
lapack_complex_double* work = NULL; lapack_complex_double* work = NULL;
if( matrix_layout != LAPACK_COL_MAJOR && matrix_layout != LAPACK_ROW_MAJOR ) { if( matrix_layout != LAPACK_COL_MAJOR && matrix_layout != LAPACK_ROW_MAJOR ) {
@ -71,8 +72,10 @@ lapack_int LAPACKE_ztpmqrt( int matrix_layout, char side, char trans,
} }
#endif #endif
/* Allocate memory for working array(s) */ /* Allocate memory for working array(s) */
lwork = LAPACKE_lsame( side, 'L' ) ? MAX(1,nb) * MAX(1,n) :
( LAPACKE_lsame( side, 'R' ) ? MAX(1,m) * MAX(1,nb) : 0 );
work = (lapack_complex_double*) work = (lapack_complex_double*)
LAPACKE_malloc( sizeof(lapack_complex_double) * MAX(1,m) * MAX(1,nb) ); LAPACKE_malloc( sizeof(lapack_complex_double) * lwork );
if( work == NULL ) { if( work == NULL ) {
info = LAPACK_WORK_MEMORY_ERROR; info = LAPACK_WORK_MEMORY_ERROR;
goto exit_level_0; goto exit_level_0;