Safe Haskell	None
Language	Haskell2010

Futhark.Pass.ExtractKernels.DistributeNests

Documentation

type KernelsStms = Stms Kernels Source #

data MapLoop Source #

Constructors

MapLoop Pattern Certificates SubExp Lambda [VName]

mapLoopStm :: MapLoop -> Stm Source #

bodyContainsParallelism :: Body -> Bool Source #

lambdaContainsParallelism :: Lambda -> Bool Source #

determineReduceOp :: (MonadBinder m, Lore m ~ Kernels) => Lambda -> [SubExp] -> m (Lambda Kernels, [SubExp], Shape) Source #

incrementalFlattening :: Bool Source #

genReduceKernel :: (MonadFreshNames m, HasScope Kernels m) => Pattern -> [(VName, SubExp)] -> [KernelInput] -> Certificates -> SubExp -> [GenReduceOp SOACS] -> Lambda Kernels -> [VName] -> m KernelsStms Source #

data DistEnv m Source #

Constructors

DistEnv
Fields distNest :: Nestings distScope :: Scope Kernels distOnTopLevelStms :: Stms SOACS -> DistNestT m (Stms Kernels) distOnInnerMap :: MapLoop -> DistAcc -> DistNestT m DistAcc distSegLevel :: MkSegLevel m

Instances

Monad m => MonadReader (DistEnv m) (DistNestT m) Source #
Instance details Defined in Futhark.Pass.ExtractKernels.DistributeNests Methods ask :: DistNestT m (DistEnv m) # local :: (DistEnv m -> DistEnv m) -> DistNestT m a -> DistNestT m a # reader :: (DistEnv m -> a) -> DistNestT m a #

data DistAcc Source #

Constructors

DistAcc
Fields distTargets :: Targets distStms :: KernelsStms

runDistNestT :: MonadLogger m => DistEnv m -> DistNestT m DistAcc -> m (Stms Kernels) Source #

data DistNestT m a Source #

Instances

MonadTrans DistNestT Source #
Instance details Defined in Futhark.Pass.ExtractKernels.DistributeNests Methods lift :: Monad m => m a -> DistNestT m a #
Monad m => LocalScope Kernels (DistNestT m) Source #
Instance details Defined in Futhark.Pass.ExtractKernels.DistributeNests Methods localScope :: Scope Kernels -> DistNestT m a -> DistNestT m a Source #
Monad m => HasScope Kernels (DistNestT m) Source #
Instance details Defined in Futhark.Pass.ExtractKernels.DistributeNests Methods lookupType :: VName -> DistNestT m Type Source # lookupInfo :: VName -> DistNestT m (NameInfo Kernels) Source # askScope :: DistNestT m (Scope Kernels) Source # asksScope :: (Scope Kernels -> a) -> DistNestT m a Source #
Monad m => Monad (DistNestT m) Source #
Instance details Defined in Futhark.Pass.ExtractKernels.DistributeNests Methods (>>=) :: DistNestT m a -> (a -> DistNestT m b) -> DistNestT m b # (>>) :: DistNestT m a -> DistNestT m b -> DistNestT m b # return :: a -> DistNestT m a # fail :: String -> DistNestT m a #
Functor m => Functor (DistNestT m) Source #
Instance details Defined in Futhark.Pass.ExtractKernels.DistributeNests Methods fmap :: (a -> b) -> DistNestT m a -> DistNestT m b # (<$) :: a -> DistNestT m b -> DistNestT m a #
Applicative m => Applicative (DistNestT m) Source #
Instance details Defined in Futhark.Pass.ExtractKernels.DistributeNests Methods pure :: a -> DistNestT m a # (<>) :: DistNestT m (a -> b) -> DistNestT m a -> DistNestT m b # liftA2 :: (a -> b -> c) -> DistNestT m a -> DistNestT m b -> DistNestT m c # (>) :: DistNestT m a -> DistNestT m b -> DistNestT m b # (<*) :: DistNestT m a -> DistNestT m b -> DistNestT m a #
Monad m => MonadLogger (DistNestT m) Source #
Instance details Defined in Futhark.Pass.ExtractKernels.DistributeNests Methods logMsg :: ToLog a => a -> DistNestT m () Source # addLog :: Log -> DistNestT m () Source #
MonadFreshNames m => MonadFreshNames (DistNestT m) Source #
Instance details Defined in Futhark.Pass.ExtractKernels.DistributeNests Methods getNameSource :: DistNestT m VNameSource Source # putNameSource :: VNameSource -> DistNestT m () Source #
Monad m => MonadReader (DistEnv m) (DistNestT m) Source #
Instance details Defined in Futhark.Pass.ExtractKernels.DistributeNests Methods ask :: DistNestT m (DistEnv m) # local :: (DistEnv m -> DistEnv m) -> DistNestT m a -> DistNestT m a # reader :: (DistEnv m -> a) -> DistNestT m a #

distributeMap :: MonadFreshNames m => MapLoop -> DistAcc -> DistNestT m DistAcc Source #

distribute :: MonadFreshNames m => DistAcc -> DistNestT m DistAcc Source #

distributeSingleStm :: MonadFreshNames m => DistAcc -> Stm -> DistNestT m (Maybe (PostKernels, Result, KernelNest, DistAcc)) Source #

distributeMapBodyStms :: MonadFreshNames m => DistAcc -> Stms SOACS -> DistNestT m DistAcc Source #

postKernelsStms :: PostKernels -> KernelsStms Source #

addStmsToKernel :: KernelsStms -> DistAcc -> DistAcc Source #

addStmToKernel :: Monad m => Stm -> DistAcc -> m DistAcc Source #

permutationAndMissing :: Pattern -> [SubExp] -> Maybe ([Int], [PatElem]) Source #

addKernels :: Monad m => PostKernels -> DistNestT m () Source #

addKernel :: Monad m => KernelsStms -> DistNestT m () Source #

inNesting :: Monad m => KernelNest -> DistNestT m a -> DistNestT m a Source #