arma_inline
eT
cblas_cx_dot(const int N, const eT* X, const eT* Y)
{
arma_type_check((is_supported_blas_type<eT>::value == false));
if(is_supported_complex_float<eT>::value)
{
typedef typename std::complex<float> T;
T out;
arma_wrapper(cblas_cdotu_sub)(N, (const T*)X, 1, (const T*)Y, 1, &out);
return eT(out);
}
else
if(is_supported_complex_double<eT>::value)
{
typedef typename std::complex<double> T;
T out;
arma_wrapper(cblas_zdotu_sub)(N, (const T*)X, 1, (const T*)Y, 1, &out);
return eT(out);
}
return eT(0);
}
inline
void
cblas_herk
(
const enum CBLAS_ORDER Order, const enum CBLAS_UPLO Uplo, const enum CBLAS_TRANSPOSE Trans,
const int N, const int K, const T alpha,
const std::complex<T>* A, const int lda, const T beta, std::complex<T>* C, const int ldc
)
{
arma_type_check((is_supported_blas_type<T>::value == false));
if(is_float<T>::value)
{
typedef float TT;
typedef std::complex<float> cx_TT;
arma_wrapper(cblas_cherk)(Order, Uplo, Trans, N, K, (const TT)alpha, (const cx_TT*)A, lda, (const TT)beta, (cx_TT*)C, ldc);
}
else
if(is_double<T>::value)
{
typedef double TT;
typedef std::complex<double> cx_TT;
arma_wrapper(cblas_zherk)(Order, Uplo, Trans, N, K, (const TT)alpha, (const cx_TT*)A, lda, (const TT)beta, (cx_TT*)C, ldc);
}
}
inline
void
gssv(superlu_options_t* options, SuperMatrix* A, int* perm_c, int* perm_r, SuperMatrix* L, SuperMatrix* U, SuperMatrix* B, SuperLUStat_t* stat, int* info)
{
arma_type_check(( is_supported_blas_type<eT>::value == false ));
if(is_float<eT>::value)
{
arma_wrapper(sgssv)(options, A, perm_c, perm_r, L, U, B, stat, info);
}
else
if(is_double<eT>::value)
{
arma_wrapper(dgssv)(options, A, perm_c, perm_r, L, U, B, stat, info);
}
else
if(is_supported_complex_float<eT>::value)
{
arma_wrapper(cgssv)(options, A, perm_c, perm_r, L, U, B, stat, info);
}
else
if(is_supported_complex_double<eT>::value)
{
arma_wrapper(zgssv)(options, A, perm_c, perm_r, L, U, B, stat, info);
}
}
inline
int
clapack_posv(const enum CBLAS_ORDER Order, const enum CBLAS_UPLO Uplo, const int N, const int NRHS, eT *A, const int lda, eT *B, const int ldb)
{
arma_type_check((is_supported_blas_type<eT>::value == false));
if(is_float<eT>::value)
{
typedef float T;
return arma_wrapper(clapack_sposv)(Order, Uplo, N, NRHS, (T*)A, lda, (T*)B, ldb);
}
else
if(is_double<eT>::value)
{
typedef double T;
return arma_wrapper(clapack_dposv)(Order, Uplo, N, NRHS, (T*)A, lda, (T*)B, ldb);
}
else
if(is_supported_complex_float<eT>::value)
{
typedef std::complex<float> T;
return arma_wrapper(clapack_cposv)(Order, Uplo, N, NRHS, (T*)A, lda, (T*)B, ldb);
}
else
if(is_supported_complex_double<eT>::value)
{
typedef std::complex<double> T;
return arma_wrapper(clapack_zposv)(Order, Uplo, N, NRHS, (T*)A, lda, (T*)B, ldb);
}
return -1;
}
inline
void
gssvx(
superlu_options_t* opts,
SuperMatrix* A,
int* perm_c, int* perm_r,
int* etree, char* equed,
typename get_pod_type<eT>::result* R, typename get_pod_type<eT>::result* C,
SuperMatrix* L, SuperMatrix* U,
void* work, int lwork,
SuperMatrix* B, SuperMatrix* X,
typename get_pod_type<eT>::result* rpg, typename get_pod_type<eT>::result* rcond,
typename get_pod_type<eT>::result* ferr, typename get_pod_type<eT>::result* berr,
mem_usage_t* mu, SuperLUStat_t* stat, int* info
)
{
arma_type_check(( is_supported_blas_type<eT>::value == false ));
if(is_float<eT>::value)
{
typedef float T;
arma_wrapper(sgssvx)(opts, A, perm_c, perm_r, etree, equed, (T*)R, (T*)C, L, U, work, lwork, B, X, (T*)rpg, (T*)rcond, (T*)ferr, (T*)berr, mu, stat, info);
}
else
if(is_double<eT>::value)
{
typedef double T;
arma_wrapper(dgssvx)(opts, A, perm_c, perm_r, etree, equed, (T*)R, (T*)C, L, U, work, lwork, B, X, (T*)rpg, (T*)rcond, (T*)ferr, (T*)berr, mu, stat, info);
}
else
if(is_supported_complex_float<eT>::value)
{
typedef float T;
arma_wrapper(cgssvx)(opts, A, perm_c, perm_r, etree, equed, (T*)R, (T*)C, L, U, work, lwork, B, X, (T*)rpg, (T*)rcond, (T*)ferr, (T*)berr, mu, stat, info);
}
else
if(is_supported_complex_double<eT>::value)
{
typedef double T;
arma_wrapper(zgssvx)(opts, A, perm_c, perm_r, etree, equed, (T*)R, (T*)C, L, U, work, lwork, B, X, (T*)rpg, (T*)rcond, (T*)ferr, (T*)berr, mu, stat, info);
}
}
arma_inline
eT
cblas_dot(const int N, const eT* X, const eT* Y)
{
arma_type_check((is_supported_blas_type<eT>::value == false));
if(is_float<eT>::value)
{
typedef float T;
return eT( arma_wrapper(cblas_sdot)(N, (const T*)X, 1, (const T*)Y, 1) );
}
else
if(is_double<eT>::value)
{
typedef double T;
return eT( arma_wrapper(cblas_ddot)(N, (const T*)X, 1, (const T*)Y, 1) );
}
return eT(0);
}
inline
void
cblas_gemv
(
const enum CBLAS_ORDER Order, const enum CBLAS_TRANSPOSE TransA,
const int M, const int N,
const eT alpha,
const eT *A, const int lda,
const eT *X, const int incX,
const eT beta,
eT *Y, const int incY
)
{
arma_type_check((is_supported_blas_type<eT>::value == false));
if(is_float<eT>::value)
{
typedef float T;
arma_wrapper(cblas_sgemv)(Order, TransA, M, N, (const T)tmp_real(alpha), (const T*)A, lda, (const T*)X, incX, (const T)tmp_real(beta), (T*)Y, incY);
}
else
if(is_double<eT>::value)
{
typedef double T;
arma_wrapper(cblas_dgemv)(Order, TransA, M, N, (const T)tmp_real(alpha), (const T*)A, lda, (const T*)X, incX, (const T)tmp_real(beta), (T*)Y, incY);
}
else
if(is_supported_complex_float<eT>::value)
{
typedef std::complex<float> T;
arma_wrapper(cblas_cgemv)(Order, TransA, M, N, (const T*)&alpha, (const T*)A, lda, (const T*)X, incX, (const T*)&beta, (T*)Y, incY);
}
else
if(is_supported_complex_double<eT>::value)
{
typedef std::complex<double> T;
arma_wrapper(cblas_zgemv)(Order, TransA, M, N, (const T*)&alpha, (const T*)A, lda, (const T*)X, incX, (const T*)&beta, (T*)Y, incY);
}
}
inline
void
cblas_syrk
(
const enum CBLAS_ORDER Order, const enum CBLAS_UPLO Uplo, const enum CBLAS_TRANSPOSE Trans,
const int N, const int K, const eT alpha,
const eT* A, const int lda, const eT beta, eT* C, const int ldc
)
{
arma_type_check((is_supported_blas_type<eT>::value == false));
if(is_float<eT>::value)
{
typedef float T;
arma_wrapper(cblas_ssyrk)(Order, Uplo, Trans, N, K, (const T)alpha, (const T*)A, lda, (const T)beta, (T*)C, ldc);
}
else
if(is_double<eT>::value)
{
typedef double T;
arma_wrapper(cblas_dsyrk)(Order, Uplo, Trans, N, K, (const T)alpha, (const T*)A, lda, (const T)beta, (T*)C, ldc);
}
}
inline
int
clapack_getri
(
const enum CBLAS_ORDER Order, const int N, eT *A,
const int lda, const int *ipiv
)
{
arma_type_check((is_supported_blas_type<eT>::value == false));
if(is_float<eT>::value)
{
typedef float T;
return arma_wrapper(clapack_sgetri)(Order, N, (T*)A, lda, ipiv);
}
else
if(is_double<eT>::value)
{
typedef double T;
return arma_wrapper(clapack_dgetri)(Order, N, (T*)A, lda, ipiv);
}
else
if(is_supported_complex_float<eT>::value)
{
typedef std::complex<float> T;
return arma_wrapper(clapack_cgetri)(Order, N, (T*)A, lda, ipiv);
}
else
if(is_supported_complex_double<eT>::value)
{
typedef std::complex<double> T;
return arma_wrapper(clapack_zgetri)(Order, N, (T*)A, lda, ipiv);
}
return -1;
}
inline
void
free_stat(SuperLUStat_t* stat)
{
arma_wrapper(StatFree)(stat);
}
inline
void
destroy_supernode_mat(SuperMatrix* a)
{
arma_wrapper(Destroy_SuperNode_Matrix)(a);
}
inline
void
destroy_compcol_mat(SuperMatrix* a)
{
arma_wrapper(Destroy_CompCol_Matrix)(a);
}
inline
void
destroy_dense_mat(SuperMatrix* a)
{
arma_wrapper(Destroy_SuperMatrix_Store)(a);
}
inline
void
init_stat(SuperLUStat_t* stat)
{
arma_wrapper(StatInit)(stat);
}
inline
void
set_default_opts(superlu_options_t* opts)
{
arma_wrapper(set_default_options)(opts);
}
inline
void*
malloc(size_t N)
{
return arma_wrapper(superlu_malloc)(N);
}
inline
void
free(void* mem)
{
arma_wrapper(superlu_free)(mem);
}
