Safe Haskell | None |
---|
Documentation
gemmComplexity :: Integral a => a -> a -> a -> Int64Source
gemvComplexity :: Integral a => a -> a -> Int64Source
coordSwapper :: Transpose -> (a, a) -> (a, a)Source
type GemmFun el orient s m = Transpose -> Transpose -> el -> el -> MDenseMatrix s orient el -> MDenseMatrix s orient el -> MDenseMatrix s orient el -> m ()Source
type GemvFun el orient s m = Transpose -> el -> el -> MDenseMatrix s orient el -> MDenseVector s Direct el -> MDenseVector s Direct el -> m ()Source
type TrsvFun el orient s m = MatUpLo -> Transpose -> MatDiag -> MDenseMatrix s orient el -> MDenseVector s Direct 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 -> MDenseMatrix (PrimState m) orient Float -> MDenseMatrix (PrimState m) orient Float -> MDenseMatrix (PrimState m) orient Float -> m ()Source
dgemm :: PrimMonad m => Transpose -> Transpose -> Double -> Double -> MDenseMatrix (PrimState m) orient Double -> MDenseMatrix (PrimState m) orient Double -> MDenseMatrix (PrimState m) orient Double -> m ()Source
cgemm :: PrimMonad m => Transpose -> Transpose -> Complex Float -> Complex Float -> MDenseMatrix (PrimState m) orient (Complex Float) -> MDenseMatrix (PrimState m) orient (Complex Float) -> MDenseMatrix (PrimState m) orient (Complex Float) -> m ()Source
zgemm :: PrimMonad m => Transpose -> Transpose -> Complex Double -> Complex Double -> MDenseMatrix (PrimState m) orient (Complex Double) -> MDenseMatrix (PrimState m) orient (Complex Double) -> MDenseMatrix (PrimState m) orient (Complex Double) -> m ()Source
gemvAbstraction :: (Storable el, PrimMonad m) => String -> GemvFunFFI scale el -> GemvFunFFI scale el -> (el -> (scale -> m ()) -> m ()) -> forall orient. GemvFun el orient (PrimState m) mSource
trsvAbstraction :: (Storable el, PrimMonad m) => String -> TrsvFunFFI el -> TrsvFunFFI el -> forall orient. TrsvFun el orient (PrimState m) mSource