lapack_logical LAPACKE_zpo_nancheck( int matrix_order, char uplo,
lapack_int n,
const lapack_complex_double *a,
lapack_int lda )
{
return LAPACKE_ztr_nancheck( matrix_order, uplo, 'n', n, a, lda );
}
lapack_int LAPACKE_ztrrfs( int matrix_layout, char uplo, char trans, char diag,
lapack_int n, lapack_int nrhs,
const lapack_complex_double* a, lapack_int lda,
const lapack_complex_double* b, lapack_int ldb,
const lapack_complex_double* x, lapack_int ldx,
double* ferr, double* berr )
{
lapack_int info = 0;
double* rwork = NULL;
lapack_complex_double* work = NULL;
if( matrix_layout != LAPACK_COL_MAJOR && matrix_layout != LAPACK_ROW_MAJOR ) {
LAPACKE_xerbla( "LAPACKE_ztrrfs", -1 );
return -1;
}
#ifndef LAPACK_DISABLE_NAN_CHECK
/* Optionally check input matrices for NaNs */
if( LAPACKE_ztr_nancheck( matrix_layout, uplo, diag, n, a, lda ) ) {
return -7;
}
if( LAPACKE_zge_nancheck( matrix_layout, n, nrhs, b, ldb ) ) {
return -9;
}
if( LAPACKE_zge_nancheck( matrix_layout, n, nrhs, x, ldx ) ) {
return -11;
}
#endif
/* Allocate memory for working array(s) */
rwork = (double*)LAPACKE_malloc( sizeof(double) * MAX(1,n) );
if( rwork == NULL ) {
info = LAPACK_WORK_MEMORY_ERROR;
goto exit_level_0;
}
work = (lapack_complex_double*)
LAPACKE_malloc( sizeof(lapack_complex_double) * MAX(1,2*n) );
if( work == NULL ) {
info = LAPACK_WORK_MEMORY_ERROR;
goto exit_level_1;
}
/* Call middle-level interface */
info = LAPACKE_ztrrfs_work( matrix_layout, uplo, trans, diag, n, nrhs, a,
lda, b, ldb, x, ldx, ferr, berr, work, rwork );
/* Release memory and exit */
LAPACKE_free( work );
exit_level_1:
LAPACKE_free( rwork );
exit_level_0:
if( info == LAPACK_WORK_MEMORY_ERROR ) {
LAPACKE_xerbla( "LAPACKE_ztrrfs", info );
}
return info;
}
lapack_logical LAPACKE_zhs_nancheck( int matrix_order, lapack_int n,
const lapack_complex_double *a,
lapack_int lda )
{
lapack_logical subdiag_nans;
if( a == NULL ) return (lapack_logical) 0;
/* Check subdiagonal first */
if( matrix_order == LAPACK_COL_MAJOR ) {
subdiag_nans = LAPACKE_z_nancheck( n-1, &a[1], lda+1 );
} else if ( matrix_order == LAPACK_ROW_MAJOR ) {
subdiag_nans = LAPACKE_z_nancheck( n-1, &a[lda], lda+1 );
} else {
return (lapack_logical) 0;
}
/* Check upper triangular if subdiagonal has no NaNs. */
return subdiag_nans || LAPACKE_ztr_nancheck( matrix_order, 'u', 'n',
n, a, lda);
}
lapack_int LAPACKE_ztrtrs( int matrix_layout, char uplo, char trans, char diag,
lapack_int n, lapack_int nrhs,
const lapack_complex_double* a, lapack_int lda,
lapack_complex_double* b, lapack_int ldb )
{
if( matrix_layout != LAPACK_COL_MAJOR && matrix_layout != LAPACK_ROW_MAJOR ) {
LAPACKE_xerbla( "LAPACKE_ztrtrs", -1 );
return -1;
}
#ifndef LAPACK_DISABLE_NAN_CHECK
/* Optionally check input matrices for NaNs */
if( LAPACKE_ztr_nancheck( matrix_layout, uplo, diag, n, a, lda ) ) {
return -7;
}
if( LAPACKE_zge_nancheck( matrix_layout, n, nrhs, b, ldb ) ) {
return -9;
}
#endif
return LAPACKE_ztrtrs_work( matrix_layout, uplo, trans, diag, n, nrhs, a,
lda, b, ldb );
}
double LAPACKE_zlantr( int matrix_layout, char norm, char uplo, char diag,
lapack_int m, lapack_int n,
const lapack_complex_double* a, lapack_int lda )
{
lapack_int info = 0;
double res = 0.;
double* work = NULL;
if( matrix_layout != LAPACK_COL_MAJOR && matrix_layout != LAPACK_ROW_MAJOR ) {
LAPACKE_xerbla( "LAPACKE_zlantr", -1 );
return -1;
}
#ifndef LAPACK_DISABLE_NAN_CHECK
if( LAPACKE_get_nancheck() ) {
/* Optionally check input matrices for NaNs */
if( LAPACKE_ztr_nancheck( matrix_layout, uplo, diag, MIN(m,n), a, lda ) ) {
return -7;
}
}
#endif
/* Allocate memory for working array(s) */
if( LAPACKE_lsame( norm, 'i' ) ) {
work = (double*)LAPACKE_malloc( sizeof(double) * MAX(1,MAX(m,n)) );
if( work == NULL ) {
info = LAPACK_WORK_MEMORY_ERROR;
goto exit_level_0;
}
}
/* Call middle-level interface */
res = LAPACKE_zlantr_work( matrix_layout, norm, uplo, diag, m, n, a, lda,
work );
/* Release memory and exit */
if( LAPACKE_lsame( norm, 'i' ) ) {
LAPACKE_free( work );
}
exit_level_0:
if( info == LAPACK_WORK_MEMORY_ERROR ) {
LAPACKE_xerbla( "LAPACKE_zlantr", info );
}
return res;
}
lapack_int LAPACKE_zlascl( int matrix_layout, char type, lapack_int kl,
lapack_int ku, double cfrom, double cto,
lapack_int m, lapack_int n, lapack_complex_double* a,
lapack_int lda )
{
if( matrix_layout != LAPACK_COL_MAJOR && matrix_layout != LAPACK_ROW_MAJOR ) {
LAPACKE_xerbla( "LAPACKE_zlascl", -1 );
return -1;
}
#ifndef LAPACK_zISABLE_NAN_CHECK
/* Optionally check input matrices for NaNs */
switch (type) {
case 'G':
if( LAPACKE_zge_nancheck( matrix_layout, lda, n, a, lda ) ) {
return -9;
}
break;
case 'L':
// TYPE = 'L' - lower triangular matrix.
if( LAPACKE_ztr_nancheck( matrix_layout, 'L', 'N', n, a, lda ) ) {
return -9;
}
break;
case 'U':
// TYPE = 'U' - upper triangular matrix
if( LAPACKE_ztr_nancheck( matrix_layout, 'U', 'N', n, a, lda ) ) {
return -9;
}
break;
case 'H':
// TYPE = 'H' - upper Hessenberg matrix
if( LAPACKE_zhs_nancheck( matrix_layout, n, a, lda ) ) {
return -9;
}
break;
case 'B':
// TYPE = 'B' - A is a symmetric band matrix with lower bandwidth KL
// and upper bandwidth KU and with the only the lower
// half stored.
if( LAPACKE_zhb_nancheck( matrix_layout, 'L', n, kl, a, lda ) ) {
return -9;
}
break;
case 'Q':
// TYPE = 'Q' - A is a symmetric band matrix with lower bandwidth KL
// and upper bandwidth KU and with the only the upper
// half stored.
if( LAPACKE_zhb_nancheck( matrix_layout, 'U', n, ku, a, lda ) ) {
return -9;
}
break;
case 'Z':
// TYPE = 'Z' - A is a band matrix with lower bandwidth KL and upper
// bandwidth KU. See DGBTRF for storage details.
if( LAPACKE_zgb_nancheck( matrix_layout, n, n, kl, kl+ku, a, lda ) ) {
return -6;
}
break;
}
#endif
return LAPACKE_zlascl_work( matrix_layout, type, kl, ku, cfrom, cto, m, n, a, lda );
}
lapack_int LAPACKE_zlarfb( int matrix_order, char side, char trans, char direct,
char storev, lapack_int m, lapack_int n,
lapack_int k, const lapack_complex_double* v,
lapack_int ldv, const lapack_complex_double* t,
lapack_int ldt, lapack_complex_double* c,
lapack_int ldc )
{
lapack_int info = 0;
lapack_int ldwork = ( side=='l')?n:(( side=='r')?m:1);
lapack_complex_double* work = NULL;
lapack_int ncols_v, nrows_v;
if( matrix_order != LAPACK_COL_MAJOR && matrix_order != LAPACK_ROW_MAJOR ) {
LAPACKE_xerbla( "LAPACKE_zlarfb", -1 );
return -1;
}
#ifndef LAPACK_DISABLE_NAN_CHECK
/* Optionally check input matrices for NaNs */
ncols_v = LAPACKE_lsame( storev, 'c' ) ? k :
( ( LAPACKE_lsame( storev, 'r' ) &&
LAPACKE_lsame( side, 'l' ) ) ? m :
( ( LAPACKE_lsame( storev, 'r' ) &&
LAPACKE_lsame( side, 'r' ) ) ? n : 1) );
nrows_v = ( LAPACKE_lsame( storev, 'c' ) &&
LAPACKE_lsame( side, 'l' ) ) ? m :
( ( LAPACKE_lsame( storev, 'c' ) &&
LAPACKE_lsame( side, 'r' ) ) ? n :
( LAPACKE_lsame( storev, 'r' ) ? k : 1) );
if( LAPACKE_zge_nancheck( matrix_order, m, n, c, ldc ) ) {
return -13;
}
if( LAPACKE_zge_nancheck( matrix_order, k, k, t, ldt ) ) {
return -11;
}
if( LAPACKE_lsame( storev, 'c' ) && LAPACKE_lsame( direct, 'f' ) ) {
if( LAPACKE_ztr_nancheck( matrix_order, 'l', 'u', k, v, ldv ) )
return -9;
if( LAPACKE_zge_nancheck( matrix_order, nrows_v-k, ncols_v, &v[k*ldv],
ldv ) )
return -9;
} else if( LAPACKE_lsame( storev, 'c' ) && LAPACKE_lsame( direct, 'b' ) ) {
if( k > nrows_v ) {
LAPACKE_xerbla( "LAPACKE_zlarfb", -8 );
return -8;
}
if( LAPACKE_ztr_nancheck( matrix_order, 'u', 'u', k,
&v[(nrows_v-k)*ldv], ldv ) )
return -9;
if( LAPACKE_zge_nancheck( matrix_order, nrows_v-k, ncols_v, v, ldv ) )
return -9;
} else if( LAPACKE_lsame( storev, 'r' ) && LAPACKE_lsame( direct, 'f' ) ) {
if( LAPACKE_ztr_nancheck( matrix_order, 'u', 'u', k, v, ldv ) )
return -9;
if( LAPACKE_zge_nancheck( matrix_order, nrows_v, ncols_v-k, &v[k],
ldv ) )
return -9;
} else if( LAPACKE_lsame( storev, 'r' ) && LAPACKE_lsame( direct, 'f' ) ) {
if( k > ncols_v ) {
LAPACKE_xerbla( "LAPACKE_zlarfb", -8 );
return -8;
}
if( LAPACKE_ztr_nancheck( matrix_order, 'l', 'u', k, &v[ncols_v-k],
ldv ) )
return -9;
if( LAPACKE_zge_nancheck( matrix_order, nrows_v, ncols_v-k, v, ldv ) )
return -9;
}
#endif
/* Allocate memory for working array(s) */
work = (lapack_complex_double*)
LAPACKE_malloc( sizeof(lapack_complex_double) * ldwork * MAX(1,k) );
if( work == NULL ) {
info = LAPACK_WORK_MEMORY_ERROR;
goto exit_level_0;
}
/* Call middle-level interface */
info = LAPACKE_zlarfb_work( matrix_order, side, trans, direct, storev, m, n,
k, v, ldv, t, ldt, c, ldc, work, ldwork );
/* Release memory and exit */
LAPACKE_free( work );
exit_level_0:
if( info == LAPACK_WORK_MEMORY_ERROR ) {
LAPACKE_xerbla( "LAPACKE_zlarfb", info );
}
return info;
}
