Futhark.Pass.ExtractKernels.BlockedKernel

Documentation

type DistLore lore = (Bindable lore, HasSegOp lore, BinderOps lore, LetDec lore ~ Type, ExpDec lore ~ (), BodyDec lore ~ ())

Constraints pertinent to performing distribution/flattening.

type MkSegLevel lore m = [SubExp] -> String -> ThreadRecommendation -> BinderT lore m (SegOpLevel lore)

data ThreadRecommendation

Constructors

ManyThreads

NoRecommendation SegVirt

segRed :: (MonadFreshNames m, DistLore lore, HasScope lore m) => SegOpLevel lore -> Pattern lore -> SubExp -> [SegBinOp lore] -> Lambda lore -> [VName] -> [(VName, SubExp)] -> [KernelInput] -> m (Stms lore)

nonSegRed :: (MonadFreshNames m, DistLore lore, HasScope lore m) => SegOpLevel lore -> Pattern lore -> SubExp -> [SegBinOp lore] -> Lambda lore -> [VName] -> m (Stms lore)

segScan :: (MonadFreshNames m, DistLore lore, HasScope lore m) => SegOpLevel lore -> Pattern lore -> SubExp -> [SegBinOp lore] -> Lambda lore -> [VName] -> [(VName, SubExp)] -> [KernelInput] -> m (Stms lore)

segHist

Arguments

:: (DistLore lore, MonadFreshNames m, HasScope lore m)

=> SegOpLevel lore

-> Pattern lore

-> SubExp

-> [(VName, SubExp)]

Segment indexes and sizes.

-> [KernelInput]

-> [HistOp lore]

-> Lambda lore

-> [VName]

-> m (Stms lore)

segMap :: (MonadFreshNames m, DistLore lore, HasScope lore m) => SegOpLevel lore -> Pattern lore -> SubExp -> Lambda lore -> [VName] -> [(VName, SubExp)] -> [KernelInput] -> m (Stms lore)

mapKernel :: (DistLore lore, HasScope lore m, MonadFreshNames m) => MkSegLevel lore m -> [(VName, SubExp)] -> [KernelInput] -> [Type] -> KernelBody lore -> m (SegOp (SegOpLevel lore) lore, Stms lore)

data KernelInput

Constructors

KernelInput

Fields

• kernelInputName :: VName
• kernelInputType :: Type
• kernelInputArray :: VName
• kernelInputIndices :: [SubExp]

Instances

Show KernelInput

Defined in Futhark.Pass.ExtractKernels.BlockedKernel

Methods

showsPrec :: Int -> KernelInput -> ShowS

show :: KernelInput -> String

showList :: [KernelInput] -> ShowS

readKernelInput :: (DistLore (Lore m), MonadBinder m) => KernelInput -> m ()

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

dummyDim :: (MonadFreshNames m, MonadBinder m, DistLore (Lore m)) => Pattern (Lore m) -> m (Pattern (Lore m), [(VName, SubExp)], m ())