lapack_int LAPACKE_chbtrd_work( int matrix_order, char vect, char uplo,
lapack_int n, lapack_int kd,
lapack_complex_float* ab, lapack_int ldab,
float* d, float* e, lapack_complex_float* q,
lapack_int ldq, lapack_complex_float* work )
{
lapack_int info = 0;
if( matrix_order == LAPACK_COL_MAJOR ) {
/* Call LAPACK function and adjust info */
LAPACK_chbtrd( &vect, &uplo, &n, &kd, ab, &ldab, d, e, q, &ldq, work,
&info );
if( info < 0 ) {
info = info - 1;
}
} else if( matrix_order == LAPACK_ROW_MAJOR ) {
lapack_int ldab_t = MAX(1,kd+1);
lapack_int ldq_t = MAX(1,n);
lapack_complex_float* ab_t = NULL;
lapack_complex_float* q_t = NULL;
/* Check leading dimension(s) */
if( ldab < n ) {
info = -7;
LAPACKE_xerbla( "LAPACKE_chbtrd_work", info );
return info;
}
if( ldq < n ) {
info = -11;
LAPACKE_xerbla( "LAPACKE_chbtrd_work", info );
return info;
}
/* Allocate memory for temporary array(s) */
ab_t = (lapack_complex_float*)
LAPACKE_malloc( sizeof(lapack_complex_float) * ldab_t * MAX(1,n) );
if( ab_t == NULL ) {
info = LAPACK_TRANSPOSE_MEMORY_ERROR;
goto exit_level_0;
}
if( LAPACKE_lsame( vect, 'u' ) || LAPACKE_lsame( vect, 'v' ) ) {
q_t = (lapack_complex_float*)
LAPACKE_malloc( sizeof(lapack_complex_float) *
ldq_t * MAX(1,n) );
if( q_t == NULL ) {
info = LAPACK_TRANSPOSE_MEMORY_ERROR;
goto exit_level_1;
}
}
/* Transpose input matrices */
LAPACKE_chb_trans( matrix_order, uplo, n, kd, ab, ldab, ab_t, ldab_t );
if( LAPACKE_lsame( vect, 'u' ) || LAPACKE_lsame( vect, 'v' ) ) {
LAPACKE_cge_trans( matrix_order, n, n, q, ldq, q_t, ldq_t );
}
/* Call LAPACK function and adjust info */
LAPACK_chbtrd( &vect, &uplo, &n, &kd, ab_t, &ldab_t, d, e, q_t, &ldq_t,
work, &info );
if( info < 0 ) {
info = info - 1;
}
/* Transpose output matrices */
LAPACKE_chb_trans( LAPACK_COL_MAJOR, uplo, n, kd, ab_t, ldab_t, ab,
ldab );
if( LAPACKE_lsame( vect, 'u' ) || LAPACKE_lsame( vect, 'v' ) ) {
LAPACKE_cge_trans( LAPACK_COL_MAJOR, n, n, q_t, ldq_t, q, ldq );
}
/* Release memory and exit */
if( LAPACKE_lsame( vect, 'u' ) || LAPACKE_lsame( vect, 'v' ) ) {
LAPACKE_free( q_t );
}
exit_level_1:
LAPACKE_free( ab_t );
exit_level_0:
if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
LAPACKE_xerbla( "LAPACKE_chbtrd_work", info );
}
} else {
info = -1;
LAPACKE_xerbla( "LAPACKE_chbtrd_work", info );
}
return info;
}
lapack_int LAPACKE_chbgst_work( int matrix_layout, char vect, char uplo,
lapack_int n, lapack_int ka, lapack_int kb,
lapack_complex_float* ab, lapack_int ldab,
const lapack_complex_float* bb, lapack_int ldbb,
lapack_complex_float* x, lapack_int ldx,
lapack_complex_float* work, float* rwork )
{
lapack_int info = 0;
if( matrix_layout == LAPACK_COL_MAJOR ) {
/* Call LAPACK function and adjust info */
LAPACK_chbgst( &vect, &uplo, &n, &ka, &kb, ab, &ldab, bb, &ldbb, x,
&ldx, work, rwork, &info );
if( info < 0 ) {
info = info - 1;
}
} else if( matrix_layout == LAPACK_ROW_MAJOR ) {
lapack_int ldab_t = MAX(1,ka+1);
lapack_int ldbb_t = MAX(1,kb+1);
lapack_int ldx_t = MAX(1,n);
lapack_complex_float* ab_t = NULL;
lapack_complex_float* bb_t = NULL;
lapack_complex_float* x_t = NULL;
/* Check leading dimension(s) */
if( ldab < n ) {
info = -8;
LAPACKE_xerbla( "LAPACKE_chbgst_work", info );
return info;
}
if( ldbb < n ) {
info = -10;
LAPACKE_xerbla( "LAPACKE_chbgst_work", info );
return info;
}
if( ldx < n ) {
info = -12;
LAPACKE_xerbla( "LAPACKE_chbgst_work", info );
return info;
}
/* Allocate memory for temporary array(s) */
ab_t = (lapack_complex_float*)
LAPACKE_malloc( sizeof(lapack_complex_float) * ldab_t * MAX(1,n) );
if( ab_t == NULL ) {
info = LAPACK_TRANSPOSE_MEMORY_ERROR;
goto exit_level_0;
}
bb_t = (lapack_complex_float*)
LAPACKE_malloc( sizeof(lapack_complex_float) * ldbb_t * MAX(1,n) );
if( bb_t == NULL ) {
info = LAPACK_TRANSPOSE_MEMORY_ERROR;
goto exit_level_1;
}
if( LAPACKE_lsame( vect, 'v' ) ) {
x_t = (lapack_complex_float*)
LAPACKE_malloc( sizeof(lapack_complex_float) *
ldx_t * MAX(1,n) );
if( x_t == NULL ) {
info = LAPACK_TRANSPOSE_MEMORY_ERROR;
goto exit_level_2;
}
}
/* Transpose input matrices */
LAPACKE_chb_trans( matrix_layout, uplo, n, ka, ab, ldab, ab_t, ldab_t );
LAPACKE_chb_trans( matrix_layout, uplo, n, kb, bb, ldbb, bb_t, ldbb_t );
/* Call LAPACK function and adjust info */
LAPACK_chbgst( &vect, &uplo, &n, &ka, &kb, ab_t, &ldab_t, bb_t, &ldbb_t,
x_t, &ldx_t, work, rwork, &info );
if( info < 0 ) {
info = info - 1;
}
/* Transpose output matrices */
LAPACKE_chb_trans( LAPACK_COL_MAJOR, uplo, n, ka, ab_t, ldab_t, ab,
ldab );
if( LAPACKE_lsame( vect, 'v' ) ) {
LAPACKE_cge_trans( LAPACK_COL_MAJOR, n, n, x_t, ldx_t, x, ldx );
}
/* Release memory and exit */
if( LAPACKE_lsame( vect, 'v' ) ) {
LAPACKE_free( x_t );
}
exit_level_2:
LAPACKE_free( bb_t );
exit_level_1:
LAPACKE_free( ab_t );
exit_level_0:
if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
LAPACKE_xerbla( "LAPACKE_chbgst_work", info );
}
} else {
info = -1;
LAPACKE_xerbla( "LAPACKE_chbgst_work", info );
}
return info;
}
lapack_int LAPACKE_chbgvd_work( int matrix_order, char jobz, char uplo,
lapack_int n, lapack_int ka, lapack_int kb,
lapack_complex_float* ab, lapack_int ldab,
lapack_complex_float* bb, lapack_int ldbb,
float* w, lapack_complex_float* z,
lapack_int ldz, lapack_complex_float* work,
lapack_int lwork, float* rwork,
lapack_int lrwork, lapack_int* iwork,
lapack_int liwork )
{
lapack_int info = 0;
if( matrix_order == LAPACK_COL_MAJOR ) {
/* Call LAPACK function and adjust info */
LAPACK_chbgvd( &jobz, &uplo, &n, &ka, &kb, ab, &ldab, bb, &ldbb, w, z,
&ldz, work, &lwork, rwork, &lrwork, iwork, &liwork,
&info );
if( info < 0 ) {
info = info - 1;
}
} else if( matrix_order == LAPACK_ROW_MAJOR ) {
lapack_int ldab_t = MAX(1,ka+1);
lapack_int ldbb_t = MAX(1,kb+1);
lapack_int ldz_t = MAX(1,n);
lapack_complex_float* ab_t = NULL;
lapack_complex_float* bb_t = NULL;
lapack_complex_float* z_t = NULL;
/* Check leading dimension(s) */
if( ldab < n ) {
info = -8;
LAPACKE_xerbla( "LAPACKE_chbgvd_work", info );
return info;
}
if( ldbb < n ) {
info = -10;
LAPACKE_xerbla( "LAPACKE_chbgvd_work", info );
return info;
}
if( ldz < n ) {
info = -13;
LAPACKE_xerbla( "LAPACKE_chbgvd_work", info );
return info;
}
/* Query optimal working array(s) size if requested */
if( liwork == -1 || lrwork == -1 || lwork == -1 ) {
LAPACK_chbgvd( &jobz, &uplo, &n, &ka, &kb, ab, &ldab_t, bb, &ldbb_t,
w, z, &ldz_t, work, &lwork, rwork, &lrwork, iwork,
&liwork, &info );
return (info < 0) ? (info - 1) : info;
}
/* Allocate memory for temporary array(s) */
ab_t = (lapack_complex_float*)
LAPACKE_malloc( sizeof(lapack_complex_float) * ldab_t * MAX(1,n) );
if( ab_t == NULL ) {
info = LAPACK_TRANSPOSE_MEMORY_ERROR;
goto exit_level_0;
}
bb_t = (lapack_complex_float*)
LAPACKE_malloc( sizeof(lapack_complex_float) * ldbb_t * MAX(1,n) );
if( bb_t == NULL ) {
info = LAPACK_TRANSPOSE_MEMORY_ERROR;
goto exit_level_1;
}
if( LAPACKE_lsame( jobz, 'v' ) ) {
z_t = (lapack_complex_float*)
LAPACKE_malloc( sizeof(lapack_complex_float) *
ldz_t * MAX(1,n) );
if( z_t == NULL ) {
info = LAPACK_TRANSPOSE_MEMORY_ERROR;
goto exit_level_2;
}
}
/* Transpose input matrices */
LAPACKE_chb_trans( matrix_order, uplo, n, ka, ab, ldab, ab_t, ldab_t );
LAPACKE_chb_trans( matrix_order, uplo, n, kb, bb, ldbb, bb_t, ldbb_t );
/* Call LAPACK function and adjust info */
LAPACK_chbgvd( &jobz, &uplo, &n, &ka, &kb, ab_t, &ldab_t, bb_t, &ldbb_t,
w, z_t, &ldz_t, work, &lwork, rwork, &lrwork, iwork,
&liwork, &info );
if( info < 0 ) {
info = info - 1;
}
/* Transpose output matrices */
LAPACKE_chb_trans( LAPACK_COL_MAJOR, uplo, n, ka, ab_t, ldab_t, ab,
ldab );
LAPACKE_chb_trans( LAPACK_COL_MAJOR, uplo, n, kb, bb_t, ldbb_t, bb,
ldbb );
if( LAPACKE_lsame( jobz, 'v' ) ) {
LAPACKE_cge_trans( LAPACK_COL_MAJOR, n, n, z_t, ldz_t, z, ldz );
}
/* Release memory and exit */
if( LAPACKE_lsame( jobz, 'v' ) ) {
LAPACKE_free( z_t );
}
exit_level_2:
LAPACKE_free( bb_t );
exit_level_1:
LAPACKE_free( ab_t );
exit_level_0:
if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
LAPACKE_xerbla( "LAPACKE_chbgvd_work", info );
}
} else {
info = -1;
LAPACKE_xerbla( "LAPACKE_chbgvd_work", info );
}
return info;
}
lapack_int LAPACKE_chbevx_work( int matrix_order, char jobz, char range,
char uplo, lapack_int n, lapack_int kd,
lapack_complex_float* ab, lapack_int ldab,
lapack_complex_float* q, lapack_int ldq,
float vl, float vu, lapack_int il,
lapack_int iu, float abstol, lapack_int* m,
float* w, lapack_complex_float* z,
lapack_int ldz, lapack_complex_float* work,
float* rwork, lapack_int* iwork,
lapack_int* ifail )
{
lapack_int info = 0;
if( matrix_order == LAPACK_COL_MAJOR ) {
/* Call LAPACK function and adjust info */
LAPACK_chbevx( &jobz, &range, &uplo, &n, &kd, ab, &ldab, q, &ldq, &vl,
&vu, &il, &iu, &abstol, m, w, z, &ldz, work, rwork,
iwork, ifail, &info );
if( info < 0 ) {
info = info - 1;
}
} else if( matrix_order == LAPACK_ROW_MAJOR ) {
lapack_int ncols_z = ( LAPACKE_lsame( range, 'a' ) ||
LAPACKE_lsame( range, 'v' ) ) ? n :
( LAPACKE_lsame( range, 'i' ) ? (iu-il+1) : 1);
lapack_int ldab_t = MAX(1,kd+1);
lapack_int ldq_t = MAX(1,n);
lapack_int ldz_t = MAX(1,n);
lapack_complex_float* ab_t = NULL;
lapack_complex_float* q_t = NULL;
lapack_complex_float* z_t = NULL;
/* Check leading dimension(s) */
if( ldab < n ) {
info = -8;
LAPACKE_xerbla( "LAPACKE_chbevx_work", info );
return info;
}
if( ldq < n ) {
info = -10;
LAPACKE_xerbla( "LAPACKE_chbevx_work", info );
return info;
}
if( ldz < ncols_z ) {
info = -19;
LAPACKE_xerbla( "LAPACKE_chbevx_work", info );
return info;
}
/* Allocate memory for temporary array(s) */
ab_t = (lapack_complex_float*)
LAPACKE_malloc( sizeof(lapack_complex_float) * ldab_t * MAX(1,n) );
if( ab_t == NULL ) {
info = LAPACK_TRANSPOSE_MEMORY_ERROR;
goto exit_level_0;
}
if( LAPACKE_lsame( jobz, 'v' ) ) {
q_t = (lapack_complex_float*)
LAPACKE_malloc( sizeof(lapack_complex_float) *
ldq_t * MAX(1,n) );
if( q_t == NULL ) {
info = LAPACK_TRANSPOSE_MEMORY_ERROR;
goto exit_level_1;
}
}
if( LAPACKE_lsame( jobz, 'v' ) ) {
z_t = (lapack_complex_float*)
LAPACKE_malloc( sizeof(lapack_complex_float) *
ldz_t * MAX(1,ncols_z) );
if( z_t == NULL ) {
info = LAPACK_TRANSPOSE_MEMORY_ERROR;
goto exit_level_2;
}
}
/* Transpose input matrices */
LAPACKE_chb_trans( matrix_order, uplo, n, kd, ab, ldab, ab_t, ldab_t );
/* Call LAPACK function and adjust info */
LAPACK_chbevx( &jobz, &range, &uplo, &n, &kd, ab_t, &ldab_t, q_t,
&ldq_t, &vl, &vu, &il, &iu, &abstol, m, w, z_t, &ldz_t,
work, rwork, iwork, ifail, &info );
if( info < 0 ) {
info = info - 1;
}
/* Transpose output matrices */
LAPACKE_chb_trans( LAPACK_COL_MAJOR, uplo, n, kd, ab_t, ldab_t, ab,
ldab );
if( LAPACKE_lsame( jobz, 'v' ) ) {
LAPACKE_cge_trans( LAPACK_COL_MAJOR, n, n, q_t, ldq_t, q, ldq );
}
if( LAPACKE_lsame( jobz, 'v' ) ) {
LAPACKE_cge_trans( LAPACK_COL_MAJOR, n, ncols_z, z_t, ldz_t, z,
ldz );
}
/* Release memory and exit */
if( LAPACKE_lsame( jobz, 'v' ) ) {
LAPACKE_free( z_t );
}
exit_level_2:
if( LAPACKE_lsame( jobz, 'v' ) ) {
LAPACKE_free( q_t );
}
exit_level_1:
LAPACKE_free( ab_t );
exit_level_0:
if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
LAPACKE_xerbla( "LAPACKE_chbevx_work", info );
}
} else {
info = -1;
LAPACKE_xerbla( "LAPACKE_chbevx_work", info );
}
return info;
}
lapack_int LAPACKE_chbev_work( int matrix_layout, char jobz, char uplo,
lapack_int n, lapack_int kd,
lapack_complex_float* ab, lapack_int ldab,
float* w, lapack_complex_float* z,
lapack_int ldz, lapack_complex_float* work,
float* rwork )
{
lapack_int info = 0;
if( matrix_layout == LAPACK_COL_MAJOR ) {
/* Call LAPACK function and adjust info */
LAPACK_chbev( &jobz, &uplo, &n, &kd, ab, &ldab, w, z, &ldz, work, rwork,
&info );
if( info < 0 ) {
info = info - 1;
}
} else if( matrix_layout == LAPACK_ROW_MAJOR ) {
lapack_int ldab_t = MAX(1,kd+1);
lapack_int ldz_t = MAX(1,n);
lapack_complex_float* ab_t = NULL;
lapack_complex_float* z_t = NULL;
/* Check leading dimension(s) */
if( ldab < n ) {
info = -7;
LAPACKE_xerbla( "LAPACKE_chbev_work", info );
return info;
}
if( ldz < n ) {
info = -10;
LAPACKE_xerbla( "LAPACKE_chbev_work", info );
return info;
}
/* Allocate memory for temporary array(s) */
ab_t = (lapack_complex_float*)
LAPACKE_malloc( sizeof(lapack_complex_float) * ldab_t * MAX(1,n) );
if( ab_t == NULL ) {
info = LAPACK_TRANSPOSE_MEMORY_ERROR;
goto exit_level_0;
}
if( LAPACKE_lsame( jobz, 'v' ) ) {
z_t = (lapack_complex_float*)
LAPACKE_malloc( sizeof(lapack_complex_float) *
ldz_t * MAX(1,n) );
if( z_t == NULL ) {
info = LAPACK_TRANSPOSE_MEMORY_ERROR;
goto exit_level_1;
}
}
/* Transpose input matrices */
LAPACKE_chb_trans( matrix_layout, uplo, n, kd, ab, ldab, ab_t, ldab_t );
/* Call LAPACK function and adjust info */
LAPACK_chbev( &jobz, &uplo, &n, &kd, ab_t, &ldab_t, w, z_t, &ldz_t,
work, rwork, &info );
if( info < 0 ) {
info = info - 1;
}
/* Transpose output matrices */
LAPACKE_chb_trans( LAPACK_COL_MAJOR, uplo, n, kd, ab_t, ldab_t, ab,
ldab );
if( LAPACKE_lsame( jobz, 'v' ) ) {
LAPACKE_cge_trans( LAPACK_COL_MAJOR, n, n, z_t, ldz_t, z, ldz );
}
/* Release memory and exit */
if( LAPACKE_lsame( jobz, 'v' ) ) {
LAPACKE_free( z_t );
}
exit_level_1:
LAPACKE_free( ab_t );
exit_level_0:
if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
LAPACKE_xerbla( "LAPACKE_chbev_work", info );
}
} else {
info = -1;
LAPACKE_xerbla( "LAPACKE_chbev_work", info );
}
return info;
}
