lapack_int LAPACKE_zunmhr( int matrix_layout, char side, char trans,
lapack_int m, lapack_int n, lapack_int ilo,
lapack_int ihi, const lapack_complex_double* a,
lapack_int lda, const lapack_complex_double* tau,
lapack_complex_double* c, lapack_int ldc )
{
lapack_int info = 0;
lapack_int lwork = -1;
lapack_complex_double* work = NULL;
lapack_complex_double work_query;
lapack_int r;
if( matrix_layout != LAPACK_COL_MAJOR && matrix_layout != LAPACK_ROW_MAJOR ) {
LAPACKE_xerbla( "LAPACKE_zunmhr", -1 );
return -1;
}
#ifndef LAPACK_DISABLE_NAN_CHECK
/* Optionally check input matrices for NaNs */
r = LAPACKE_lsame( side, 'l' ) ? m : n;
if( LAPACKE_zge_nancheck( matrix_layout, r, r, a, lda ) ) {
return -8;
}
if( LAPACKE_zge_nancheck( matrix_layout, m, n, c, ldc ) ) {
return -11;
}
if( LAPACKE_z_nancheck( m-1, tau, 1 ) ) {
return -10;
}
#endif
/* Query optimal working array(s) size */
info = LAPACKE_zunmhr_work( matrix_layout, side, trans, m, n, ilo, ihi, a,
lda, tau, c, ldc, &work_query, lwork );
if( info != 0 ) {
goto exit_level_0;
}
lwork = LAPACK_Z2INT( work_query );
/* Allocate memory for work arrays */
work = (lapack_complex_double*)
LAPACKE_malloc( sizeof(lapack_complex_double) * lwork );
if( work == NULL ) {
info = LAPACK_WORK_MEMORY_ERROR;
goto exit_level_0;
}
/* Call middle-level interface */
info = LAPACKE_zunmhr_work( matrix_layout, side, trans, m, n, ilo, ihi, a,
lda, tau, c, ldc, work, lwork );
/* Release memory and exit */
LAPACKE_free( work );
exit_level_0:
if( info == LAPACK_WORK_MEMORY_ERROR ) {
LAPACKE_xerbla( "LAPACKE_zunmhr", info );
}
return info;
}
int main(void)
{
/* Local scalars */
char side, side_i;
char trans, trans_i;
lapack_int m, m_i;
lapack_int n, n_i;
lapack_int ilo, ilo_i;
lapack_int ihi, ihi_i;
lapack_int lda, lda_i;
lapack_int lda_r;
lapack_int ldc, ldc_i;
lapack_int ldc_r;
lapack_int lwork, lwork_i;
lapack_int info, info_i;
/* Declare scalars */
lapack_int r;
lapack_int i;
int failed;
/* Local arrays */
lapack_complex_double *a = NULL, *a_i = NULL;
lapack_complex_double *tau = NULL, *tau_i = NULL;
lapack_complex_double *c = NULL, *c_i = NULL;
lapack_complex_double *work = NULL, *work_i = NULL;
lapack_complex_double *c_save = NULL;
lapack_complex_double *a_r = NULL;
lapack_complex_double *c_r = NULL;
/* Iniitialize the scalar parameters */
init_scalars_zunmhr( &side, &trans, &m, &n, &ilo, &ihi, &lda, &ldc,
&lwork );
r = LAPACKE_lsame( side, 'l' ) ? m : n;
lda_r = r+2;
ldc_r = n+2;
side_i = side;
trans_i = trans;
m_i = m;
n_i = n;
ilo_i = ilo;
ihi_i = ihi;
lda_i = lda;
ldc_i = ldc;
lwork_i = lwork;
/* Allocate memory for the LAPACK routine arrays */
a = (lapack_complex_double *)
LAPACKE_malloc( lda*m * sizeof(lapack_complex_double) );
tau = (lapack_complex_double *)
LAPACKE_malloc( (m-1) * sizeof(lapack_complex_double) );
c = (lapack_complex_double *)
LAPACKE_malloc( ldc*n * sizeof(lapack_complex_double) );
work = (lapack_complex_double *)
LAPACKE_malloc( lwork * sizeof(lapack_complex_double) );
/* Allocate memory for the C interface function arrays */
a_i = (lapack_complex_double *)
LAPACKE_malloc( lda*m * sizeof(lapack_complex_double) );
tau_i = (lapack_complex_double *)
LAPACKE_malloc( (m-1) * sizeof(lapack_complex_double) );
c_i = (lapack_complex_double *)
LAPACKE_malloc( ldc*n * sizeof(lapack_complex_double) );
work_i = (lapack_complex_double *)
LAPACKE_malloc( lwork * sizeof(lapack_complex_double) );
/* Allocate memory for the backup arrays */
c_save = (lapack_complex_double *)
LAPACKE_malloc( ldc*n * sizeof(lapack_complex_double) );
/* Allocate memory for the row-major arrays */
a_r = (lapack_complex_double *)
LAPACKE_malloc( r*(r+2) * sizeof(lapack_complex_double) );
c_r = (lapack_complex_double *)
LAPACKE_malloc( m*(n+2) * sizeof(lapack_complex_double) );
/* Initialize input arrays */
init_a( lda*m, a );
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 */
zunmhr_( &side, &trans, &m, &n, &ilo, &ihi, a, &lda, tau, c, &ldc, work,
&lwork, &info );
/* Initialize input data, call the column-major middle-level
* interface to LAPACK routine and check the results */
for( i = 0; i < lda*m; i++ ) {
a_i[i] = a[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_zunmhr_work( LAPACK_COL_MAJOR, side_i, trans_i, m_i, n_i,
ilo_i, ihi_i, a_i, lda_i, tau_i, c_i, ldc_i,
work_i, lwork_i );
failed = compare_zunmhr( c, c_i, info, info_i, ldc, n );
if( failed == 0 ) {
printf( "PASSED: column-major middle-level interface to zunmhr\n" );
} else {
printf( "FAILED: column-major middle-level interface to zunmhr\n" );
}
/* Initialize input data, call the column-major high-level
* interface to LAPACK routine and check the results */
for( i = 0; i < lda*m; i++ ) {
a_i[i] = a[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_zunmhr( LAPACK_COL_MAJOR, side_i, trans_i, m_i, n_i, ilo_i,
ihi_i, a_i, lda_i, tau_i, c_i, ldc_i );
failed = compare_zunmhr( c, c_i, info, info_i, ldc, n );
if( failed == 0 ) {
printf( "PASSED: column-major high-level interface to zunmhr\n" );
} else {
printf( "FAILED: column-major high-level interface to zunmhr\n" );
}
/* Initialize input data, call the row-major middle-level
* interface to LAPACK routine and check the results */
for( i = 0; i < lda*m; i++ ) {
a_i[i] = a[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_zge_trans( LAPACK_COL_MAJOR, r, r, a_i, lda, a_r, r+2 );
LAPACKE_zge_trans( LAPACK_COL_MAJOR, m, n, c_i, ldc, c_r, n+2 );
info_i = LAPACKE_zunmhr_work( LAPACK_ROW_MAJOR, side_i, trans_i, m_i, n_i,
ilo_i, ihi_i, a_r, lda_r, tau_i, c_r, ldc_r,
work_i, lwork_i );
LAPACKE_zge_trans( LAPACK_ROW_MAJOR, m, n, c_r, n+2, c_i, ldc );
failed = compare_zunmhr( c, c_i, info, info_i, ldc, n );
if( failed == 0 ) {
printf( "PASSED: row-major middle-level interface to zunmhr\n" );
} else {
printf( "FAILED: row-major middle-level interface to zunmhr\n" );
}
/* Initialize input data, call the row-major high-level
* interface to LAPACK routine and check the results */
for( i = 0; i < lda*m; i++ ) {
a_i[i] = a[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_zge_trans( LAPACK_COL_MAJOR, r, r, a_i, lda, a_r, r+2 );
LAPACKE_zge_trans( LAPACK_COL_MAJOR, m, n, c_i, ldc, c_r, n+2 );
info_i = LAPACKE_zunmhr( LAPACK_ROW_MAJOR, side_i, trans_i, m_i, n_i, ilo_i,
ihi_i, a_r, lda_r, tau_i, c_r, ldc_r );
LAPACKE_zge_trans( LAPACK_ROW_MAJOR, m, n, c_r, n+2, c_i, ldc );
failed = compare_zunmhr( c, c_i, info, info_i, ldc, n );
if( failed == 0 ) {
printf( "PASSED: row-major high-level interface to zunmhr\n" );
} else {
printf( "FAILED: row-major high-level interface to zunmhr\n" );
}
/* Release memory */
if( a != NULL ) {
LAPACKE_free( a );
}
if( a_i != NULL ) {
LAPACKE_free( a_i );
}
if( a_r != NULL ) {
LAPACKE_free( a_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;
}
