Futhark.Pass.ExtractKernels.BlockedKernel

Documentation

blockedReduction :: (MonadFreshNames m, HasScope Kernels m) => Pattern Kernels -> SubExp -> Commutativity -> Lambda InKernel -> Lambda InKernel -> [(VName, SubExp)] -> [SubExp] -> [VName] -> m (Stms Kernels)

blockedReductionStream :: (MonadFreshNames m, HasScope Kernels m) => Pattern Kernels -> SubExp -> Commutativity -> Lambda InKernel -> Lambda InKernel -> [(VName, SubExp)] -> [SubExp] -> [VName] -> m (Stms Kernels)

blockedGenReduce

Arguments

:: (MonadFreshNames m, HasScope Kernels m)

=> SubExp

-> [(VName, SubExp)]

Segment indexes and sizes.

-> [KernelInput]

-> [GenReduceOp InKernel]

-> Lambda InKernel

-> [VName]

-> m ([VName], Stms Kernels)

blockedMap :: (MonadFreshNames m, HasScope Kernels m) => Pattern Kernels -> SubExp -> StreamOrd -> Lambda InKernel -> [SubExp] -> [VName] -> m (Stm Kernels, Stms Kernels)

blockedScan :: (MonadBinder m, Lore m ~ Kernels) => Pattern Kernels -> SubExp -> Scan InKernel -> Reduce InKernel -> Lambda InKernel -> SubExp -> [(VName, SubExp)] -> [KernelInput] -> [VName] -> m [VName]

The VNames returned are the names of variables bound to the carry-out of the last thread. You can ignore them if you don't need them.

segRed :: (MonadFreshNames m, HasScope Kernels m) => Pattern Kernels -> SubExp -> SubExp -> Commutativity -> Lambda InKernel -> Lambda InKernel -> [SubExp] -> [VName] -> [(VName, SubExp)] -> [KernelInput] -> m (Stms Kernels)

nonSegRed :: (MonadFreshNames m, HasScope Kernels m) => Pattern Kernels -> SubExp -> Commutativity -> Lambda InKernel -> Lambda InKernel -> [SubExp] -> [VName] -> m (Stms Kernels)

mapKernel :: (HasScope Kernels m, MonadFreshNames m) => SubExp -> SpaceStructure -> [KernelInput] -> [Type] -> KernelBody InKernel -> m (Stms Kernels, Kernel InKernel)

mapKernelFromBody :: (HasScope Kernels m, MonadFreshNames m) => SubExp -> SpaceStructure -> [KernelInput] -> [Type] -> Body InKernel -> m (Stms Kernels, Kernel InKernel)

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 :: (HasScope scope m, Monad m) => KernelInput -> m (Stm InKernel)

kerneliseLambda :: MonadFreshNames m => [SubExp] -> Lambda InKernel -> m (Lambda InKernel)

Given a chunked fold lambda that takes its initial accumulator value as parameters, bind those parameters to the neutral element instead.

newKernelSpace :: MonadFreshNames m => (SubExp, SubExp, SubExp) -> SpaceStructure -> m KernelSpace

chunkLambda :: (MonadFreshNames m, HasScope Kernels m) => Pattern Kernels -> [SubExp] -> Lambda InKernel -> m (Lambda InKernel)

Requires a fold lambda that includes accumulator parameters.

splitArrays :: (MonadBinder m, Lore m ~ InKernel) => VName -> [VName] -> SplitOrdering -> SubExp -> SubExp -> SubExp -> [VName] -> m ()

getSize :: (MonadBinder m, Op (Lore m) ~ Kernel innerlore) => String -> SizeClass -> m SubExp