Chris@49: // Copyright (C) 2008-2012 NICTA (www.nicta.com.au) Chris@49: // Copyright (C) 2008-2012 Conrad Sanderson Chris@49: // Copyright (C) 2009 Edmund Highcock Chris@49: // Copyright (C) 2011 James Sanders Chris@49: // Copyright (C) 2012 Eric Jon Sundstrom Chris@49: // Chris@49: // This Source Code Form is subject to the terms of the Mozilla Public Chris@49: // License, v. 2.0. If a copy of the MPL was not distributed with this Chris@49: // file, You can obtain one at http://mozilla.org/MPL/2.0/. Chris@49: Chris@49: Chris@49: Chris@49: #ifdef ARMA_USE_LAPACK Chris@49: Chris@49: Chris@49: //! \namespace lapack namespace for LAPACK functions Chris@49: namespace lapack Chris@49: { Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: getrf(blas_int* m, blas_int* n, eT* a, blas_int* lda, blas_int* ipiv, blas_int* info) Chris@49: { Chris@49: arma_type_check(( is_supported_blas_type::value == false )); Chris@49: Chris@49: if(is_float::value == true) Chris@49: { Chris@49: typedef float T; Chris@49: arma_fortran(arma_sgetrf)(m, n, (T*)a, lda, ipiv, info); Chris@49: } Chris@49: else Chris@49: if(is_double::value == true) Chris@49: { Chris@49: typedef double T; Chris@49: arma_fortran(arma_dgetrf)(m, n, (T*)a, lda, ipiv, info); Chris@49: } Chris@49: else Chris@49: if(is_supported_complex_float::value == true) Chris@49: { Chris@49: typedef std::complex T; Chris@49: arma_fortran(arma_cgetrf)(m, n, (T*)a, lda, ipiv, info); Chris@49: } Chris@49: else Chris@49: if(is_supported_complex_double::value == true) Chris@49: { Chris@49: typedef std::complex T; Chris@49: arma_fortran(arma_zgetrf)(m, n, (T*)a, lda, ipiv, info); Chris@49: } Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: getri(blas_int* n, eT* a, blas_int* lda, blas_int* ipiv, eT* work, blas_int* lwork, blas_int* info) Chris@49: { Chris@49: arma_type_check(( is_supported_blas_type::value == false )); Chris@49: Chris@49: if(is_float::value == true) Chris@49: { Chris@49: typedef float T; Chris@49: arma_fortran(arma_sgetri)(n, (T*)a, lda, ipiv, (T*)work, lwork, info); Chris@49: } Chris@49: else Chris@49: if(is_double::value == true) Chris@49: { Chris@49: typedef double T; Chris@49: arma_fortran(arma_dgetri)(n, (T*)a, lda, ipiv, (T*)work, lwork, info); Chris@49: } Chris@49: else Chris@49: if(is_supported_complex_float::value == true) Chris@49: { Chris@49: typedef std::complex T; Chris@49: arma_fortran(arma_cgetri)(n, (T*)a, lda, ipiv, (T*)work, lwork, info); Chris@49: } Chris@49: else Chris@49: if(is_supported_complex_double::value == true) Chris@49: { Chris@49: typedef std::complex T; Chris@49: arma_fortran(arma_zgetri)(n, (T*)a, lda, ipiv, (T*)work, lwork, info); Chris@49: } Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: trtri(char* uplo, char* diag, blas_int* n, eT* a, blas_int* lda, blas_int* info) Chris@49: { Chris@49: arma_type_check(( is_supported_blas_type::value == false )); Chris@49: Chris@49: if(is_float::value == true) Chris@49: { Chris@49: typedef float T; Chris@49: arma_fortran(arma_strtri)(uplo, diag, n, (T*)a, lda, info); Chris@49: } Chris@49: else Chris@49: if(is_double::value == true) Chris@49: { Chris@49: typedef double T; Chris@49: arma_fortran(arma_dtrtri)(uplo, diag, n, (T*)a, lda, info); Chris@49: } Chris@49: else Chris@49: if(is_supported_complex_float::value == true) Chris@49: { Chris@49: typedef std::complex T; Chris@49: arma_fortran(arma_ctrtri)(uplo, diag, n, (T*)a, lda, info); Chris@49: } Chris@49: else Chris@49: if(is_supported_complex_double::value == true) Chris@49: { Chris@49: typedef std::complex T; Chris@49: arma_fortran(arma_ztrtri)(uplo, diag, n, (T*)a, lda, info); Chris@49: } Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: syev(char* jobz, char* uplo, blas_int* n, eT* a, blas_int* lda, eT* w, eT* work, blas_int* lwork, blas_int* info) Chris@49: { Chris@49: arma_type_check(( is_supported_blas_type::value == false )); Chris@49: Chris@49: if(is_float::value == true) Chris@49: { Chris@49: typedef float T; Chris@49: arma_fortran(arma_ssyev)(jobz, uplo, n, (T*)a, lda, (T*)w, (T*)work, lwork, info); Chris@49: } Chris@49: else Chris@49: if(is_double::value == true) Chris@49: { Chris@49: typedef double T; Chris@49: arma_fortran(arma_dsyev)(jobz, uplo, n, (T*)a, lda, (T*)w, (T*)work, lwork, info); Chris@49: } Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: syevd(char* jobz, char* uplo, blas_int* n, eT* a, blas_int* lda, eT* w, eT* work, blas_int* lwork, blas_int* iwork, blas_int* liwork, blas_int* info) Chris@49: { Chris@49: arma_type_check(( is_supported_blas_type::value == false )); Chris@49: Chris@49: if(is_float::value == true) Chris@49: { Chris@49: typedef float T; Chris@49: arma_fortran(arma_ssyevd)(jobz, uplo, n, (T*)a, lda, (T*)w, (T*)work, lwork, iwork, liwork, info); Chris@49: } Chris@49: else Chris@49: if(is_double::value == true) Chris@49: { Chris@49: typedef double T; Chris@49: arma_fortran(arma_dsyevd)(jobz, uplo, n, (T*)a, lda, (T*)w, (T*)work, lwork, iwork, liwork, info); Chris@49: } Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: heev Chris@49: ( Chris@49: char* jobz, char* uplo, blas_int* n, Chris@49: eT* a, blas_int* lda, typename eT::value_type* w, Chris@49: eT* work, blas_int* lwork, typename eT::value_type* rwork, Chris@49: blas_int* info Chris@49: ) Chris@49: { Chris@49: arma_type_check(( is_supported_blas_type::value == false )); Chris@49: Chris@49: if(is_supported_complex_float::value == true) Chris@49: { Chris@49: typedef float T; Chris@49: typedef typename std::complex cx_T; Chris@49: arma_fortran(arma_cheev)(jobz, uplo, n, (cx_T*)a, lda, (T*)w, (cx_T*)work, lwork, (T*)rwork, info); Chris@49: } Chris@49: else Chris@49: if(is_supported_complex_double::value == true) Chris@49: { Chris@49: typedef double T; Chris@49: typedef typename std::complex cx_T; Chris@49: arma_fortran(arma_zheev)(jobz, uplo, n, (cx_T*)a, lda, (T*)w, (cx_T*)work, lwork, (T*)rwork, info); Chris@49: } Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: heevd Chris@49: ( Chris@49: char* jobz, char* uplo, blas_int* n, Chris@49: eT* a, blas_int* lda, typename eT::value_type* w, Chris@49: eT* work, blas_int* lwork, typename eT::value_type* rwork, Chris@49: blas_int* lrwork, blas_int* iwork, blas_int* liwork, Chris@49: blas_int* info Chris@49: ) Chris@49: { Chris@49: arma_type_check(( is_supported_blas_type::value == false )); Chris@49: Chris@49: if(is_supported_complex_float::value == true) Chris@49: { Chris@49: typedef float T; Chris@49: typedef typename std::complex cx_T; Chris@49: arma_fortran(arma_cheevd)(jobz, uplo, n, (cx_T*)a, lda, (T*)w, (cx_T*)work, lwork, (T*)rwork, lrwork, iwork, liwork, info); Chris@49: } Chris@49: else Chris@49: if(is_supported_complex_double::value == true) Chris@49: { Chris@49: typedef double T; Chris@49: typedef typename std::complex cx_T; Chris@49: arma_fortran(arma_zheevd)(jobz, uplo, n, (cx_T*)a, lda, (T*)w, (cx_T*)work, lwork, (T*)rwork, lrwork, iwork, liwork, info); Chris@49: } Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: geev Chris@49: ( Chris@49: char* jobvl, char* jobvr, blas_int* n, Chris@49: eT* a, blas_int* lda, eT* wr, eT* wi, eT* vl, Chris@49: blas_int* ldvl, eT* vr, blas_int* ldvr, Chris@49: eT* work, blas_int* lwork, Chris@49: blas_int* info Chris@49: ) Chris@49: { Chris@49: arma_type_check(( is_supported_blas_type::value == false )); Chris@49: Chris@49: if(is_float::value == true) Chris@49: { Chris@49: typedef float T; Chris@49: arma_fortran(arma_sgeev)(jobvl, jobvr, n, (T*)a, lda, (T*)wr, (T*)wi, (T*)vl, ldvl, (T*)vr, ldvr, (T*)work, lwork, info); Chris@49: } Chris@49: else Chris@49: if(is_double::value == true) Chris@49: { Chris@49: typedef double T; Chris@49: arma_fortran(arma_dgeev)(jobvl, jobvr, n, (T*)a, lda, (T*)wr, (T*)wi, (T*)vl, ldvl, (T*)vr, ldvr, (T*)work, lwork, info); Chris@49: } Chris@49: } Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: cx_geev Chris@49: ( Chris@49: char* jobvl, char* jobvr, blas_int* n, Chris@49: eT* a, blas_int* lda, eT* w, Chris@49: eT* vl, blas_int* ldvl, Chris@49: eT* vr, blas_int* ldvr, Chris@49: eT* work, blas_int* lwork, typename eT::value_type* rwork, Chris@49: blas_int* info Chris@49: ) Chris@49: { Chris@49: arma_type_check(( is_supported_blas_type::value == false )); Chris@49: Chris@49: if(is_supported_complex_float::value == true) Chris@49: { Chris@49: typedef float T; Chris@49: typedef typename std::complex cx_T; Chris@49: arma_fortran(arma_cgeev)(jobvl, jobvr, n, (cx_T*)a, lda, (cx_T*)w, (cx_T*)vl, ldvl, (cx_T*)vr, ldvr, (cx_T*)work, lwork, (T*)rwork, info); Chris@49: } Chris@49: else Chris@49: if(is_supported_complex_double::value == true) Chris@49: { Chris@49: typedef double T; Chris@49: typedef typename std::complex cx_T; Chris@49: arma_fortran(arma_zgeev)(jobvl, jobvr, n, (cx_T*)a, lda, (cx_T*)w, (cx_T*)vl, ldvl, (cx_T*)vr, ldvr, (cx_T*)work, lwork, (T*)rwork, info); Chris@49: } Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: potrf(char* uplo, blas_int* n, eT* a, blas_int* lda, blas_int* info) Chris@49: { Chris@49: arma_type_check(( is_supported_blas_type::value == false )); Chris@49: Chris@49: if(is_float::value == true) Chris@49: { Chris@49: typedef float T; Chris@49: arma_fortran(arma_spotrf)(uplo, n, (T*)a, lda, info); Chris@49: } Chris@49: else Chris@49: if(is_double::value == true) Chris@49: { Chris@49: typedef double T; Chris@49: arma_fortran(arma_dpotrf)(uplo, n, (T*)a, lda, info); Chris@49: } Chris@49: else Chris@49: if(is_supported_complex_float::value == true) Chris@49: { Chris@49: typedef std::complex T; Chris@49: arma_fortran(arma_cpotrf)(uplo, n, (T*)a, lda, info); Chris@49: } Chris@49: else Chris@49: if(is_supported_complex_double::value == true) Chris@49: { Chris@49: typedef std::complex T; Chris@49: arma_fortran(arma_zpotrf)(uplo, n, (T*)a, lda, info); Chris@49: } Chris@49: Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: potri(char* uplo, blas_int* n, eT* a, blas_int* lda, blas_int* info) Chris@49: { Chris@49: arma_type_check(( is_supported_blas_type::value == false )); Chris@49: Chris@49: if(is_float::value == true) Chris@49: { Chris@49: typedef float T; Chris@49: arma_fortran(arma_spotri)(uplo, n, (T*)a, lda, info); Chris@49: } Chris@49: else Chris@49: if(is_double::value == true) Chris@49: { Chris@49: typedef double T; Chris@49: arma_fortran(arma_dpotri)(uplo, n, (T*)a, lda, info); Chris@49: } Chris@49: else Chris@49: if(is_supported_complex_float::value == true) Chris@49: { Chris@49: typedef std::complex T; Chris@49: arma_fortran(arma_cpotri)(uplo, n, (T*)a, lda, info); Chris@49: } Chris@49: else Chris@49: if(is_supported_complex_double::value == true) Chris@49: { Chris@49: typedef std::complex T; Chris@49: arma_fortran(arma_zpotri)(uplo, n, (T*)a, lda, info); Chris@49: } Chris@49: Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: geqrf(blas_int* m, blas_int* n, eT* a, blas_int* lda, eT* tau, eT* work, blas_int* lwork, blas_int* info) Chris@49: { Chris@49: arma_type_check(( is_supported_blas_type::value == false )); Chris@49: Chris@49: if(is_float::value == true) Chris@49: { Chris@49: typedef float T; Chris@49: arma_fortran(arma_sgeqrf)(m, n, (T*)a, lda, (T*)tau, (T*)work, lwork, info); Chris@49: } Chris@49: else Chris@49: if(is_double::value == true) Chris@49: { Chris@49: typedef double T; Chris@49: arma_fortran(arma_dgeqrf)(m, n, (T*)a, lda, (T*)tau, (T*)work, lwork, info); Chris@49: } Chris@49: else Chris@49: if(is_supported_complex_float::value == true) Chris@49: { Chris@49: typedef std::complex T; Chris@49: arma_fortran(arma_cgeqrf)(m, n, (T*)a, lda, (T*)tau, (T*)work, lwork, info); Chris@49: } Chris@49: else Chris@49: if(is_supported_complex_double::value == true) Chris@49: { Chris@49: typedef std::complex T; Chris@49: arma_fortran(arma_zgeqrf)(m, n, (T*)a, lda, (T*)tau, (T*)work, lwork, info); Chris@49: } Chris@49: Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: orgqr(blas_int* m, blas_int* n, blas_int* k, eT* a, blas_int* lda, eT* tau, eT* work, blas_int* lwork, blas_int* info) Chris@49: { Chris@49: arma_type_check(( is_supported_blas_type::value == false )); Chris@49: Chris@49: if(is_float::value == true) Chris@49: { Chris@49: typedef float T; Chris@49: arma_fortran(arma_sorgqr)(m, n, k, (T*)a, lda, (T*)tau, (T*)work, lwork, info); Chris@49: } Chris@49: else Chris@49: if(is_double::value == true) Chris@49: { Chris@49: typedef double T; Chris@49: arma_fortran(arma_dorgqr)(m, n, k, (T*)a, lda, (T*)tau, (T*)work, lwork, info); Chris@49: } Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: ungqr(blas_int* m, blas_int* n, blas_int* k, eT* a, blas_int* lda, eT* tau, eT* work, blas_int* lwork, blas_int* info) Chris@49: { Chris@49: arma_type_check(( is_supported_blas_type::value == false )); Chris@49: Chris@49: if(is_supported_complex_float::value == true) Chris@49: { Chris@49: typedef float T; Chris@49: arma_fortran(arma_cungqr)(m, n, k, (T*)a, lda, (T*)tau, (T*)work, lwork, info); Chris@49: } Chris@49: else Chris@49: if(is_supported_complex_double::value == true) Chris@49: { Chris@49: typedef double T; Chris@49: arma_fortran(arma_zungqr)(m, n, k, (T*)a, lda, (T*)tau, (T*)work, lwork, info); Chris@49: } Chris@49: } Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: gesvd Chris@49: ( Chris@49: char* jobu, char* jobvt, blas_int* m, blas_int* n, eT* a, blas_int* lda, Chris@49: eT* s, eT* u, blas_int* ldu, eT* vt, blas_int* ldvt, Chris@49: eT* work, blas_int* lwork, blas_int* info Chris@49: ) Chris@49: { Chris@49: arma_type_check(( is_supported_blas_type::value == false )); Chris@49: Chris@49: if(is_float::value == true) Chris@49: { Chris@49: typedef float T; Chris@49: arma_fortran(arma_sgesvd)(jobu, jobvt, m, n, (T*)a, lda, (T*)s, (T*)u, ldu, (T*)vt, ldvt, (T*)work, lwork, info); Chris@49: } Chris@49: else Chris@49: if(is_double::value == true) Chris@49: { Chris@49: typedef double T; Chris@49: arma_fortran(arma_dgesvd)(jobu, jobvt, m, n, (T*)a, lda, (T*)s, (T*)u, ldu, (T*)vt, ldvt, (T*)work, lwork, info); Chris@49: } Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: cx_gesvd Chris@49: ( Chris@49: char* jobu, char* jobvt, blas_int* m, blas_int* n, std::complex* a, blas_int* lda, Chris@49: T* s, std::complex* u, blas_int* ldu, std::complex* vt, blas_int* ldvt, Chris@49: std::complex* work, blas_int* lwork, T* rwork, blas_int* info Chris@49: ) Chris@49: { Chris@49: arma_type_check(( is_supported_blas_type::value == false )); Chris@49: arma_type_check(( is_supported_blas_type< std::complex >::value == false )); Chris@49: Chris@49: if(is_float::value == true) Chris@49: { Chris@49: typedef float bT; Chris@49: arma_fortran(arma_cgesvd) Chris@49: ( Chris@49: jobu, jobvt, m, n, (std::complex*)a, lda, Chris@49: (bT*)s, (std::complex*)u, ldu, (std::complex*)vt, ldvt, Chris@49: (std::complex*)work, lwork, (bT*)rwork, info Chris@49: ); Chris@49: } Chris@49: else Chris@49: if(is_double::value == true) Chris@49: { Chris@49: typedef double bT; Chris@49: arma_fortran(arma_zgesvd) Chris@49: ( Chris@49: jobu, jobvt, m, n, (std::complex*)a, lda, Chris@49: (bT*)s, (std::complex*)u, ldu, (std::complex*)vt, ldvt, Chris@49: (std::complex*)work, lwork, (bT*)rwork, info Chris@49: ); Chris@49: } Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: gesdd Chris@49: ( Chris@49: char* jobz, blas_int* m, blas_int* n, Chris@49: eT* a, blas_int* lda, eT* s, eT* u, blas_int* ldu, eT* vt, blas_int* ldvt, Chris@49: eT* work, blas_int* lwork, blas_int* iwork, blas_int* info Chris@49: ) Chris@49: { Chris@49: arma_type_check(( is_supported_blas_type::value == false )); Chris@49: Chris@49: if(is_float::value == true) Chris@49: { Chris@49: typedef float T; Chris@49: arma_fortran(arma_sgesdd)(jobz, m, n, (T*)a, lda, (T*)s, (T*)u, ldu, (T*)vt, ldvt, (T*)work, lwork, iwork, info); Chris@49: } Chris@49: else Chris@49: if(is_double::value == true) Chris@49: { Chris@49: typedef double T; Chris@49: arma_fortran(arma_dgesdd)(jobz, m, n, (T*)a, lda, (T*)s, (T*)u, ldu, (T*)vt, ldvt, (T*)work, lwork, iwork, info); Chris@49: } Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: cx_gesdd Chris@49: ( Chris@49: char* jobz, blas_int* m, blas_int* n, Chris@49: std::complex* a, blas_int* lda, T* s, std::complex* u, blas_int* ldu, std::complex* vt, blas_int* ldvt, Chris@49: std::complex* work, blas_int* lwork, T* rwork, blas_int* iwork, blas_int* info Chris@49: ) Chris@49: { Chris@49: arma_type_check(( is_supported_blas_type::value == false )); Chris@49: arma_type_check(( is_supported_blas_type< std::complex >::value == false )); Chris@49: Chris@49: if(is_float::value == true) Chris@49: { Chris@49: typedef float bT; Chris@49: arma_fortran(arma_cgesdd) Chris@49: ( Chris@49: jobz, m, n, Chris@49: (std::complex*)a, lda, (bT*)s, (std::complex*)u, ldu, (std::complex*)vt, ldvt, Chris@49: (std::complex*)work, lwork, (bT*)rwork, iwork, info Chris@49: ); Chris@49: } Chris@49: else Chris@49: if(is_double::value == true) Chris@49: { Chris@49: typedef double bT; Chris@49: arma_fortran(arma_zgesdd) Chris@49: ( Chris@49: jobz, m, n, Chris@49: (std::complex*)a, lda, (bT*)s, (std::complex*)u, ldu, (std::complex*)vt, ldvt, Chris@49: (std::complex*)work, lwork, (bT*)rwork, iwork, info Chris@49: ); Chris@49: } Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: gesv(blas_int* n, blas_int* nrhs, eT* a, blas_int* lda, blas_int* ipiv, eT* b, blas_int* ldb, blas_int* info) Chris@49: { Chris@49: arma_type_check(( is_supported_blas_type::value == false )); Chris@49: Chris@49: if(is_float::value == true) Chris@49: { Chris@49: typedef float T; Chris@49: arma_fortran(arma_sgesv)(n, nrhs, (T*)a, lda, ipiv, (T*)b, ldb, info); Chris@49: } Chris@49: else Chris@49: if(is_double::value == true) Chris@49: { Chris@49: typedef double T; Chris@49: arma_fortran(arma_dgesv)(n, nrhs, (T*)a, lda, ipiv, (T*)b, ldb, info); Chris@49: } Chris@49: else Chris@49: if(is_supported_complex_float::value == true) Chris@49: { Chris@49: typedef std::complex T; Chris@49: arma_fortran(arma_cgesv)(n, nrhs, (T*)a, lda, ipiv, (T*)b, ldb, info); Chris@49: } Chris@49: else Chris@49: if(is_supported_complex_double::value == true) Chris@49: { Chris@49: typedef std::complex T; Chris@49: arma_fortran(arma_zgesv)(n, nrhs, (T*)a, lda, ipiv, (T*)b, ldb, info); Chris@49: } Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: gels(char* trans, blas_int* m, blas_int* n, blas_int* nrhs, eT* a, blas_int* lda, eT* b, blas_int* ldb, eT* work, blas_int* lwork, blas_int* info) Chris@49: { Chris@49: arma_type_check(( is_supported_blas_type::value == false )); Chris@49: Chris@49: if(is_float::value == true) Chris@49: { Chris@49: typedef float T; Chris@49: arma_fortran(arma_sgels)(trans, m, n, nrhs, (T*)a, lda, (T*)b, ldb, (T*)work, lwork, info); Chris@49: } Chris@49: else Chris@49: if(is_double::value == true) Chris@49: { Chris@49: typedef double T; Chris@49: arma_fortran(arma_dgels)(trans, m, n, nrhs, (T*)a, lda, (T*)b, ldb, (T*)work, lwork, info); Chris@49: } Chris@49: else Chris@49: if(is_supported_complex_float::value == true) Chris@49: { Chris@49: typedef std::complex T; Chris@49: arma_fortran(arma_cgels)(trans, m, n, nrhs, (T*)a, lda, (T*)b, ldb, (T*)work, lwork, info); Chris@49: } Chris@49: else Chris@49: if(is_supported_complex_double::value == true) Chris@49: { Chris@49: typedef std::complex T; Chris@49: arma_fortran(arma_zgels)(trans, m, n, nrhs, (T*)a, lda, (T*)b, ldb, (T*)work, lwork, info); Chris@49: } Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: trtrs(char* uplo, char* trans, char* diag, blas_int* n, blas_int* nrhs, const eT* a, blas_int* lda, eT* b, blas_int* ldb, blas_int* info) Chris@49: { Chris@49: arma_type_check(( is_supported_blas_type::value == false )); Chris@49: Chris@49: if(is_float::value == true) Chris@49: { Chris@49: typedef float T; Chris@49: arma_fortran(arma_strtrs)(uplo, trans, diag, n, nrhs, (T*)a, lda, (T*)b, ldb, info); Chris@49: } Chris@49: else Chris@49: if(is_double::value == true) Chris@49: { Chris@49: typedef double T; Chris@49: arma_fortran(arma_dtrtrs)(uplo, trans, diag, n, nrhs, (T*)a, lda, (T*)b, ldb, info); Chris@49: } Chris@49: else Chris@49: if(is_supported_complex_float::value == true) Chris@49: { Chris@49: typedef std::complex T; Chris@49: arma_fortran(arma_ctrtrs)(uplo, trans, diag, n, nrhs, (T*)a, lda, (T*)b, ldb, info); Chris@49: } Chris@49: else Chris@49: if(is_supported_complex_double::value == true) Chris@49: { Chris@49: typedef std::complex T; Chris@49: arma_fortran(arma_ztrtrs)(uplo, trans, diag, n, nrhs, (T*)a, lda, (T*)b, ldb, info); Chris@49: } Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: gees(char* jobvs, char* sort, blas_int* select, blas_int* n, eT* a, blas_int* lda, blas_int* sdim, eT* wr, eT* wi, eT* vs, blas_int* ldvs, eT* work, blas_int* lwork, blas_int* bwork, blas_int* info) Chris@49: { Chris@49: arma_type_check(( is_supported_blas_type::value == false )); Chris@49: Chris@49: if(is_float::value == true) Chris@49: { Chris@49: typedef float T; Chris@49: arma_fortran(arma_sgees)(jobvs, sort, select, n, (T*)a, lda, sdim, (T*)wr, (T*)wi, (T*)vs, ldvs, (T*)work, lwork, bwork, info); Chris@49: } Chris@49: else Chris@49: if(is_double::value == true) Chris@49: { Chris@49: typedef double T; Chris@49: arma_fortran(arma_dgees)(jobvs, sort, select, n, (T*)a, lda, sdim, (T*)wr, (T*)wi, (T*)vs, ldvs, (T*)work, lwork, bwork, info); Chris@49: } Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: cx_gees(char* jobvs, char* sort, blas_int* select, blas_int* n, std::complex* a, blas_int* lda, blas_int* sdim, std::complex* w, std::complex* vs, blas_int* ldvs, std::complex* work, blas_int* lwork, T* rwork, blas_int* bwork, blas_int* info) Chris@49: { Chris@49: arma_type_check(( is_supported_blas_type::value == false )); Chris@49: arma_type_check(( is_supported_blas_type< std::complex >::value == false )); Chris@49: Chris@49: if(is_float::value == true) Chris@49: { Chris@49: typedef float bT; Chris@49: typedef std::complex cT; Chris@49: arma_fortran(arma_cgees)(jobvs, sort, select, n, (cT*)a, lda, sdim, (cT*)w, (cT*)vs, ldvs, (cT*)work, lwork, (bT*)rwork, bwork, info); Chris@49: } Chris@49: else Chris@49: if(is_double::value == true) Chris@49: { Chris@49: typedef double bT; Chris@49: typedef std::complex cT; Chris@49: arma_fortran(arma_zgees)(jobvs, sort, select, n, (cT*)a, lda, sdim, (cT*)w, (cT*)vs, ldvs, (cT*)work, lwork, (bT*)rwork, bwork, info); Chris@49: } Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: trsyl(char* transa, char* transb, blas_int* isgn, blas_int* m, blas_int* n, const eT* a, blas_int* lda, const eT* b, blas_int* ldb, eT* c, blas_int* ldc, eT* scale, blas_int* info) Chris@49: { Chris@49: arma_type_check(( is_supported_blas_type::value == false )); Chris@49: Chris@49: if(is_float::value == true) Chris@49: { Chris@49: typedef float T; Chris@49: arma_fortran(arma_strsyl)(transa, transb, isgn, m, n, (T*)a, lda, (T*)b, ldb, (T*)c, ldc, (T*)scale, info); Chris@49: } Chris@49: else Chris@49: if(is_double::value == true) Chris@49: { Chris@49: typedef double T; Chris@49: arma_fortran(arma_dtrsyl)(transa, transb, isgn, m, n, (T*)a, lda, (T*)b, ldb, (T*)c, ldc, (T*)scale, info); Chris@49: } Chris@49: else Chris@49: if(is_supported_complex_float::value == true) Chris@49: { Chris@49: typedef std::complex T; Chris@49: arma_fortran(arma_ctrsyl)(transa, transb, isgn, m, n, (T*)a, lda, (T*)b, ldb, (T*)c, ldc, (float*)scale, info); Chris@49: } Chris@49: else Chris@49: if(is_supported_complex_double::value == true) Chris@49: { Chris@49: typedef std::complex T; Chris@49: arma_fortran(arma_ztrsyl)(transa, transb, isgn, m, n, (T*)a, lda, (T*)b, ldb, (T*)c, ldc, (double*)scale, info); Chris@49: } Chris@49: } Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: sytrf(char* uplo, blas_int* n, eT* a, blas_int* lda, blas_int* ipiv, eT* work, blas_int* lwork, blas_int* info) Chris@49: { Chris@49: arma_type_check(( is_supported_blas_type::value == false )); Chris@49: Chris@49: if(is_float::value == true) Chris@49: { Chris@49: typedef float T; Chris@49: arma_fortran(arma_ssytrf)(uplo, n, (T*)a, lda, ipiv, (T*)work, lwork, info); Chris@49: } Chris@49: else Chris@49: if(is_double::value == true) Chris@49: { Chris@49: typedef double T; Chris@49: arma_fortran(arma_dsytrf)(uplo, n, (T*)a, lda, ipiv, (T*)work, lwork, info); Chris@49: } Chris@49: else Chris@49: if(is_supported_complex_float::value == true) Chris@49: { Chris@49: typedef std::complex T; Chris@49: arma_fortran(arma_csytrf)(uplo, n, (T*)a, lda, ipiv, (T*)work, lwork, info); Chris@49: } Chris@49: else Chris@49: if(is_supported_complex_double::value == true) Chris@49: { Chris@49: typedef std::complex T; Chris@49: arma_fortran(arma_zsytrf)(uplo, n, (T*)a, lda, ipiv, (T*)work, lwork, info); Chris@49: } Chris@49: } Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: sytri(char* uplo, blas_int* n, eT* a, blas_int* lda, blas_int* ipiv, eT* work, blas_int* info) Chris@49: { Chris@49: arma_type_check(( is_supported_blas_type::value == false )); Chris@49: Chris@49: if(is_float::value == true) Chris@49: { Chris@49: typedef float T; Chris@49: arma_fortran(arma_ssytri)(uplo, n, (T*)a, lda, ipiv, (T*)work, info); Chris@49: } Chris@49: else Chris@49: if(is_double::value == true) Chris@49: { Chris@49: typedef double T; Chris@49: arma_fortran(arma_dsytri)(uplo, n, (T*)a, lda, ipiv, (T*)work, info); Chris@49: } Chris@49: else Chris@49: if(is_supported_complex_float::value == true) Chris@49: { Chris@49: typedef std::complex T; Chris@49: arma_fortran(arma_csytri)(uplo, n, (T*)a, lda, ipiv, (T*)work, info); Chris@49: } Chris@49: else Chris@49: if(is_supported_complex_double::value == true) Chris@49: { Chris@49: typedef std::complex T; Chris@49: arma_fortran(arma_zsytri)(uplo, n, (T*)a, lda, ipiv, (T*)work, info); Chris@49: } Chris@49: } Chris@49: Chris@49: Chris@49: } Chris@49: Chris@49: Chris@49: #endif