Add interfaces for [cz]unhr_col and [sd]orhr_col (Reference-LAPACK PR 827)
This commit is contained in:
parent
bc967e7709
commit
4f10899f5a
|
@ -0,0 +1,24 @@
|
|||
#include "lapacke_utils.h"
|
||||
|
||||
lapack_int LAPACKE_cunhr_col( int matrix_layout, lapack_int m, lapack_int n,
|
||||
lapack_int nb, lapack_complex_float* a,
|
||||
lapack_int lda, lapack_complex_float* t,
|
||||
lapack_int ldt, lapack_complex_float* d)
|
||||
{
|
||||
lapack_int info = 0;
|
||||
if( matrix_layout != LAPACK_COL_MAJOR && matrix_layout != LAPACK_ROW_MAJOR ) {
|
||||
LAPACKE_xerbla( "LAPACKE_cunhr_col", -1 );
|
||||
return -1;
|
||||
}
|
||||
#ifndef LAPACK_DISABLE_NAN_CHECK
|
||||
if( LAPACKE_get_nancheck() ) {
|
||||
/* Optionally check input matrices for NaNs */
|
||||
if( LAPACKE_cge_nancheck( matrix_layout, m, n, a, lda ) ) {
|
||||
return -5;
|
||||
}
|
||||
}
|
||||
#endif
|
||||
/* Call middle-level interface */
|
||||
info = LAPACKE_cunhr_col_work( matrix_layout, m, n, nb, a, lda, t, ldt, d );
|
||||
return info;
|
||||
}
|
|
@ -0,0 +1,69 @@
|
|||
#include "lapacke_utils.h"
|
||||
|
||||
lapack_int LAPACKE_cunhr_col_work( int matrix_layout, lapack_int m, lapack_int n,
|
||||
lapack_int nb, lapack_complex_float* a,
|
||||
lapack_int lda, lapack_complex_float* t,
|
||||
lapack_int ldt, lapack_complex_float* d )
|
||||
{
|
||||
lapack_int info = 0;
|
||||
if( matrix_layout == LAPACK_COL_MAJOR ) {
|
||||
/* Call LAPACK function and adjust info */
|
||||
LAPACK_cunhr_col( &m, &n, &nb, a, &lda, t, &ldt, d, &info );
|
||||
if( info < 0 ) {
|
||||
info = info - 1;
|
||||
}
|
||||
} else if( matrix_layout == LAPACK_ROW_MAJOR ) {
|
||||
lapack_int lda_t = MAX(1,m);
|
||||
lapack_int ldt_t = MAX(1,MIN(nb,n));
|
||||
lapack_complex_float* a_t = NULL;
|
||||
lapack_complex_float* t_t = NULL;
|
||||
/* Check leading dimension(s) */
|
||||
if( lda < n ) {
|
||||
info = -6;
|
||||
LAPACKE_xerbla( "LAPACKE_cunhr_col_work", info );
|
||||
return info;
|
||||
}
|
||||
if( ldt < n ) {
|
||||
info = -8;
|
||||
LAPACKE_xerbla( "LAPACKE_cunhr_col_work", info );
|
||||
return info;
|
||||
}
|
||||
/* Allocate memory for temporary array(s) */
|
||||
a_t = (lapack_complex_float*)
|
||||
LAPACKE_malloc( sizeof(lapack_complex_float) * lda_t * MAX(1,n) );
|
||||
if( a_t == NULL ) {
|
||||
info = LAPACK_TRANSPOSE_MEMORY_ERROR;
|
||||
goto exit_level_0;
|
||||
}
|
||||
t_t = (lapack_complex_float*)
|
||||
LAPACKE_malloc( sizeof(lapack_complex_float) *
|
||||
ldt_t * MAX(1,n) );
|
||||
if( t_t == NULL ) {
|
||||
info = LAPACK_TRANSPOSE_MEMORY_ERROR;
|
||||
goto exit_level_1;
|
||||
}
|
||||
/* Transpose input matrices */
|
||||
LAPACKE_cge_trans( matrix_layout, m, n, a, lda, a_t, lda_t );
|
||||
/* Call LAPACK function and adjust info */
|
||||
LAPACK_cunhr_col( &m, &n, &nb, a_t, &lda_t, t_t, &ldt_t, d, &info );
|
||||
if( info < 0 ) {
|
||||
info = info - 1;
|
||||
}
|
||||
/* Transpose output matrices */
|
||||
LAPACKE_cge_trans( LAPACK_COL_MAJOR, m, n, a_t, lda_t, a, lda );
|
||||
LAPACKE_cge_trans( LAPACK_COL_MAJOR, ldt, n, t_t, ldt_t, t,
|
||||
ldt );
|
||||
/* Release memory and exit */
|
||||
LAPACKE_free( t_t );
|
||||
exit_level_1:
|
||||
LAPACKE_free( a_t );
|
||||
exit_level_0:
|
||||
if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
|
||||
LAPACKE_xerbla( "LAPACKE_cunhr_col_work", info );
|
||||
}
|
||||
} else {
|
||||
info = -1;
|
||||
LAPACKE_xerbla( "LAPACKE_cunhr_col_work", info );
|
||||
}
|
||||
return info;
|
||||
}
|
|
@ -0,0 +1,24 @@
|
|||
#include "lapacke_utils.h"
|
||||
|
||||
lapack_int LAPACKE_dorhr_col( int matrix_layout, lapack_int m, lapack_int n,
|
||||
lapack_int nb, double* a,
|
||||
lapack_int lda, double* t,
|
||||
lapack_int ldt, double* d)
|
||||
{
|
||||
lapack_int info = 0;
|
||||
if( matrix_layout != LAPACK_COL_MAJOR && matrix_layout != LAPACK_ROW_MAJOR ) {
|
||||
LAPACKE_xerbla( "LAPACKE_dorhr_col", -1 );
|
||||
return -1;
|
||||
}
|
||||
#ifndef LAPACK_DISABLE_NAN_CHECK
|
||||
if( LAPACKE_get_nancheck() ) {
|
||||
/* Optionally check input matrices for NaNs */
|
||||
if( LAPACKE_dge_nancheck( matrix_layout, m, n, a, lda ) ) {
|
||||
return -5;
|
||||
}
|
||||
}
|
||||
#endif
|
||||
/* Call middle-level interface */
|
||||
info = LAPACKE_dorhr_col_work( matrix_layout, m, n, nb, a, lda, t, ldt, d );
|
||||
return info;
|
||||
}
|
|
@ -0,0 +1,69 @@
|
|||
#include "lapacke_utils.h"
|
||||
|
||||
lapack_int LAPACKE_dorhr_col_work( int matrix_layout, lapack_int m, lapack_int n,
|
||||
lapack_int nb, double* a,
|
||||
lapack_int lda, double* t,
|
||||
lapack_int ldt, double* d )
|
||||
{
|
||||
lapack_int info = 0;
|
||||
if( matrix_layout == LAPACK_COL_MAJOR ) {
|
||||
/* Call LAPACK function and adjust info */
|
||||
LAPACK_dorhr_col( &m, &n, &nb, a, &lda, t, &ldt, d, &info );
|
||||
if( info < 0 ) {
|
||||
info = info - 1;
|
||||
}
|
||||
} else if( matrix_layout == LAPACK_ROW_MAJOR ) {
|
||||
lapack_int lda_t = MAX(1,m);
|
||||
lapack_int ldt_t = MAX(1,MIN(nb,n));
|
||||
double* a_t = NULL;
|
||||
double* t_t = NULL;
|
||||
/* Check leading dimension(s) */
|
||||
if( lda < n ) {
|
||||
info = -6;
|
||||
LAPACKE_xerbla( "LAPACKE_dorhr_col_work", info );
|
||||
return info;
|
||||
}
|
||||
if( ldt < n ) {
|
||||
info = -8;
|
||||
LAPACKE_xerbla( "LAPACKE_dorhr_col_work", info );
|
||||
return info;
|
||||
}
|
||||
/* Allocate memory for temporary array(s) */
|
||||
a_t = (double*)
|
||||
LAPACKE_malloc( sizeof(double) * lda_t * MAX(1,n) );
|
||||
if( a_t == NULL ) {
|
||||
info = LAPACK_TRANSPOSE_MEMORY_ERROR;
|
||||
goto exit_level_0;
|
||||
}
|
||||
t_t = (double*)
|
||||
LAPACKE_malloc( sizeof(double) *
|
||||
ldt_t * MAX(1,n) );
|
||||
if( t_t == NULL ) {
|
||||
info = LAPACK_TRANSPOSE_MEMORY_ERROR;
|
||||
goto exit_level_1;
|
||||
}
|
||||
/* Transpose input matrices */
|
||||
LAPACKE_dge_trans( matrix_layout, m, n, a, lda, a_t, lda_t );
|
||||
/* Call LAPACK function and adjust info */
|
||||
LAPACK_dorhr_col( &m, &n, &nb, a_t, &lda_t, t_t, &ldt_t, d, &info );
|
||||
if( info < 0 ) {
|
||||
info = info - 1;
|
||||
}
|
||||
/* Transpose output matrices */
|
||||
LAPACKE_dge_trans( LAPACK_COL_MAJOR, m, n, a_t, lda_t, a, lda );
|
||||
LAPACKE_dge_trans( LAPACK_COL_MAJOR, ldt, n, t_t, ldt_t, t,
|
||||
ldt );
|
||||
/* Release memory and exit */
|
||||
LAPACKE_free( t_t );
|
||||
exit_level_1:
|
||||
LAPACKE_free( a_t );
|
||||
exit_level_0:
|
||||
if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
|
||||
LAPACKE_xerbla( "LAPACKE_dorhr_col_work", info );
|
||||
}
|
||||
} else {
|
||||
info = -1;
|
||||
LAPACKE_xerbla( "LAPACKE_dorhr_col_work", info );
|
||||
}
|
||||
return info;
|
||||
}
|
|
@ -0,0 +1,24 @@
|
|||
#include "lapacke_utils.h"
|
||||
|
||||
lapack_int LAPACKE_sorhr_col( int matrix_layout, lapack_int m, lapack_int n,
|
||||
lapack_int nb, float* a,
|
||||
lapack_int lda, float* t,
|
||||
lapack_int ldt, float* d)
|
||||
{
|
||||
lapack_int info = 0;
|
||||
if( matrix_layout != LAPACK_COL_MAJOR && matrix_layout != LAPACK_ROW_MAJOR ) {
|
||||
LAPACKE_xerbla( "LAPACKE_sorhr_col", -1 );
|
||||
return -1;
|
||||
}
|
||||
#ifndef LAPACK_DISABLE_NAN_CHECK
|
||||
if( LAPACKE_get_nancheck() ) {
|
||||
/* Optionally check input matrices for NaNs */
|
||||
if( LAPACKE_sge_nancheck( matrix_layout, m, n, a, lda ) ) {
|
||||
return -5;
|
||||
}
|
||||
}
|
||||
#endif
|
||||
/* Call middle-level interface */
|
||||
info = LAPACKE_sorhr_col_work( matrix_layout, m, n, nb, a, lda, t, ldt, d );
|
||||
return info;
|
||||
}
|
|
@ -0,0 +1,69 @@
|
|||
#include "lapacke_utils.h"
|
||||
|
||||
lapack_int LAPACKE_sorhr_col_work( int matrix_layout, lapack_int m, lapack_int n,
|
||||
lapack_int nb, float* a,
|
||||
lapack_int lda, float* t,
|
||||
lapack_int ldt, float* d )
|
||||
{
|
||||
lapack_int info = 0;
|
||||
if( matrix_layout == LAPACK_COL_MAJOR ) {
|
||||
/* Call LAPACK function and adjust info */
|
||||
LAPACK_sorhr_col( &m, &n, &nb, a, &lda, t, &ldt, d, &info );
|
||||
if( info < 0 ) {
|
||||
info = info - 1;
|
||||
}
|
||||
} else if( matrix_layout == LAPACK_ROW_MAJOR ) {
|
||||
lapack_int lda_t = MAX(1,m);
|
||||
lapack_int ldt_t = MAX(1,MIN(nb,n));
|
||||
float* a_t = NULL;
|
||||
float* t_t = NULL;
|
||||
/* Check leading dimension(s) */
|
||||
if( lda < n ) {
|
||||
info = -6;
|
||||
LAPACKE_xerbla( "LAPACKE_sorhr_col_work", info );
|
||||
return info;
|
||||
}
|
||||
if( ldt < n ) {
|
||||
info = -8;
|
||||
LAPACKE_xerbla( "LAPACKE_sorhr_col_work", info );
|
||||
return info;
|
||||
}
|
||||
/* Allocate memory for temporary array(s) */
|
||||
a_t = (float*)
|
||||
LAPACKE_malloc( sizeof(float) * lda_t * MAX(1,n) );
|
||||
if( a_t == NULL ) {
|
||||
info = LAPACK_TRANSPOSE_MEMORY_ERROR;
|
||||
goto exit_level_0;
|
||||
}
|
||||
t_t = (float*)
|
||||
LAPACKE_malloc( sizeof(float) *
|
||||
ldt_t * MAX(1,n) );
|
||||
if( t_t == NULL ) {
|
||||
info = LAPACK_TRANSPOSE_MEMORY_ERROR;
|
||||
goto exit_level_1;
|
||||
}
|
||||
/* Transpose input matrices */
|
||||
LAPACKE_sge_trans( matrix_layout, m, n, a, lda, a_t, lda_t );
|
||||
/* Call LAPACK function and adjust info */
|
||||
LAPACK_sorhr_col( &m, &n, &nb, a_t, &lda_t, t_t, &ldt_t, d, &info );
|
||||
if( info < 0 ) {
|
||||
info = info - 1;
|
||||
}
|
||||
/* Transpose output matrices */
|
||||
LAPACKE_sge_trans( LAPACK_COL_MAJOR, m, n, a_t, lda_t, a, lda );
|
||||
LAPACKE_sge_trans( LAPACK_COL_MAJOR, ldt, n, t_t, ldt_t, t,
|
||||
ldt );
|
||||
/* Release memory and exit */
|
||||
LAPACKE_free( t_t );
|
||||
exit_level_1:
|
||||
LAPACKE_free( a_t );
|
||||
exit_level_0:
|
||||
if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
|
||||
LAPACKE_xerbla( "LAPACKE_sorhr_col_work", info );
|
||||
}
|
||||
} else {
|
||||
info = -1;
|
||||
LAPACKE_xerbla( "LAPACKE_sorhr_col_work", info );
|
||||
}
|
||||
return info;
|
||||
}
|
|
@ -0,0 +1,24 @@
|
|||
#include "lapacke_utils.h"
|
||||
|
||||
lapack_int LAPACKE_zunhr_col( int matrix_layout, lapack_int m, lapack_int n,
|
||||
lapack_int nb, lapack_complex_double* a,
|
||||
lapack_int lda, lapack_complex_double* t,
|
||||
lapack_int ldt, lapack_complex_double* d)
|
||||
{
|
||||
lapack_int info = 0;
|
||||
if( matrix_layout != LAPACK_COL_MAJOR && matrix_layout != LAPACK_ROW_MAJOR ) {
|
||||
LAPACKE_xerbla( "LAPACKE_zunhr_col", -1 );
|
||||
return -1;
|
||||
}
|
||||
#ifndef LAPACK_DISABLE_NAN_CHECK
|
||||
if( LAPACKE_get_nancheck() ) {
|
||||
/* Optionally check input matrices for NaNs */
|
||||
if( LAPACKE_zge_nancheck( matrix_layout, m, n, a, lda ) ) {
|
||||
return -5;
|
||||
}
|
||||
}
|
||||
#endif
|
||||
/* Call middle-level interface */
|
||||
info = LAPACKE_zunhr_col_work( matrix_layout, m, n, nb, a, lda, t, ldt, d );
|
||||
return info;
|
||||
}
|
|
@ -0,0 +1,69 @@
|
|||
#include "lapacke_utils.h"
|
||||
|
||||
lapack_int LAPACKE_zunhr_col_work( int matrix_layout, lapack_int m, lapack_int n,
|
||||
lapack_int nb, lapack_complex_double* a,
|
||||
lapack_int lda, lapack_complex_double* t,
|
||||
lapack_int ldt, lapack_complex_double* d )
|
||||
{
|
||||
lapack_int info = 0;
|
||||
if( matrix_layout == LAPACK_COL_MAJOR ) {
|
||||
/* Call LAPACK function and adjust info */
|
||||
LAPACK_zunhr_col( &m, &n, &nb, a, &lda, t, &ldt, d, &info );
|
||||
if( info < 0 ) {
|
||||
info = info - 1;
|
||||
}
|
||||
} else if( matrix_layout == LAPACK_ROW_MAJOR ) {
|
||||
lapack_int lda_t = MAX(1,m);
|
||||
lapack_int ldt_t = MAX(1,MIN(nb,n));
|
||||
lapack_complex_double* a_t = NULL;
|
||||
lapack_complex_double* t_t = NULL;
|
||||
/* Check leading dimension(s) */
|
||||
if( lda < n ) {
|
||||
info = -6;
|
||||
LAPACKE_xerbla( "LAPACKE_zunhr_col_work", info );
|
||||
return info;
|
||||
}
|
||||
if( ldt < n ) {
|
||||
info = -8;
|
||||
LAPACKE_xerbla( "LAPACKE_zunhr_col_work", info );
|
||||
return info;
|
||||
}
|
||||
/* Allocate memory for temporary array(s) */
|
||||
a_t = (lapack_complex_double*)
|
||||
LAPACKE_malloc( sizeof(lapack_complex_double) * lda_t * MAX(1,n) );
|
||||
if( a_t == NULL ) {
|
||||
info = LAPACK_TRANSPOSE_MEMORY_ERROR;
|
||||
goto exit_level_0;
|
||||
}
|
||||
t_t = (lapack_complex_double*)
|
||||
LAPACKE_malloc( sizeof(lapack_complex_double) *
|
||||
ldt_t * MAX(1,n) );
|
||||
if( t_t == NULL ) {
|
||||
info = LAPACK_TRANSPOSE_MEMORY_ERROR;
|
||||
goto exit_level_1;
|
||||
}
|
||||
/* Transpose input matrices */
|
||||
LAPACKE_zge_trans( matrix_layout, m, n, a, lda, a_t, lda_t );
|
||||
/* Call LAPACK function and adjust info */
|
||||
LAPACK_zunhr_col( &m, &n, &nb, a_t, &lda_t, t_t, &ldt_t, d, &info );
|
||||
if( info < 0 ) {
|
||||
info = info - 1;
|
||||
}
|
||||
/* Transpose output matrices */
|
||||
LAPACKE_zge_trans( LAPACK_COL_MAJOR, m, n, a_t, lda_t, a, lda );
|
||||
LAPACKE_zge_trans( LAPACK_COL_MAJOR, ldt, n, t_t, ldt_t, t,
|
||||
ldt );
|
||||
/* Release memory and exit */
|
||||
LAPACKE_free( t_t );
|
||||
exit_level_1:
|
||||
LAPACKE_free( a_t );
|
||||
exit_level_0:
|
||||
if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
|
||||
LAPACKE_xerbla( "LAPACKE_zunhr_col_work", info );
|
||||
}
|
||||
} else {
|
||||
info = -1;
|
||||
LAPACKE_xerbla( "LAPACKE_zunhr_col_work", info );
|
||||
}
|
||||
return info;
|
||||
}
|
Loading…
Reference in New Issue