Futhark.Pass.ExtractKernels.BlockedKernel

Documentation

type DistRep rep = (Buildable rep, HasSegOp rep, BuilderOps rep, LetDec rep ~ Type, ExpDec rep ~ (), BodyDec rep ~ (), AliasableRep rep)

Constraints pertinent to performing distribution/flattening.

type MkSegLevel rep m = [SubExp] -> String -> ThreadRecommendation -> BuilderT rep m (SegOpLevel rep)

data ThreadRecommendation

Constructors

ManyThreads

NoRecommendation SegVirt

segRed :: (MonadFreshNames m, DistRep rep, HasScope rep m) => SegOpLevel rep -> Pat (LetDec rep) -> Certs -> SubExp -> [SegBinOp rep] -> Lambda rep -> [VName] -> [(VName, SubExp)] -> [KernelInput] -> m (Stms rep)

nonSegRed :: (MonadFreshNames m, DistRep rep, HasScope rep m) => SegOpLevel rep -> Pat Type -> SubExp -> [SegBinOp rep] -> Lambda rep -> [VName] -> m (Stms rep)

segScan :: (MonadFreshNames m, DistRep rep, HasScope rep m) => SegOpLevel rep -> Pat (LetDec rep) -> Certs -> SubExp -> [SegBinOp rep] -> Lambda rep -> [VName] -> [(VName, SubExp)] -> [KernelInput] -> m (Stms rep)

segHist

Arguments

:: (DistRep rep, MonadFreshNames m, HasScope rep m)

=> SegOpLevel rep

-> Pat Type

-> SubExp

-> [(VName, SubExp)]

Segment indexes and sizes.

-> [KernelInput]

-> [HistOp rep]

-> Lambda rep

-> [VName]

-> m (Stms rep)

segMap :: (MonadFreshNames m, DistRep rep, HasScope rep m) => SegOpLevel rep -> Pat (LetDec rep) -> SubExp -> Lambda rep -> [VName] -> [(VName, SubExp)] -> [KernelInput] -> m (Stms rep)

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

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 :: (DistRep (Rep m), MonadBuilder m) => KernelInput -> m ()

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

dummyDim :: MonadBuilder m => Pat Type -> m (Pat Type, [(VName, SubExp)], m ())