Documentation

flopsThreshold :: IntSource

gemmComplexity :: Num a => a -> a -> a -> aSource

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

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

encodeNiceOrder :: SOrientation x -> CBLAS_ORDERTSource

encodeFFITranpose :: Transpose -> CBLAS_TRANSPOSETSource

encodeNiceTranpose :: Transpose -> BLAS_TransposeSource

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) mSource

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