Safe Haskell	None
Language	Haskell2010

Futhark.Pass.ExtractKernels.BlockedKernel

Synopsis

type MkSegLevel m = [SubExp] -> String -> ThreadRecommendation -> BinderT Kernels m SegLevel
data ThreadRecommendation
- = ManyThreads
- | NoRecommendation SegVirt
segRed :: (MonadFreshNames m, HasScope Kernels m) => SegLevel -> Pattern Kernels -> SubExp -> [SegRedOp Kernels] -> Lambda Kernels -> [VName] -> [(VName, SubExp)] -> [KernelInput] -> m (Stms Kernels)
nonSegRed :: (MonadFreshNames m, HasScope Kernels m) => SegLevel -> Pattern Kernels -> SubExp -> [SegRedOp Kernels] -> Lambda Kernels -> [VName] -> m (Stms Kernels)
segScan :: (MonadFreshNames m, HasScope Kernels m) => SegLevel -> Pattern Kernels -> SubExp -> Lambda Kernels -> Lambda Kernels -> [SubExp] -> [VName] -> [(VName, SubExp)] -> [KernelInput] -> m (Stms Kernels)
segGenRed :: (MonadFreshNames m, HasScope Kernels m) => Pattern Kernels -> SubExp -> [(VName, SubExp)] -> [KernelInput] -> [GenReduceOp Kernels] -> Lambda Kernels -> [VName] -> m (Stms Kernels)
segMap :: (MonadFreshNames m, HasScope Kernels m) => SegLevel -> Pattern Kernels -> SubExp -> Lambda Kernels -> [VName] -> [(VName, SubExp)] -> [KernelInput] -> m (Stms Kernels)
streamRed :: (MonadFreshNames m, HasScope Kernels m) => Pattern Kernels -> SubExp -> Commutativity -> Lambda Kernels -> Lambda Kernels -> [SubExp] -> [VName] -> m (Stms Kernels)
streamMap :: (MonadFreshNames m, HasScope Kernels m) => [String] -> [PatElem Kernels] -> SubExp -> Commutativity -> Lambda Kernels -> [SubExp] -> [VName] -> m ((SubExp, [VName]), Stms Kernels)
mapKernel :: (HasScope Kernels m, MonadFreshNames m) => MkSegLevel m -> [(VName, SubExp)] -> [KernelInput] -> [Type] -> KernelBody Kernels -> m (SegOp Kernels, Stms Kernels)
data KernelInput = KernelInput {
- kernelInputName :: VName
- kernelInputType :: Type
- kernelInputArray :: VName
- kernelInputIndices :: [SubExp]
}
readKernelInput :: (MonadBinder m, Lore m ~ Kernels) => KernelInput -> m ()
soacsLambdaToKernels :: Lambda SOACS -> Lambda Kernels
soacsStmToKernels :: Stm SOACS -> Stm Kernels
scopeForKernels :: Scope SOACS -> Scope Kernels
scopeForSOACs :: Scope Kernels -> Scope SOACS
getSize :: (MonadBinder m, Op (Lore m) ~ HostOp (Lore m) inner) => String -> SizeClass -> m SubExp
segThread :: (MonadBinder m, Op (Lore m) ~ HostOp (Lore m) inner) => String -> m SegLevel
segThreadCapped :: MonadFreshNames m => MkSegLevel m
mkSegSpace :: MonadFreshNames m => [(VName, SubExp)] -> m SegSpace

Documentation

type MkSegLevel m = [SubExp] -> String -> ThreadRecommendation -> BinderT Kernels m SegLevel Source #

data ThreadRecommendation Source #

Constructors

ManyThreads
NoRecommendation SegVirt

segRed :: (MonadFreshNames m, HasScope Kernels m) => SegLevel -> Pattern Kernels -> SubExp -> [SegRedOp Kernels] -> Lambda Kernels -> [VName] -> [(VName, SubExp)] -> [KernelInput] -> m (Stms Kernels) Source #

nonSegRed :: (MonadFreshNames m, HasScope Kernels m) => SegLevel -> Pattern Kernels -> SubExp -> [SegRedOp Kernels] -> Lambda Kernels -> [VName] -> m (Stms Kernels) Source #

segScan :: (MonadFreshNames m, HasScope Kernels m) => SegLevel -> Pattern Kernels -> SubExp -> Lambda Kernels -> Lambda Kernels -> [SubExp] -> [VName] -> [(VName, SubExp)] -> [KernelInput] -> m (Stms Kernels) Source #

segGenRed Source #

Arguments

:: (MonadFreshNames m, HasScope Kernels m)
=> Pattern Kernels
-> SubExp
-> [(VName, SubExp)]	Segment indexes and sizes.
-> [KernelInput]
-> [GenReduceOp Kernels]
-> Lambda Kernels
-> [VName]
-> m (Stms Kernels)

segMap :: (MonadFreshNames m, HasScope Kernels m) => SegLevel -> Pattern Kernels -> SubExp -> Lambda Kernels -> [VName] -> [(VName, SubExp)] -> [KernelInput] -> m (Stms Kernels) Source #

streamRed :: (MonadFreshNames m, HasScope Kernels m) => Pattern Kernels -> SubExp -> Commutativity -> Lambda Kernels -> Lambda Kernels -> [SubExp] -> [VName] -> m (Stms Kernels) Source #

streamMap :: (MonadFreshNames m, HasScope Kernels m) => [String] -> [PatElem Kernels] -> SubExp -> Commutativity -> Lambda Kernels -> [SubExp] -> [VName] -> m ((SubExp, [VName]), Stms Kernels) Source #

mapKernel :: (HasScope Kernels m, MonadFreshNames m) => MkSegLevel m -> [(VName, SubExp)] -> [KernelInput] -> [Type] -> KernelBody Kernels -> m (SegOp Kernels, Stms Kernels) Source #

data KernelInput Source #

Constructors

KernelInput
Fields kernelInputName :: VName kernelInputType :: Type kernelInputArray :: VName kernelInputIndices :: [SubExp]

Instances

Show KernelInput Source #
Instance details Defined in Futhark.Pass.ExtractKernels.BlockedKernel Methods showsPrec :: Int -> KernelInput -> ShowS # show :: KernelInput -> String # showList :: [KernelInput] -> ShowS #

readKernelInput :: (MonadBinder m, Lore m ~ Kernels) => KernelInput -> m () Source #

soacsLambdaToKernels :: Lambda SOACS -> Lambda Kernels Source #

soacsStmToKernels :: Stm SOACS -> Stm Kernels Source #

scopeForKernels :: Scope SOACS -> Scope Kernels Source #

scopeForSOACs :: Scope Kernels -> Scope SOACS Source #

getSize :: (MonadBinder m, Op (Lore m) ~ HostOp (Lore m) inner) => String -> SizeClass -> m SubExp Source #

segThread :: (MonadBinder m, Op (Lore m) ~ HostOp (Lore m) inner) => String -> m SegLevel Source #

segThreadCapped :: MonadFreshNames m => MkSegLevel m Source #

Like segThread, but cap the thread count to the input size. This is more efficient for small kernels, e.g. summing a small array.

mkSegSpace :: MonadFreshNames m => [(VName, SubExp)] -> m SegSpace Source #