Documentation

flopsThreshold :: Int Source

gemmComplexity :: Num a => a -> a -> a -> a Source

isBadGemm :: (Ord a, Num a) => Transpose -> Transpose -> a -> a -> a -> a -> a -> a -> Bool Source

coordSwapper :: Transpose -> (a, a) -> (a, a) Source

encodeNiceOrder :: SOrientation x -> CBLAS_ORDERT Source

encodeFFITranpose :: Transpose -> CBLAS_TRANSPOSET Source

encodeNiceTranpose :: Transpose -> BLAS_Transpose Source

type GemmFun el orient s m = Transpose -> Transpose -> el -> el -> MutDenseMatrix s orient el -> MutDenseMatrix s orient el -> MutDenseMatrix s orient el -> m () Source

gemmAbstraction :: (Storable el, PrimMonad m) => String -> GemmFunFFI scale el -> GemmFunFFI scale el -> (el -> (scale -> m ()) -> m ()) -> forall orient. GemmFun el orient (PrimState m) m Source

sgemm :: PrimMonad m => Transpose -> Transpose -> Float -> Float -> MutDenseMatrix (PrimState m) orient Float -> MutDenseMatrix (PrimState m) orient Float -> MutDenseMatrix (PrimState m) orient Float -> m () Source

dgemm :: PrimMonad m => Transpose -> Transpose -> Double -> Double -> MutDenseMatrix (PrimState m) orient Double -> MutDenseMatrix (PrimState m) orient Double -> MutDenseMatrix (PrimState m) orient Double -> m () Source

cgemm :: PrimMonad m => Transpose -> Transpose -> Complex Float -> Complex Float -> MutDenseMatrix (PrimState m) orient (Complex Float) -> MutDenseMatrix (PrimState m) orient (Complex Float) -> MutDenseMatrix (PrimState m) orient (Complex Float) -> m () Source

zgemm :: PrimMonad m => Transpose -> Transpose -> Complex Double -> Complex Double -> MutDenseMatrix (PrimState m) orient (Complex Double) -> MutDenseMatrix (PrimState m) orient (Complex Double) -> MutDenseMatrix (PrimState m) orient (Complex Double) -> m () Source