lapack_int LAPACKE_ssbgvd_work( int matrix_layout, char jobz, char uplo, lapack_int n, lapack_int ka, lapack_int kb, float* ab, lapack_int ldab, float* bb, lapack_int ldbb, float* w, float* z, lapack_int ldz, float* work, lapack_int lwork, lapack_int* iwork, lapack_int liwork ) { lapack_int info = 0; if( matrix_layout == LAPACK_COL_MAJOR ) { /* Call LAPACK function and adjust info */ LAPACK_ssbgvd( &jobz, &uplo, &n, &ka, &kb, ab, &ldab, bb, &ldbb, w, z, &ldz, work, &lwork, iwork, &liwork, &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 ldz_t = MAX(1,n); float* ab_t = NULL; float* bb_t = NULL; float* z_t = NULL; /* Check leading dimension(s) */ if( ldab < n ) { info = -8; LAPACKE_xerbla( "LAPACKE_ssbgvd_work", info ); return info; } if( ldbb < n ) { info = -10; LAPACKE_xerbla( "LAPACKE_ssbgvd_work", info ); return info; } if( ldz < n ) { info = -13; LAPACKE_xerbla( "LAPACKE_ssbgvd_work", info ); return info; } /* Query optimal working array(s) size if requested */ if( liwork == -1 || lwork == -1 ) { LAPACK_ssbgvd( &jobz, &uplo, &n, &ka, &kb, ab, &ldab_t, bb, &ldbb_t, w, z, &ldz_t, work, &lwork, iwork, &liwork, &info ); return (info < 0) ? (info - 1) : info; } /* Allocate memory for temporary array(s) */ ab_t = (float*)LAPACKE_malloc( sizeof(float) * ldab_t * MAX(1,n) ); if( ab_t == NULL ) { info = LAPACK_TRANSPOSE_MEMORY_ERROR; goto exit_level_0; } bb_t = (float*)LAPACKE_malloc( sizeof(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 = (float*)LAPACKE_malloc( sizeof(float) * ldz_t * MAX(1,n) ); if( z_t == NULL ) { info = LAPACK_TRANSPOSE_MEMORY_ERROR; goto exit_level_2; } } /* Transpose input matrices */ LAPACKE_ssb_trans( matrix_layout, uplo, n, ka, ab, ldab, ab_t, ldab_t ); LAPACKE_ssb_trans( matrix_layout, uplo, n, kb, bb, ldbb, bb_t, ldbb_t ); /* Call LAPACK function and adjust info */ LAPACK_ssbgvd( &jobz, &uplo, &n, &ka, &kb, ab_t, &ldab_t, bb_t, &ldbb_t, w, z_t, &ldz_t, work, &lwork, iwork, &liwork, &info ); if( info < 0 ) { info = info - 1; } /* Transpose output matrices */ LAPACKE_ssb_trans( LAPACK_COL_MAJOR, uplo, n, ka, ab_t, ldab_t, ab, ldab ); LAPACKE_ssb_trans( LAPACK_COL_MAJOR, uplo, n, kb, bb_t, ldbb_t, bb, ldbb ); if( LAPACKE_lsame( jobz, 'v' ) ) { LAPACKE_sge_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_ssbgvd_work", info ); } } else { info = -1; LAPACKE_xerbla( "LAPACKE_ssbgvd_work", info ); } return info; }
lapack_int LAPACKE_ssbevx_work( int matrix_layout, char jobz, char range, char uplo, lapack_int n, lapack_int kd, float* ab, lapack_int ldab, float* q, lapack_int ldq, float vl, float vu, lapack_int il, lapack_int iu, float abstol, lapack_int* m, float* w, float* z, lapack_int ldz, float* work, lapack_int* iwork, lapack_int* ifail ) { lapack_int info = 0; if( matrix_layout == LAPACK_COL_MAJOR ) { /* Call LAPACK function and adjust info */ LAPACK_ssbevx( &jobz, &range, &uplo, &n, &kd, ab, &ldab, q, &ldq, &vl, &vu, &il, &iu, &abstol, m, w, z, &ldz, work, iwork, ifail, &info ); if( info < 0 ) { info = info - 1; } } else if( matrix_layout == 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); float* ab_t = NULL; float* q_t = NULL; float* z_t = NULL; /* Check leading dimension(s) */ if( ldab < n ) { info = -8; LAPACKE_xerbla( "LAPACKE_ssbevx_work", info ); return info; } if( ldq < n ) { info = -10; LAPACKE_xerbla( "LAPACKE_ssbevx_work", info ); return info; } if( ldz < ncols_z ) { info = -19; LAPACKE_xerbla( "LAPACKE_ssbevx_work", info ); return info; } /* Allocate memory for temporary array(s) */ ab_t = (float*)LAPACKE_malloc( sizeof(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 = (float*)LAPACKE_malloc( sizeof(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 = (float*) LAPACKE_malloc( sizeof(float) * ldz_t * MAX(1,ncols_z) ); if( z_t == NULL ) { info = LAPACK_TRANSPOSE_MEMORY_ERROR; goto exit_level_2; } } /* Transpose input matrices */ LAPACKE_ssb_trans( matrix_layout, uplo, n, kd, ab, ldab, ab_t, ldab_t ); /* Call LAPACK function and adjust info */ LAPACK_ssbevx( &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, iwork, ifail, &info ); if( info < 0 ) { info = info - 1; } /* Transpose output matrices */ LAPACKE_ssb_trans( LAPACK_COL_MAJOR, uplo, n, kd, ab_t, ldab_t, ab, ldab ); if( LAPACKE_lsame( jobz, 'v' ) ) { LAPACKE_sge_trans( LAPACK_COL_MAJOR, n, n, q_t, ldq_t, q, ldq ); } if( LAPACKE_lsame( jobz, 'v' ) ) { LAPACKE_sge_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_ssbevx_work", info ); } } else { info = -1; LAPACKE_xerbla( "LAPACKE_ssbevx_work", info ); } return info; }
lapack_int LAPACKE_ssbgst_work( int matrix_order, char vect, char uplo, lapack_int n, lapack_int ka, lapack_int kb, float* ab, lapack_int ldab, const float* bb, lapack_int ldbb, float* x, lapack_int ldx, float* work ) { lapack_int info = 0; if( matrix_order == LAPACK_COL_MAJOR ) { /* Call LAPACK function and adjust info */ LAPACK_ssbgst( &vect, &uplo, &n, &ka, &kb, ab, &ldab, bb, &ldbb, x, &ldx, work, &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 ldx_t = MAX(1,n); float* ab_t = NULL; float* bb_t = NULL; float* x_t = NULL; /* Check leading dimension(s) */ if( ldab < n ) { info = -8; LAPACKE_xerbla( "LAPACKE_ssbgst_work", info ); return info; } if( ldbb < n ) { info = -10; LAPACKE_xerbla( "LAPACKE_ssbgst_work", info ); return info; } if( ldx < n ) { info = -12; LAPACKE_xerbla( "LAPACKE_ssbgst_work", info ); return info; } /* Allocate memory for temporary array(s) */ ab_t = (float*)LAPACKE_malloc( sizeof(float) * ldab_t * MAX(1,n) ); if( ab_t == NULL ) { info = LAPACK_TRANSPOSE_MEMORY_ERROR; goto exit_level_0; } bb_t = (float*)LAPACKE_malloc( sizeof(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 = (float*)LAPACKE_malloc( sizeof(float) * ldx_t * MAX(1,n) ); if( x_t == NULL ) { info = LAPACK_TRANSPOSE_MEMORY_ERROR; goto exit_level_2; } } /* Transpose input matrices */ LAPACKE_ssb_trans( matrix_order, uplo, n, ka, ab, ldab, ab_t, ldab_t ); LAPACKE_ssb_trans( matrix_order, uplo, n, kb, bb, ldbb, bb_t, ldbb_t ); /* Call LAPACK function and adjust info */ LAPACK_ssbgst( &vect, &uplo, &n, &ka, &kb, ab_t, &ldab_t, bb_t, &ldbb_t, x_t, &ldx_t, work, &info ); if( info < 0 ) { info = info - 1; } /* Transpose output matrices */ LAPACKE_ssb_trans( LAPACK_COL_MAJOR, uplo, n, ka, ab_t, ldab_t, ab, ldab ); if( LAPACKE_lsame( vect, 'v' ) ) { LAPACKE_sge_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_ssbgst_work", info ); } } else { info = -1; LAPACKE_xerbla( "LAPACKE_ssbgst_work", info ); } return info; }