lapack_int LAPACKE_dtrexc( int matrix_layout, char compq, lapack_int n,
double* t, lapack_int ldt, double* q, lapack_int ldq,
lapack_int* ifst, lapack_int* ilst )
{
lapack_int info = 0;
double* work = NULL;
if( matrix_layout != LAPACK_COL_MAJOR && matrix_layout != LAPACK_ROW_MAJOR ) {
LAPACKE_xerbla( "LAPACKE_dtrexc", -1 );
return -1;
}
#ifndef LAPACK_DISABLE_NAN_CHECK
/* Optionally check input matrices for NaNs */
if( LAPACKE_lsame( compq, 'v' ) ) {
if( LAPACKE_dge_nancheck( matrix_layout, n, n, q, ldq ) ) {
return -6;
}
}
if( LAPACKE_dge_nancheck( matrix_layout, n, n, t, ldt ) ) {
return -4;
}
#endif
/* Allocate memory for working array(s) */
work = (double*)LAPACKE_malloc( sizeof(double) * MAX(1,n) );
if( work == NULL ) {
info = LAPACK_WORK_MEMORY_ERROR;
goto exit_level_0;
}
/* Call middle-level interface */
info = LAPACKE_dtrexc_work( matrix_layout, compq, n, t, ldt, q, ldq, ifst,
ilst, work );
/* Release memory and exit */
LAPACKE_free( work );
exit_level_0:
if( info == LAPACK_WORK_MEMORY_ERROR ) {
LAPACKE_xerbla( "LAPACKE_dtrexc", info );
}
return info;
}
int main(void)
{
/* Local scalars */
char compq, compq_i;
lapack_int n, n_i;
lapack_int ldt, ldt_i;
lapack_int ldt_r;
lapack_int ldq, ldq_i;
lapack_int ldq_r;
lapack_int ifst, ifst_i, ifst_save;
lapack_int ilst, ilst_i, ilst_save;
lapack_int info, info_i;
lapack_int i;
int failed;
/* Local arrays */
double *t = NULL, *t_i = NULL;
double *q = NULL, *q_i = NULL;
double *work = NULL, *work_i = NULL;
double *t_save = NULL;
double *q_save = NULL;
double *t_r = NULL;
double *q_r = NULL;
/* Iniitialize the scalar parameters */
init_scalars_dtrexc( &compq, &n, &ldt, &ldq, &ifst, &ilst );
ldt_r = n+2;
ldq_r = n+2;
compq_i = compq;
n_i = n;
ldt_i = ldt;
ldq_i = ldq;
ifst_i = ifst_save = ifst;
ilst_i = ilst_save = ilst;
/* Allocate memory for the LAPACK routine arrays */
t = (double *)LAPACKE_malloc( ldt*n * sizeof(double) );
q = (double *)LAPACKE_malloc( ldq*n * sizeof(double) );
work = (double *)LAPACKE_malloc( n * sizeof(double) );
/* Allocate memory for the C interface function arrays */
t_i = (double *)LAPACKE_malloc( ldt*n * sizeof(double) );
q_i = (double *)LAPACKE_malloc( ldq*n * sizeof(double) );
work_i = (double *)LAPACKE_malloc( n * sizeof(double) );
/* Allocate memory for the backup arrays */
t_save = (double *)LAPACKE_malloc( ldt*n * sizeof(double) );
q_save = (double *)LAPACKE_malloc( ldq*n * sizeof(double) );
/* Allocate memory for the row-major arrays */
t_r = (double *)LAPACKE_malloc( n*(n+2) * sizeof(double) );
q_r = (double *)LAPACKE_malloc( n*(n+2) * sizeof(double) );
/* Initialize input arrays */
init_t( ldt*n, t );
init_q( ldq*n, q );
init_work( n, work );
/* Backup the ouptut arrays */
for( i = 0; i < ldt*n; i++ ) {
t_save[i] = t[i];
}
for( i = 0; i < ldq*n; i++ ) {
q_save[i] = q[i];
}
/* Call the LAPACK routine */
dtrexc_( &compq, &n, t, &ldt, q, &ldq, &ifst, &ilst, work, &info );
/* Initialize input data, call the column-major middle-level
* interface to LAPACK routine and check the results */
ifst_i = ifst_save;
ilst_i = ilst_save;
for( i = 0; i < ldt*n; i++ ) {
t_i[i] = t_save[i];
}
for( i = 0; i < ldq*n; i++ ) {
q_i[i] = q_save[i];
}
for( i = 0; i < n; i++ ) {
work_i[i] = work[i];
}
info_i = LAPACKE_dtrexc_work( LAPACK_COL_MAJOR, compq_i, n_i, t_i, ldt_i,
q_i, ldq_i, &ifst_i, &ilst_i, work_i );
failed = compare_dtrexc( t, t_i, q, q_i, ifst, ifst_i, ilst, ilst_i, info,
info_i, compq, ldq, ldt, n );
if( failed == 0 ) {
printf( "PASSED: column-major middle-level interface to dtrexc\n" );
} else {
printf( "FAILED: column-major middle-level interface to dtrexc\n" );
}
/* Initialize input data, call the column-major high-level
* interface to LAPACK routine and check the results */
ifst_i = ifst_save;
ilst_i = ilst_save;
for( i = 0; i < ldt*n; i++ ) {
t_i[i] = t_save[i];
}
for( i = 0; i < ldq*n; i++ ) {
q_i[i] = q_save[i];
}
for( i = 0; i < n; i++ ) {
work_i[i] = work[i];
}
info_i = LAPACKE_dtrexc( LAPACK_COL_MAJOR, compq_i, n_i, t_i, ldt_i, q_i,
ldq_i, &ifst_i, &ilst_i );
failed = compare_dtrexc( t, t_i, q, q_i, ifst, ifst_i, ilst, ilst_i, info,
info_i, compq, ldq, ldt, n );
if( failed == 0 ) {
printf( "PASSED: column-major high-level interface to dtrexc\n" );
} else {
printf( "FAILED: column-major high-level interface to dtrexc\n" );
}
/* Initialize input data, call the row-major middle-level
* interface to LAPACK routine and check the results */
ifst_i = ifst_save;
ilst_i = ilst_save;
for( i = 0; i < ldt*n; i++ ) {
t_i[i] = t_save[i];
}
for( i = 0; i < ldq*n; i++ ) {
q_i[i] = q_save[i];
}
for( i = 0; i < n; i++ ) {
work_i[i] = work[i];
}
LAPACKE_dge_trans( LAPACK_COL_MAJOR, n, n, t_i, ldt, t_r, n+2 );
if( LAPACKE_lsame( compq, 'v' ) ) {
LAPACKE_dge_trans( LAPACK_COL_MAJOR, n, n, q_i, ldq, q_r, n+2 );
}
info_i = LAPACKE_dtrexc_work( LAPACK_ROW_MAJOR, compq_i, n_i, t_r, ldt_r,
q_r, ldq_r, &ifst_i, &ilst_i, work_i );
LAPACKE_dge_trans( LAPACK_ROW_MAJOR, n, n, t_r, n+2, t_i, ldt );
if( LAPACKE_lsame( compq, 'v' ) ) {
LAPACKE_dge_trans( LAPACK_ROW_MAJOR, n, n, q_r, n+2, q_i, ldq );
}
failed = compare_dtrexc( t, t_i, q, q_i, ifst, ifst_i, ilst, ilst_i, info,
info_i, compq, ldq, ldt, n );
if( failed == 0 ) {
printf( "PASSED: row-major middle-level interface to dtrexc\n" );
} else {
printf( "FAILED: row-major middle-level interface to dtrexc\n" );
}
/* Initialize input data, call the row-major high-level
* interface to LAPACK routine and check the results */
ifst_i = ifst_save;
ilst_i = ilst_save;
for( i = 0; i < ldt*n; i++ ) {
t_i[i] = t_save[i];
}
for( i = 0; i < ldq*n; i++ ) {
q_i[i] = q_save[i];
}
for( i = 0; i < n; i++ ) {
work_i[i] = work[i];
}
/* Init row_major arrays */
LAPACKE_dge_trans( LAPACK_COL_MAJOR, n, n, t_i, ldt, t_r, n+2 );
if( LAPACKE_lsame( compq, 'v' ) ) {
LAPACKE_dge_trans( LAPACK_COL_MAJOR, n, n, q_i, ldq, q_r, n+2 );
}
info_i = LAPACKE_dtrexc( LAPACK_ROW_MAJOR, compq_i, n_i, t_r, ldt_r, q_r,
ldq_r, &ifst_i, &ilst_i );
LAPACKE_dge_trans( LAPACK_ROW_MAJOR, n, n, t_r, n+2, t_i, ldt );
if( LAPACKE_lsame( compq, 'v' ) ) {
LAPACKE_dge_trans( LAPACK_ROW_MAJOR, n, n, q_r, n+2, q_i, ldq );
}
failed = compare_dtrexc( t, t_i, q, q_i, ifst, ifst_i, ilst, ilst_i, info,
info_i, compq, ldq, ldt, n );
if( failed == 0 ) {
printf( "PASSED: row-major high-level interface to dtrexc\n" );
} else {
printf( "FAILED: row-major high-level interface to dtrexc\n" );
}
/* Release memory */
if( t != NULL ) {
LAPACKE_free( t );
}
if( t_i != NULL ) {
LAPACKE_free( t_i );
}
if( t_r != NULL ) {
LAPACKE_free( t_r );
}
if( t_save != NULL ) {
LAPACKE_free( t_save );
}
if( q != NULL ) {
LAPACKE_free( q );
}
if( q_i != NULL ) {
LAPACKE_free( q_i );
}
if( q_r != NULL ) {
LAPACKE_free( q_r );
}
if( q_save != NULL ) {
LAPACKE_free( q_save );
}
if( work != NULL ) {
LAPACKE_free( work );
}
if( work_i != NULL ) {
LAPACKE_free( work_i );
}
return 0;
}
