lapack_int LAPACKE_cupmtr( int matrix_order, char side, char uplo, char trans,
lapack_int m, lapack_int n,
const lapack_complex_float* ap,
const lapack_complex_float* tau,
lapack_complex_float* c, lapack_int ldc )
{
lapack_int info = 0;
/* Additional scalars declarations for work arrays */
lapack_int lwork;
lapack_complex_float* work = NULL;
lapack_int r;
if( matrix_order != LAPACK_COL_MAJOR && matrix_order != LAPACK_ROW_MAJOR ) {
LAPACKE_xerbla( "LAPACKE_cupmtr", -1 );
return -1;
}
#ifndef LAPACK_DISABLE_NAN_CHECK
/* Optionally check input matrices for NaNs */
r = LAPACKE_lsame( side, 'l' ) ? m : n;
if( LAPACKE_cpp_nancheck( r, ap ) ) {
return -7;
}
if( LAPACKE_cge_nancheck( matrix_order, m, n, c, ldc ) ) {
return -9;
}
if( LAPACKE_c_nancheck( m-1, tau, 1 ) ) {
return -8;
}
#endif
/* Additional scalars initializations for work arrays */
if( LAPACKE_lsame( side, 'l' ) ) {
lwork = MAX(1,n);
} else if( LAPACKE_lsame( side, 'r' ) ) {
lwork = MAX(1,m);
} else {
lwork = 1; /* Any value */
}
/* Allocate memory for working array(s) */
work = (lapack_complex_float*)
LAPACKE_malloc( sizeof(lapack_complex_float) * lwork );
if( work == NULL ) {
info = LAPACK_WORK_MEMORY_ERROR;
goto exit_level_0;
}
/* Call middle-level interface */
info = LAPACKE_cupmtr_work( matrix_order, side, uplo, trans, m, n, ap, tau,
c, ldc, work );
/* Release memory and exit */
LAPACKE_free( work );
exit_level_0:
if( info == LAPACK_WORK_MEMORY_ERROR ) {
LAPACKE_xerbla( "LAPACKE_cupmtr", info );
}
return info;
}
int main(void)
{
/* Local scalars */
char side, side_i;
char uplo, uplo_i;
char trans, trans_i;
lapack_int m, m_i;
lapack_int n, n_i;
lapack_int ldc, ldc_i;
lapack_int ldc_r;
lapack_int info, info_i;
/* Declare scalars */
lapack_int lwork;
lapack_int i;
int failed;
/* Local arrays */
lapack_complex_float *ap = NULL, *ap_i = NULL;
lapack_complex_float *tau = NULL, *tau_i = NULL;
lapack_complex_float *c = NULL, *c_i = NULL;
lapack_complex_float *work = NULL, *work_i = NULL;
lapack_complex_float *c_save = NULL;
lapack_complex_float *ap_r = NULL;
lapack_complex_float *c_r = NULL;
/* Iniitialize the scalar parameters */
init_scalars_cupmtr( &side, &uplo, &trans, &m, &n, &ldc );
lwork = MAX(m,n);
ldc_r = n+2;
side_i = side;
uplo_i = uplo;
trans_i = trans;
m_i = m;
n_i = n;
ldc_i = ldc;
/* Allocate memory for the LAPACK routine arrays */
ap = (lapack_complex_float *)
LAPACKE_malloc( ((m*(m+1)/2)) * sizeof(lapack_complex_float) );
tau = (lapack_complex_float *)
LAPACKE_malloc( (m-1) * sizeof(lapack_complex_float) );
c = (lapack_complex_float *)
LAPACKE_malloc( ldc*n * sizeof(lapack_complex_float) );
work = (lapack_complex_float *)
LAPACKE_malloc( lwork * sizeof(lapack_complex_float) );
/* Allocate memory for the C interface function arrays */
ap_i = (lapack_complex_float *)
LAPACKE_malloc( ((m*(m+1)/2)) * sizeof(lapack_complex_float) );
tau_i = (lapack_complex_float *)
LAPACKE_malloc( (m-1) * sizeof(lapack_complex_float) );
c_i = (lapack_complex_float *)
LAPACKE_malloc( ldc*n * sizeof(lapack_complex_float) );
work_i = (lapack_complex_float *)
LAPACKE_malloc( lwork * sizeof(lapack_complex_float) );
/* Allocate memory for the backup arrays */
c_save = (lapack_complex_float *)
LAPACKE_malloc( ldc*n * sizeof(lapack_complex_float) );
/* Allocate memory for the row-major arrays */
ap_r = (lapack_complex_float *)
LAPACKE_malloc( m*(m+1)/2 * sizeof(lapack_complex_float) );
c_r = (lapack_complex_float *)
LAPACKE_malloc( m*(n+2) * sizeof(lapack_complex_float) );
/* Initialize input arrays */
init_ap( (m*(m+1)/2), ap );
init_tau( (m-1), tau );
init_c( ldc*n, c );
init_work( lwork, work );
/* Backup the ouptut arrays */
for( i = 0; i < ldc*n; i++ ) {
c_save[i] = c[i];
}
/* Call the LAPACK routine */
cupmtr_( &side, &uplo, &trans, &m, &n, ap, tau, c, &ldc, work, &info );
/* Initialize input data, call the column-major middle-level
* interface to LAPACK routine and check the results */
for( i = 0; i < (m*(m+1)/2); i++ ) {
ap_i[i] = ap[i];
}
for( i = 0; i < (m-1); i++ ) {
tau_i[i] = tau[i];
}
for( i = 0; i < ldc*n; i++ ) {
c_i[i] = c_save[i];
}
for( i = 0; i < lwork; i++ ) {
work_i[i] = work[i];
}
info_i = LAPACKE_cupmtr_work( LAPACK_COL_MAJOR, side_i, uplo_i, trans_i,
m_i, n_i, ap_i, tau_i, c_i, ldc_i, work_i );
failed = compare_cupmtr( c, c_i, info, info_i, ldc, n );
if( failed == 0 ) {
printf( "PASSED: column-major middle-level interface to cupmtr\n" );
} else {
printf( "FAILED: column-major middle-level interface to cupmtr\n" );
}
/* Initialize input data, call the column-major high-level
* interface to LAPACK routine and check the results */
for( i = 0; i < (m*(m+1)/2); i++ ) {
ap_i[i] = ap[i];
}
for( i = 0; i < (m-1); i++ ) {
tau_i[i] = tau[i];
}
for( i = 0; i < ldc*n; i++ ) {
c_i[i] = c_save[i];
}
for( i = 0; i < lwork; i++ ) {
work_i[i] = work[i];
}
info_i = LAPACKE_cupmtr( LAPACK_COL_MAJOR, side_i, uplo_i, trans_i, m_i,
n_i, ap_i, tau_i, c_i, ldc_i );
failed = compare_cupmtr( c, c_i, info, info_i, ldc, n );
if( failed == 0 ) {
printf( "PASSED: column-major high-level interface to cupmtr\n" );
} else {
printf( "FAILED: column-major high-level interface to cupmtr\n" );
}
/* Initialize input data, call the row-major middle-level
* interface to LAPACK routine and check the results */
for( i = 0; i < (m*(m+1)/2); i++ ) {
ap_i[i] = ap[i];
}
for( i = 0; i < (m-1); i++ ) {
tau_i[i] = tau[i];
}
for( i = 0; i < ldc*n; i++ ) {
c_i[i] = c_save[i];
}
for( i = 0; i < lwork; i++ ) {
work_i[i] = work[i];
}
LAPACKE_cpp_trans( LAPACK_COL_MAJOR, uplo, m, ap_i, ap_r );
LAPACKE_cge_trans( LAPACK_COL_MAJOR, m, n, c_i, ldc, c_r, n+2 );
info_i = LAPACKE_cupmtr_work( LAPACK_ROW_MAJOR, side_i, uplo_i, trans_i,
m_i, n_i, ap_r, tau_i, c_r, ldc_r, work_i );
LAPACKE_cge_trans( LAPACK_ROW_MAJOR, m, n, c_r, n+2, c_i, ldc );
failed = compare_cupmtr( c, c_i, info, info_i, ldc, n );
if( failed == 0 ) {
printf( "PASSED: row-major middle-level interface to cupmtr\n" );
} else {
printf( "FAILED: row-major middle-level interface to cupmtr\n" );
}
/* Initialize input data, call the row-major high-level
* interface to LAPACK routine and check the results */
for( i = 0; i < (m*(m+1)/2); i++ ) {
ap_i[i] = ap[i];
}
for( i = 0; i < (m-1); i++ ) {
tau_i[i] = tau[i];
}
for( i = 0; i < ldc*n; i++ ) {
c_i[i] = c_save[i];
}
for( i = 0; i < lwork; i++ ) {
work_i[i] = work[i];
}
/* Init row_major arrays */
LAPACKE_cpp_trans( LAPACK_COL_MAJOR, uplo, m, ap_i, ap_r );
LAPACKE_cge_trans( LAPACK_COL_MAJOR, m, n, c_i, ldc, c_r, n+2 );
info_i = LAPACKE_cupmtr( LAPACK_ROW_MAJOR, side_i, uplo_i, trans_i, m_i,
n_i, ap_r, tau_i, c_r, ldc_r );
LAPACKE_cge_trans( LAPACK_ROW_MAJOR, m, n, c_r, n+2, c_i, ldc );
failed = compare_cupmtr( c, c_i, info, info_i, ldc, n );
if( failed == 0 ) {
printf( "PASSED: row-major high-level interface to cupmtr\n" );
} else {
printf( "FAILED: row-major high-level interface to cupmtr\n" );
}
/* Release memory */
if( ap != NULL ) {
LAPACKE_free( ap );
}
if( ap_i != NULL ) {
LAPACKE_free( ap_i );
}
if( ap_r != NULL ) {
LAPACKE_free( ap_r );
}
if( tau != NULL ) {
LAPACKE_free( tau );
}
if( tau_i != NULL ) {
LAPACKE_free( tau_i );
}
if( c != NULL ) {
LAPACKE_free( c );
}
if( c_i != NULL ) {
LAPACKE_free( c_i );
}
if( c_r != NULL ) {
LAPACKE_free( c_r );
}
if( c_save != NULL ) {
LAPACKE_free( c_save );
}
if( work != NULL ) {
LAPACKE_free( work );
}
if( work_i != NULL ) {
LAPACKE_free( work_i );
}
return 0;
}
