{-# LANGUAGE ForeignFunctionInterface, GeneralizedNewtypeDeriving #-} module Numerical.HBLAS.BLAS.FFI where import Foreign.Ptr import Foreign() import Foreign.C.Types import Data.Complex -- /*Set the number of threads on runtime.*/ --foreign import ccall unsafe "openblas_set_num_threads" openblas_set_num_threads_unsafe :: CInt -> IO () --foreign import ccall unsafe "goto_set_num_threads" goto_set_num_threads_unsafe :: CInt -> IO () {- | For All of the BlAS FFI operations, -} {- typedef enum CBLAS_ORDER {CblasRowMajor=101, CblasColMajor=102} CBLAS_ORDER; typedef enum CBLAS_TRANSPOSE {CblasNoTrans=111, CblasTrans=112, CblasConjTrans=113, CblasConjNoTrans=114} CBLAS_TRANSPOSE; typedef enum CBLAS_UPLO {CblasUpper=121, CblasLower=122} CBLAS_UPLO; typedef enum CBLAS_DIAG {CblasNonUnit=131, CblasUnit=132} CBLAS_DIAG; typedef enum CBLAS_SIDE {CblasLeft=141, CblasRight=142} CBLAS_SIDE; -} newtype CBLAS_INDEX = CBIndex CSize deriving (Eq,Show) newtype CBLAS_ORDERT = CBOInt CInt deriving (Eq,Show) data BLASOrder = BLASRowMajor | BLASColMajor deriving (Eq,Show) encodeOrder :: BLASOrder -> CBLAS_ORDERT encodeOrder BLASRowMajor = CBOInt 101 encodeOrder BLASColMajor = CBOInt 102 newtype CBLAS_TRANSPOSET = CBLAS_TransposeT{ unCBLAS_TransposeT :: CInt } deriving (Eq, Show) data BLAS_Transpose = BlasNoTranspose | BlasTranspose | BlasConjTranspose | BlasConjNoTranspose encodeTranpose :: BLAS_Transpose -> CBLAS_TRANSPOSET encodeTranpose BlasNoTranspose = CBLAS_TransposeT 111 encodeTranpose BlasTranspose = CBLAS_TransposeT 112 encodeTranpose BlasConjTranspose = CBLAS_TransposeT 113 encodeTranpose BlasConjNoTranspose = CBLAS_TransposeT 114 newtype CBLAS_UPLOT = CBlasUPLO CInt deriving (Eq,Show) data BLASUplo = BUpper | BLower deriving (Eq,Show) encodeUPLO :: BLASUplo -> CBLAS_UPLOT encodeUPLO BUpper = CBlasUPLO 121 encodeUPLO BLower = CBlasUPLO 122 newtype CBLAS_DIAGT = CBLAS_DiagT CUChar deriving (Show,Eq) data BlasDiag = BlasNonUnit | BlasUnit deriving (Eq,Show ) encodeDiag :: BlasDiag -> CBLAS_DIAGT encodeDiag BlasNonUnit = CBLAS_DiagT 131 encodeDiag BlasUnit = CBLAS_DiagT 132 newtype CBLAS_SIDET = CBLAS_SideT { unCBLAS_SideT :: CUChar } deriving (Eq, Show) data BlasSide = BlasLeft | BlasRight deriving (Eq,Show) encodeSide :: BlasSide -> CBLAS_SIDET encodeSide BlasLeft = CBLAS_SideT 141 encodeSide BlasRight = CBLAS_SideT 142 --typedef enum CBLAS_ORDER {CblasRowMajor=101, CblasColMajor=102} CBLAS_ORDER; --typedef enum CBLAS_TRANSPOSE {CblasNoTrans=111, CblasTrans=112, CblasConjTrans=113, CblasConjNoTrans=114} CBLAS_TRANSPOSE; --typedef enum CBLAS_UPLO {CblasUpper=121, CblasLower=122} CBLAS_UPLO; --typedef enum CBLAS_DIAG {CblasNonUnit=131, CblasUnit=132} CBLAS_DIAG; --typedef enum CBLAS_SIDE {CblasLeft=141, CblasRight=142} CBLAS_SIDE; --dot products foreign import ccall unsafe "cblas_sdsdot" cblas_sdsdot_unsafe :: CInt -> Float -> Ptr Float -> CInt -> Ptr Float -> CInt -> IO Float foreign import ccall unsafe "cblas_dsdot" cblas_dsdot_unsafe :: CInt -> Ptr Float -> CInt -> Ptr Float -> CInt -> IO Double foreign import ccall unsafe "cblas_sdot" cblas_sdot_unsafe :: CInt -> Ptr Float -> CInt -> Ptr Float -> CInt -> IO Float foreign import ccall unsafe "cblas_ddot" cblas_ddot_unsafe :: CInt -> Ptr Double -> CInt -> Ptr Double -> CInt -> IO Double --Float cblas_sdsdot( CInt n, Float alpha, Float *x, CInt incx, Float *y, CInt incy); --Double cblas_dsdot ( CInt n, Float *x, CInt incx, Float *y, CInt incy); --Float cblas_sdot( CInt n, Float *x, CInt incx, Float *y, CInt incy); --Double cblas_ddot( CInt n, Double *x, CInt incx, Double *y, CInt incy); {- not doing these right now, because requires handling return value as a complex number, we can only handle pointers to complex numbers right now -} --openblas_complex_Float cblas_cdotu( CInt n, Float *x, CInt incx, Float *y, CInt incy); --openblas_complex_Float cblas_cdotc( CInt n, Float *x, CInt incx, Float *y, CInt incy); --openblas_complex_Double cblas_zdotu( CInt n, Double *x, CInt incx, Double *y, CInt incy); --openblas_complex_Double cblas_zdotc( CInt n, Double *x, CInt incx, Double *y, CInt incy); --- not sure what to do for these complex --void cblas_cdotu_sub( CInt n, Float *x, CInt incx, Float *y, CInt incy, openblas_complex_Float *ret); --void cblas_cdotc_sub( CInt n, Float *x, CInt incx, Float *y, CInt incy, openblas_complex_Float *ret); --void cblas_zdotu_sub( CInt n, Double *x, CInt incx, Double *y, CInt incy, openblas_complex_Double *ret); --void cblas_zdotc_sub( CInt n, Double *x, CInt incx, Double *y, CInt incy, openblas_complex_Double *ret); ---- absolute value foreign import ccall unsafe "cblas_sasum" cblas_sasum_unsafe:: CInt -> Ptr Float -> CInt -> IO Float foreign import ccall unsafe "cblas_dasum" cblas_dasum_unsafe :: CInt -> Ptr Double -> CInt -> IO Double foreign import ccall unsafe "cblas_scasum" cblas_casum_unsafe :: CInt -> Ptr (Complex Float)-> CInt -> IO Float foreign import ccall unsafe "cblas_dzasum" cblas_zasum_unsafe :: CInt -> Ptr (Complex Double) -> CInt -> IO Double --Float cblas_sasum ( CInt n, Float *x, CInt incx); --Double cblas_dasum ( CInt n, Double *x, CInt incx); --Float cblas_scasum( CInt n, Float *x, CInt incx); --Double cblas_dzasum( CInt n, Double *x, CInt incx); foreign import ccall unsafe "cblas_snrm2" cblas_snrm2_unsafe :: CInt -> Ptr Float -> CInt -> IO Float foreign import ccall unsafe "cblas_dnrm2" cblas_dnrm2_unsafe :: CInt -> Ptr Double -> CInt -> IO Double foreign import ccall unsafe "cblas_scnrm2" cblas_scnrm2_unsafe :: CInt -> Ptr (Complex Float)-> CInt -> IO Float foreign import ccall unsafe "cblas_dznrm2" cblas_dznrm2_unsafe :: CInt -> Ptr (Complex Double) -> CInt -> IO Double --Float cblas_snrm2 ( CInt N, Float *X, CInt incX); --Double cblas_dnrm2 ( CInt N, Double *X, CInt incX); --Float cblas_scnrm2( CInt N, Float *X, CInt incX); --Double cblas_dznrm2( CInt N, Double *X, CInt incX); foreign import ccall unsafe "cblas_isamax" cblas_isamax_unsafe :: CInt -> Ptr Float -> CInt -> IO CInt foreign import ccall unsafe "cblas_idamax" cblas_idamax_unsafe :: CInt -> Ptr Float -> CInt -> IO CInt foreign import ccall unsafe "cblas_icamax" cblas_icamax_unsafe :: CInt -> Ptr (Complex Float) -> CInt -> IO CInt foreign import ccall unsafe "cblas_izamax" cblas_izamax_unsafe :: CInt -> Ptr (Complex Double) -> CInt -> IO CInt --CBLAS_INDEX cblas_isamax( CInt n, Float *x, CInt incx); --CBLAS_INDEX cblas_idamax( CInt n, Double *x, CInt incx); --CBLAS_INDEX cblas_icamax( CInt n, Float *x, CInt incx); --CBLAS_INDEX cblas_izamax( CInt n, Double *x, CInt incx); --void cblas_saxpy( CInt n, Float alpha, Float *x, CInt incx, Float *y, CInt incy); --void cblas_daxpy( CInt n, Double alpha, Double *x, CInt incx, Double *y, CInt incy); --void cblas_caxpy( CInt n, Float *alpha, Float *x, CInt incx, Float *y, CInt incy); --void cblas_zaxpy( CInt n, Double *alpha, Double *x, CInt incx, Double *y, CInt incy); foreign import ccall unsafe "cblas_scopy" cblas_scopy_unsafe :: CInt -> Ptr Float -> CInt -> Ptr Float -> CInt -> IO () foreign import ccall unsafe "cblas_dcopy" cblas_dcopy_unsafe :: CInt -> Ptr Double-> CInt -> Ptr Double -> CInt -> IO () foreign import ccall unsafe "cblas_ccopy" cblas_ccopy_unsafe :: CInt -> Ptr (Complex Float) -> CInt -> Ptr (Complex Float) -> CInt -> IO () foreign import ccall unsafe "cblas_zcopy" cblas_zcopy_unsafe :: CInt -> Ptr (Complex Double)-> CInt -> Ptr (Complex Double) -> CInt -> IO () --void cblas_scopy( CInt n, Float *x, CInt incx, Float *y, CInt incy); --void cblas_dcopy( CInt n, Double *x, CInt incx, Double *y, CInt incy); --void cblas_ccopy( CInt n, Float *x, CInt incx, Float *y, CInt incy); --void cblas_zcopy( CInt n, Double *x, CInt incx, Double *y, CInt incy); foreign import ccall unsafe "cblas_scopy" cblas_sswap_unsafe :: CInt -> Ptr Float -> CInt -> Ptr Float -> CInt -> IO () foreign import ccall unsafe "cblas_dcopy" cblas_dswap_unsafe :: CInt -> Ptr Double-> CInt -> Ptr Double -> CInt -> IO () foreign import ccall unsafe "cblas_ccopy" cblas_cswap_unsafe :: CInt -> Ptr (Complex Float) -> CInt -> Ptr (Complex Float) -> CInt -> IO () foreign import ccall unsafe "cblas_zcopy" cblas_zswap_unsafe :: CInt -> Ptr (Complex Double)-> CInt -> Ptr (Complex Double) -> CInt -> IO () --void cblas_sswap( CInt n, Float *x, CInt incx, Float *y, CInt incy); --void cblas_dswap( CInt n, Double *x, CInt incx, Double *y, CInt incy); --void cblas_cswap( CInt n, Float *x, CInt incx, Float *y, CInt incy); --void cblas_zswap( CInt n, Double *x, CInt incx, Double *y, CInt incy); --void cblas_srot( CInt N, Float *X, CInt incX, Float *Y, CInt incY, Float c, Float s); --void cblas_drot( CInt N, Double *X, CInt incX, Double *Y, CInt incY, Double c, Double s); --void cblas_srotg(Float *a, Float *b, Float *c, Float *s); --void cblas_drotg(Double *a, Double *b, Double *c, Double *s); --void cblas_srotm( CInt N, Float *X, CInt incX, Float *Y, CInt incY, Float *P); --void cblas_drotm( CInt N, Double *X, CInt incX, Double *Y, CInt incY, Double *P); --void cblas_srotmg(Float *d1, Float *d2, Float *b1, Float b2, Float *P); --void cblas_drotmg(Double *d1, Double *d2, Double *b1, Double b2, Double *P); --void cblas_sscal( CInt N, Float alpha, Float *X, CInt incX); --void cblas_dscal( CInt N, Double alpha, Double *X, CInt incX); --void cblas_cscal( CInt N, Float *alpha, Float *X, CInt incX); --void cblas_zscal( CInt N, Double *alpha, Double *X, CInt incX); --void cblas_csscal( CInt N, Float alpha, Float *X, CInt incX); --void cblas_zdscal( CInt N, Double alpha, Double *X, CInt incX); --------------------- ----- BLAS LEVEL 2 --------------------- {- matrix vector product for general matrices perform one of the matrix-vector operations y := alpha*A*x + beta*y, or y := alpha*A'*x + beta*y, -} --void cblas_sgemv( enum CBLAS_ORDER order, enum CBLAS_TRANSPOSE trans, CInt m, CInt n, -- Float alpha, Float *a, CInt lda, Float *x, CInt incx, Float beta, Float *y, CInt incy); --void cblas_dgemv( enum CBLAS_ORDER order, enum CBLAS_TRANSPOSE trans, CInt m, CInt n, -- Double alpha, Double *a, CInt lda, Double *x, CInt incx, Double beta, Double *y, CInt incy); --void cblas_cgemv( enum CBLAS_ORDER order, enum CBLAS_TRANSPOSE trans, CInt m, CInt n, -- Float *alpha, Float *a, CInt lda, Float *x, CInt incx, Float *beta, Float *y, CInt incy); --void cblas_zgemv( enum CBLAS_ORDER order, enum CBLAS_TRANSPOSE trans, CInt m, CInt n, -- Double *alpha, Double *a, CInt lda, Double *x, CInt incx, Double *beta, Double *y, CInt incy); -- perform the rank 1 operation A := alpha*x*y' + A, --void cblas_sger ( enum CBLAS_ORDER order, CInt M, CInt N, Float alpha, Float *X, CInt incX, Float *Y, CInt incY, Float *A, CInt lda); --void cblas_dger ( enum CBLAS_ORDER order, CInt M, CInt N, Double alpha, Double *X, CInt incX, Double *Y, CInt incY, Double *A, CInt lda); --void cblas_cgeru( enum CBLAS_ORDER order, CInt M, CInt N, Float *alpha, Float *X, CInt incX, Float *Y, CInt incY, Float *A, CInt lda); --void cblas_cgerc( enum CBLAS_ORDER order, CInt M, CInt N, Float *alpha, Float *X, CInt incX, Float *Y, CInt incY, Float *A, CInt lda); --void cblas_zgeru( enum CBLAS_ORDER order, CInt M, CInt N, Double *alpha, Double *X, CInt incX, Double *Y, CInt incY, Double *A, CInt lda); --void cblas_zgerc( enum CBLAS_ORDER order, CInt M, CInt N, Double *alpha, Double *X, CInt incX, Double *Y, CInt incY, Double *A, CInt lda); --STRSV - solve one of the systems of equations A*x = b, or A'*x = b, where A is a (non)unit upper(/lower) triangular matrix --void cblas_strsv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, enum CBLAS_DIAG Diag, CInt N, Float *A, CInt lda, Float *X, CInt incX); --void cblas_dtrsv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, enum CBLAS_DIAG Diag, CInt N, Double *A, CInt lda, Double *X, CInt incX); --void cblas_ctrsv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, enum CBLAS_DIAG Diag, CInt N, Float *A, CInt lda, Float *X, CInt incX); --void cblas_ztrsv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, enum CBLAS_DIAG Diag, CInt N, Double *A, CInt lda, Double *X, CInt incX); --void cblas_strmv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, enum CBLAS_DIAG Diag, CInt N, Float *A, CInt lda, Float *X, CInt incX); --void cblas_dtrmv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, enum CBLAS_DIAG Diag, CInt N, Double *A, CInt lda, Double *X, CInt incX); --void cblas_ctrmv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, enum CBLAS_DIAG Diag, CInt N, Float *A, CInt lda, Float *X, CInt incX); --void cblas_ztrmv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, enum CBLAS_DIAG Diag, CInt N, Double *A, CInt lda, Double *X, CInt incX); --void cblas_ssyr( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, CInt N, Float alpha, Float *X, CInt incX, Float *A, CInt lda); --void cblas_dsyr( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, CInt N, Double alpha, Double *X, CInt incX, Double *A, CInt lda); --void cblas_cher( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, CInt N, Float alpha, Float *X, CInt incX, Float *A, CInt lda); --void cblas_zher( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, CInt N, Double alpha, Double *X, CInt incX, Double *A, CInt lda); --void cblas_ssyr2( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, CInt N, Float alpha, Float *X, -- CInt incX, Float *Y, CInt incY, Float *A, CInt lda); --void cblas_dsyr2( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, CInt N, Double alpha, Double *X, -- CInt incX, Double *Y, CInt incY, Double *A, CInt lda); --void cblas_cher2( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, CInt N, Float *alpha, Float *X, CInt incX, -- Float *Y, CInt incY, Float *A, CInt lda); --void cblas_zher2( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, CInt N, Double *alpha, Double *X, CInt incX, -- Double *Y, CInt incY, Double *A, CInt lda); --void cblas_sgbmv( enum CBLAS_ORDER order, enum CBLAS_TRANSPOSE TransA, CInt M, CInt N, -- CInt KL, CInt KU, Float alpha, Float *A, CInt lda, Float *X, CInt incX, Float beta, Float *Y, CInt incY); --void cblas_dgbmv( enum CBLAS_ORDER order, enum CBLAS_TRANSPOSE TransA, CInt M, CInt N, -- CInt KL, CInt KU, Double alpha, Double *A, CInt lda, Double *X, CInt incX, Double beta, Double *Y, CInt incY); --void cblas_cgbmv( enum CBLAS_ORDER order, enum CBLAS_TRANSPOSE TransA, CInt M, CInt N, -- CInt KL, CInt KU, Float *alpha, Float *A, CInt lda, Float *X, CInt incX, Float *beta, Float *Y, CInt incY); --void cblas_zgbmv( enum CBLAS_ORDER order, enum CBLAS_TRANSPOSE TransA, CInt M, CInt N, -- CInt KL, CInt KU, Double *alpha, Double *A, CInt lda, Double *X, CInt incX, Double *beta, Double *Y, CInt incY); --void cblas_ssbmv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, CInt N, CInt K, Float alpha, Float *A, -- CInt lda, Float *X, CInt incX, Float beta, Float *Y, CInt incY); --void cblas_dsbmv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, CInt N, CInt K, Double alpha, Double *A, -- CInt lda, Double *X, CInt incX, Double beta, Double *Y, CInt incY); --void cblas_stbmv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, enum CBLAS_DIAG Diag, -- CInt N, CInt K, Float *A, CInt lda, Float *X, CInt incX); --void cblas_dtbmv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, enum CBLAS_DIAG Diag, -- CInt N, CInt K, Double *A, CInt lda, Double *X, CInt incX); --void cblas_ctbmv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, enum CBLAS_DIAG Diag, -- CInt N, CInt K, Float *A, CInt lda, Float *X, CInt incX); --void cblas_ztbmv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, enum CBLAS_DIAG Diag, -- CInt N, CInt K, Double *A, CInt lda, Double *X, CInt incX); ---------------- --- | solves Ax=v where A is k+1 banded triangular matrix, and x and ---------------- --void cblas_stbsv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, enum CBLAS_DIAG Diag, -- CInt N, CInt K, Float *A, CInt lda, Float *X, CInt incX); --void cblas_dtbsv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, enum CBLAS_DIAG Diag, -- CInt N, CInt K, Double *A, CInt lda, Double *X, CInt incX); --void cblas_ctbsv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, enum CBLAS_DIAG Diag, -- CInt N, CInt K, Float *A, CInt lda, Float *X, CInt incX); --void cblas_ztbsv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, enum CBLAS_DIAG Diag, -- CInt N, CInt K, Double *A, CInt lda, Double *X, CInt incX); ------------------------------------------------------------------------- -- | matrix vector product Av, writes result into v, where A is a packed triangular nxn matrix ------------------------------------------------------------------------- --void cblas_stpmv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, enum CBLAS_DIAG Diag, -- CInt N, Float *Ap, Float *X, CInt incX); --void cblas_dtpmv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, enum CBLAS_DIAG Diag, -- CInt N, Double *Ap, Double *X, CInt incX); --void cblas_ctpmv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, enum CBLAS_DIAG Diag, -- CInt N, Float *Ap, Float *X, CInt incX); --void cblas_ztpmv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, enum CBLAS_DIAG Diag, -- CInt N, Double *Ap, Double *X, CInt incX); -------------------------------------------------- --- | solve Ax=v where A is a nxn packed triangular matrix, v vector input, writes the solution into x. -------------------------------------------------- --void cblas_stpsv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, enum CBLAS_DIAG Diag, -- CInt N, Float *Ap, Float *X, CInt incX); --void cblas_dtpsv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, enum CBLAS_DIAG Diag, -- CInt N, Double *Ap, Double *X, CInt incX); --void cblas_ctpsv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, enum CBLAS_DIAG Diag, -- CInt N, Float *Ap, Float *X, CInt incX); --void cblas_ztpsv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, enum CBLAS_DIAG Diag, -- CInt N, Double *Ap, Double *X, CInt incX); ---------------------------------- ---- | (unpacked) symmetric matrix vector product x:=Av, writes result x into v --------------------------------- type SymvFunFFI el = CBLAS_ORDERT -> CBLAS_UPLOT -> CInt -> el -> Ptr el -> CInt -> Ptr el -> CInt -> el -> Ptr el -> CInt -> IO () --void cblas_ssymv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, CInt N, Float alpha, Float *A, -- CInt lda, Float *X, CInt incX, Float beta, Float *Y, CInt incY); --void cblas_dsymv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, CInt N, Double alpha, Double *A, -- CInt lda, Double *X, CInt incX, Double beta, Double *Y, CInt incY); -------------------------------- ---- | hermitian matrix vector product x:=Av, writes result x into v -------------------------------- --void cblas_chemv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, CInt N, Float *alpha, Float *A, -- CInt lda, Float *X, CInt incX, Float *beta, Float *Y, CInt incY); --void cblas_zhemv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, CInt N, Double *alpha, Double *A, -- CInt lda, Double *X, CInt incX, Double *beta, Double *Y, CInt incY); --------------- --- | packed symmetric matrix * vector product y:= alpha * Av + beta * y --------------- --void cblas_sspmv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, CInt N, Float alpha, Float *Ap, -- Float *X, CInt incX, Float beta, Float *Y, CInt incY); --void cblas_dspmv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, CInt N, Double alpha, Double *Ap, -- Double *X, CInt incX, Double beta, Double *Y, CInt incY); --void cblas_sspr( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, CInt N, Float alpha, Float *X, CInt incX, Float *Ap); --void cblas_dspr( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, CInt N, Double alpha, Double *X, CInt incX, Double *Ap); --void cblas_chpr( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, CInt N, Float alpha, Float *X, CInt incX, Float *A); --void cblas_zhpr( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, CInt N, Double alpha, Double *X, CInt incX, Double *A); --void cblas_sspr2( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, CInt N, Float alpha, Float *X, CInt incX, Float *Y, CInt incY, Float *A); --void cblas_dspr2( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, CInt N, Double alpha, Double *X, CInt incX, Double *Y, CInt incY, Double *A); --void cblas_chpr2( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, CInt N, Float *alpha, Float *X, CInt incX, Float *Y, CInt incY, Float *Ap); --void cblas_zhpr2( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, CInt N, Double *alpha, Double *X, CInt incX, Double *Y, CInt incY, Double *Ap); --void cblas_chbmv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, CInt N, CInt K, -- Float *alpha, Float *A, CInt lda, Float *X, CInt incX, Float *beta, Float *Y, CInt incY); --void cblas_zhbmv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, CInt N, CInt K, -- Double *alpha, Double *A, CInt lda, Double *X, CInt incX, Double *beta, Double *Y, CInt incY); --void cblas_chpmv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, CInt N, -- Float *alpha, Float *Ap, Float *X, CInt incX, Float *beta, Float *Y, CInt incY); --void cblas_zhpmv( enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, CInt N, -- Double *alpha, Double *Ap, Double *X, CInt incX, Double *beta, Double *Y, CInt incY); {- -} -------------------------------------------------------------------------------- -------------------------------------------------------------------------------- ------------------------------ | BLAS LEVEL 3 ROUTINES -------------------------------------------------------------------------------- ----------------------- | Level 3 ops are faster than Levels 1 or 2 -------------------------------------------------------------------------------- --void cblas_sgemm(const enum CBLAS_ORDER Order, const enum CBLAS_TRANSPOSE TransA, const enum CBLAS_TRANSPOSE TransB, const blasint M, const blasint N, const blasint K, -- const float alpha, const float *A, const blasint lda, const float *B, const blasint ldb, const float beta, float *C, const blasint ldc); --void cblas_dgemm(const enum CBLAS_ORDER Order, const enum CBLAS_TRANSPOSE TransA, const enum CBLAS_TRANSPOSE TransB, const blasint M, const blasint N, const blasint K, -- const double alpha, const double *A, const blasint lda, const double *B, const blasint ldb, const double beta, double *C, const blasint ldc); --void cblas_cgemm(const enum CBLAS_ORDER Order, const enum CBLAS_TRANSPOSE TransA, const enum CBLAS_TRANSPOSE TransB, const blasint M, const blasint N, const blasint K, -- const float *alpha, const float *A, const blasint lda, const float *B, const blasint ldb, const float *beta, float *C, const blasint ldc); --void cblas_zgemm(const enum CBLAS_ORDER Order, const enum CBLAS_TRANSPOSE TransA, const enum CBLAS_TRANSPOSE TransB, const blasint M, const blasint N, const blasint K, -- const double *alpha, const double *A, const blasint lda, const double *B, const blasint ldb, const double *beta, double *C, const blasint ldc); -- | Matrix mult for general dense matrices type GemmFunFFI scale el = CBLAS_ORDERT -> CBLAS_TRANSPOSET -> CBLAS_TRANSPOSET-> CInt -> CInt -> CInt -> {- scal A * B -} scale -> {- Matrix A-} Ptr el -> CInt -> {- B -} Ptr el -> CInt-> scale -> {- C -} Ptr el -> CInt -> IO () {- C := alpha*op( A )*op( B ) + beta*C , -} -- matrix mult! foreign import ccall unsafe "cblas_sgemm" cblas_sgemm_unsafe :: GemmFunFFI Float Float foreign import ccall unsafe "cblas_dgemm" cblas_dgemm_unsafe :: GemmFunFFI Double Double foreign import ccall unsafe "cblas_cgemm" cblas_cgemm_unsafe :: GemmFunFFI (Ptr(Complex Float)) (Complex Float) foreign import ccall unsafe "cblas_zgemm" cblas_zgemm_unsafe :: GemmFunFFI (Ptr (Complex Double)) (Complex Double) -- safe ffi variant for large inputs foreign import ccall "cblas_sgemm" cblas_sgemm_safe :: GemmFunFFI Float Float foreign import ccall "cblas_dgemm" cblas_dgemm_safe :: GemmFunFFI Double Double foreign import ccall "cblas_cgemm" cblas_cgemm_safe :: GemmFunFFI (Ptr(Complex Float)) (Complex Float) foreign import ccall "cblas_zgemm" cblas_zgemm_safe :: GemmFunFFI (Ptr (Complex Double)) (Complex Double) ----------------------------------------- ----- | Matrix mult for Symmetric Matrices ----------------------------------------- --void cblas_ssymm(const enum CBLAS_ORDER Order, const enum CBLAS_SIDE Side, const enum CBLAS_UPLO Uplo, const blasint M, const blasint N, -- const float alpha, const float *A, const blasint lda, const float *B, const blasint ldb, const float beta, float *C, const blasint ldc); --void cblas_dsymm(const enum CBLAS_ORDER Order, const enum CBLAS_SIDE Side, const enum CBLAS_UPLO Uplo, const blasint M, const blasint N, -- const double alpha, const double *A, const blasint lda, const double *B, const blasint ldb, const double beta, double *C, const blasint ldc); --void cblas_csymm(const enum CBLAS_ORDER Order, const enum CBLAS_SIDE Side, const enum CBLAS_UPLO Uplo, const blasint M, const blasint N, -- const float *alpha, const float *A, const blasint lda, const float *B, const blasint ldb, const float *beta, float *C, const blasint ldc); --void cblas_zsymm(const enum CBLAS_ORDER Order, const enum CBLAS_SIDE Side, const enum CBLAS_UPLO Uplo, const blasint M, const blasint N, -- const double *alpha, const double *A, const blasint lda, const double *B, const blasint ldb, const double *beta, double *C, const blasint ldc); type SymmFunFFI scale el = CBLAS_ORDERT -> CBLAS_SIDET -> CBLAS_UPLOT -> CInt->CInt -> scale -> Ptr el -> CInt -> Ptr el -> CInt -> scale ->Ptr el -> CInt -> IO () foreign import ccall unsafe "cblas_ssymm" cblas_ssymm_unsafe :: SymmFunFFI Float Float foreign import ccall unsafe "cblas_dsymm" cblas_dsymm_unsafe :: SymmFunFFI Double Double foreign import ccall unsafe "cblas_csymm" cblas_csymm_unsafe :: SymmFunFFI (Ptr (Complex Float )) (Complex Float) foreign import ccall unsafe "cblas_zsymm" cblas_zsymm_unsafe :: SymmFunFFI (Ptr (Complex Double)) (Complex Double) -- safe ffi variant, foreign import ccall "cblas_ssymm" cblas_ssymm_safe :: SymmFunFFI Float Float foreign import ccall "cblas_dsymm" cblas_dsymm_safe :: SymmFunFFI Double Double foreign import ccall "cblas_csymm" cblas_csymm_safe :: SymmFunFFI (Ptr (Complex Float )) (Complex Float) foreign import ccall "cblas_zsymm" cblas_zsymm_safe :: SymmFunFFI (Ptr (Complex Double)) (Complex Double) ----------------------------------- --- | symmetric rank k matrix update, C := alpha*A*A' + beta*C --- or C = alpha*A'*A + beta*C ------------------------------------ --void cblas_ssyrk(const enum CBLAS_ORDER Order, const enum CBLAS_UPLO Uplo, const enum CBLAS_TRANSPOSE Trans, -- const blasint N, const blasint K, const float alpha, const float *A, const blasint lda, const float beta, float *C, const blasint ldc); --void cblas_dsyrk(const enum CBLAS_ORDER Order, const enum CBLAS_UPLO Uplo, const enum CBLAS_TRANSPOSE Trans, -- const blasint N, const blasint K, const double alpha, const double *A, const blasint lda, const double beta, double *C, const blasint ldc); --void cblas_csyrk(const enum CBLAS_ORDER Order, const enum CBLAS_UPLO Uplo, const enum CBLAS_TRANSPOSE Trans, -- const blasint N, const blasint K, const float *alpha, const float *A, const blasint lda, const float *beta, float *C, const blasint ldc); --void cblas_zsyrk(const enum CBLAS_ORDER Order, const enum CBLAS_UPLO Uplo, const enum CBLAS_TRANSPOSE Trans, -- const blasint N, const blasint K, const double *alpha, const double *A, const blasint lda, const double *beta, double *C, const blasint ldc); type SyrkFunFFI scale el = CBLAS_ORDERT -> CBLAS_UPLOT -> CBLAS_TRANSPOSET -> CInt->CInt -> scale -> Ptr el -> CInt -> Ptr el -> CInt ->scale -> Ptr el -> CInt -> IO () foreign import ccall unsafe "cblas_ssyrk" cblas_ssyrk_unsafe :: SyrkFunFFI Float Float foreign import ccall unsafe "cblas_dsyrk" cblas_dsyrk_unsafe :: SyrkFunFFI Double Double foreign import ccall unsafe "cblas_csyrk" cblas_csyrk_unsafe :: SyrkFunFFI (Ptr(Complex Float)) (Complex Float) foreign import ccall unsafe "cblas_zsyrk" cblas_zsyrk_unsafe :: SyrkFunFFI (Ptr(Complex Double)) (Complex Double) ---------------------- ----- | Symmetric Rank 2k matrix update, C= alpha* A*B' + alpha* B*A' + beta * C ----- or C= alpha* A'*B + alpha* B'*A + beta * C ------------------- --void cblas_ssyr2k(const enum CBLAS_ORDER Order, const enum CBLAS_UPLO Uplo, const enum CBLAS_TRANSPOSE Trans, -- const blasint N, const blasint K, const float alpha, const float *A, const blasint lda, const float *B, const blasint ldb, const float beta, float *C, const blasint ldc); --void cblas_dsyr2k(const enum CBLAS_ORDER Order, const enum CBLAS_UPLO Uplo, const enum CBLAS_TRANSPOSE Trans, -- const blasint N, const blasint K, const double alpha, const double *A, const blasint lda, const double *B, const blasint ldb, const double beta, double *C, const blasint ldc); --void cblas_csyr2k(const enum CBLAS_ORDER Order, const enum CBLAS_UPLO Uplo, const enum CBLAS_TRANSPOSE Trans, -- const blasint N, const blasint K, const float *alpha, const float *A, const blasint lda, const float *B, const blasint ldb, const float *beta, float *C, const blasint ldc); --void cblas_zsyr2k(const enum CBLAS_ORDER Order, const enum CBLAS_UPLO Uplo, const enum CBLAS_TRANSPOSE Trans, --const blasint N, const blasint K, const double *alpha, const double *A, const blasint lda, const double *B, const blasint ldb, const double *beta, double *C, const blasint ldc); type Syr2kFunFFI scale el = CBLAS_ORDERT -> CBLAS_UPLOT -> CBLAS_TRANSPOSET -> CInt->CInt -> scale -> Ptr el -> CInt -> Ptr el -> CInt -> scale ->Ptr el -> CInt -> IO () foreign import ccall unsafe "cblas_ssyr2k" cblas_ssyr2k_unsafe :: Syr2kFunFFI Float Float foreign import ccall unsafe "cblas_dsyr2k" cblas_dsyr2k_unsafe :: Syr2kFunFFI Double Double foreign import ccall unsafe "cblas_csyr2k" cblas_csyr2k_unsafe :: Syr2kFunFFI (Ptr (Complex Float)) Float foreign import ccall unsafe "cblas_zsyr2k" cblas_zsyr2k_unsafe :: Syr2kFunFFI (Ptr (Complex Double)) Double ------------------------------- -------- | matrix matrix product for triangular matrices ------------------------------ type TrmmFunFFI scale el = CBLAS_ORDERT -> CBLAS_SIDET -> CBLAS_UPLOT -> CBLAS_TRANSPOSET -> CBLAS_DIAGT -> CInt->CInt -> scale -> Ptr el -> CInt -> Ptr el -> CInt -> Ptr el -> CInt -> IO () foreign import ccall unsafe "cblas_strmm" cblas_strmm_unsafe :: TrmmFunFFI Float Float foreign import ccall unsafe "cblas_dtrmm" cblas_dtrmm_unsafe :: TrmmFunFFI Double Double foreign import ccall unsafe "cblas_ctrmm" cblas_ctrmm_unsafe :: TrmmFunFFI (Ptr (Complex Float )) (Complex Float) foreign import ccall unsafe "cblas_ztrmm" cblas_ztrmm_unsafe :: TrmmFunFFI (Ptr (Complex Double )) (Complex Double) --void cblas_strmm( enum CBLAS_ORDER Order, enum CBLAS_SIDE Side, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, -- enum CBLAS_DIAG Diag, CInt M, CInt N, Float alpha, Float *A, CInt lda, Float *B, CInt ldb); --void cblas_dtrmm( enum CBLAS_ORDER Order, enum CBLAS_SIDE Side, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, -- enum CBLAS_DIAG Diag, CInt M, CInt N, Double alpha, Double *A, CInt lda, Double *B, CInt ldb); --void cblas_ctrmm( enum CBLAS_ORDER Order, enum CBLAS_SIDE Side, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, -- enum CBLAS_DIAG Diag, CInt M, CInt N, Float *alpha, Float *A, CInt lda, Float *B, CInt ldb); --void cblas_ztrmm( enum CBLAS_ORDER Order, enum CBLAS_SIDE Side, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, -- enum CBLAS_DIAG Diag, CInt M, CInt N, Double *alpha, Double *A, CInt lda, Double *B, CInt ldb); ------------------------ -- | triangular solvers ----------------------- type TrsmFunFFI scale el = CBLAS_ORDERT -> CBLAS_SIDET -> CBLAS_UPLOT -> CBLAS_TRANSPOSET -> CBLAS_DIAGT -> CInt->CInt -> scale -> Ptr el -> CInt -> Ptr el -> CInt -> Ptr el -> CInt -> IO () foreign import ccall unsafe "cblas_strsm" cblas_strsm_unsafe :: TrmmFunFFI Float Float foreign import ccall unsafe "cblas_dtrsm" cblas_dtrsm_unsafe :: TrmmFunFFI Double Double foreign import ccall unsafe "cblas_ctrsm" cblas_ctrsm_unsafe :: TrmmFunFFI (Ptr (Complex Float )) (Complex Float) foreign import ccall unsafe "cblas_ztrsm" cblas_ztrsm_unsafe :: TrmmFunFFI (Ptr (Complex Double )) (Complex Double) --void cblas_strsm( enum CBLAS_ORDER Order, enum CBLAS_SIDE Side, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, -- enum CBLAS_DIAG Diag, CInt M, CInt N, Float alpha, Float *A, CInt lda, Float *B, CInt ldb); --void cblas_dtrsm( enum CBLAS_ORDER Order, enum CBLAS_SIDE Side, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, -- enum CBLAS_DIAG Diag, CInt M, CInt N, Double alpha, Double *A, CInt lda, Double *B, CInt ldb); --void cblas_ctrsm( enum CBLAS_ORDER Order, enum CBLAS_SIDE Side, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, -- enum CBLAS_DIAG Diag, CInt M, CInt N, Float *alpha, Float *A, CInt lda, Float *B, CInt ldb); --void cblas_ztrsm( enum CBLAS_ORDER Order, enum CBLAS_SIDE Side, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE TransA, -- enum CBLAS_DIAG Diag, CInt M, CInt N, Double *alpha, Double *A, CInt lda, Double *B, CInt ldb); ------------------------- -- | hermitian matrix mult ------------------------ type HemmFunFFI el = CBLAS_ORDERT -> CBLAS_SIDET -> CBLAS_UPLOT -> CInt->CInt -> Ptr el -> Ptr el -> CInt -> Ptr el -> CInt -> Ptr el -> CInt -> IO () foreign import ccall unsafe "cblas_chemm" cblas_chemm_unsafe :: HemmFunFFI (Complex Float) foreign import ccall unsafe "cblas_zhemm" cblas_zhemm_unsafe :: HemmFunFFI (Complex Double) --void cblas_chemm( enum CBLAS_ORDER Order, enum CBLAS_SIDE Side, enum CBLAS_UPLO Uplo, CInt M, CInt N, -- Float *alpha, Float *A, CInt lda, Float *B, CInt ldb, Float *beta, Float *C, CInt ldc); --void cblas_zhemm( enum CBLAS_ORDER Order, enum CBLAS_SIDE Side, enum CBLAS_UPLO Uplo, CInt M, CInt N, -- Double *alpha, Double *A, CInt lda, Double *B, CInt ldb, Double *beta, Double *C, CInt ldc); type HerkFun scale el = CBLAS_ORDERT -> CBLAS_SIDET-> CBLAS_TRANSPOSET -> CInt->CInt -> scale -> Ptr el -> CInt -> Ptr el -> CInt ->scale ->Ptr el -> CInt -> IO () foreign import ccall unsafe "cblas_cherk" cblas_cherk_unsafe :: HerkFun Float (Complex Float) foreign import ccall unsafe "cblas_zherk" cblas_zherk_unsafe :: HerkFun Double (Complex Double) --void cblas_cherk( enum CBLAS_ORDER Order, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE Trans, CInt N, CInt K, -- Float alpha, Float *A, CInt lda, Float beta, Float *C, CInt ldc); --void cblas_zherk( enum CBLAS_ORDER Order, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE Trans, CInt N, CInt K, -- Double alpha, Double *A, CInt lda, Double beta, Double *C, CInt ldc); type Her2kFunFFI scale el = CBLAS_ORDERT -> CBLAS_SIDET -> CBLAS_TRANSPOSET -> CInt->CInt -> Ptr el -> Ptr el -> CInt -> Ptr el -> CInt ->scale ->Ptr el -> CInt -> IO () foreign import ccall unsafe "cblas_cher2k" cblas_cher2k_unsafe :: Her2kFunFFI Float (Complex Float) foreign import ccall unsafe "cblas_zher2k" cblas_zher2k_unsafe :: Her2kFunFFI Double (Complex Double) --void cblas_cher2k( enum CBLAS_ORDER Order, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE Trans, CInt N, CInt K, -- Float *alpha, Float *A, CInt lda, Float *B, CInt ldb, Float beta, Float *C, CInt ldc); --void cblas_zher2k( enum CBLAS_ORDER Order, enum CBLAS_UPLO Uplo, enum CBLAS_TRANSPOSE Trans, CInt N, CInt K, -- Double *alpha, Double *A, CInt lda, Double *B, CInt ldb, Double beta, Double *C, CInt ldc); ----void cblas_xerbla(CInt p, char *rout, char *form, ...);