Futhark.Pass.ExtractKernels.BlockedKernel

Documentation

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

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

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

segGenRed

Arguments

:: (MonadFreshNames m, HasScope Kernels m)

=> Pattern Kernels

-> SubExp

-> [(VName, SubExp)]

Segment indexes and sizes.

-> [KernelInput]

-> [GenReduceOp InKernel]

-> Lambda InKernel

-> [VName]

-> m (Stms Kernels)

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

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

mapKernel :: (HasScope Kernels m, MonadFreshNames m) => SubExp -> SpaceStructure -> [KernelInput] -> [Type] -> KernelBody 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)

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

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