Futhark.Pass.ExtractKernels.DistributeNests

Documentation

type KernelsStms = Stms Kernels

data MapLoop

Constructors

MapLoop Pattern Certificates SubExp Lambda [VName]

mapLoopStm :: MapLoop -> Stm

bodyContainsParallelism :: Body -> Bool

lambdaContainsParallelism :: Lambda -> Bool

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

incrementalFlattening :: Bool

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

data DistEnv m

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)

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

Constructors

DistAcc

Fields

• distTargets :: Targets
• distStms :: KernelsStms

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

data DistNestT m a

Instances

MonadTrans DistNestT

Defined in Futhark.Pass.ExtractKernels.DistributeNests

Methods

lift :: Monad m => m a -> DistNestT m a

Monad m => LocalScope Kernels (DistNestT m)

Defined in Futhark.Pass.ExtractKernels.DistributeNests

Methods

localScope :: Scope Kernels -> DistNestT m a -> DistNestT m a

Monad m => HasScope Kernels (DistNestT m)

Defined in Futhark.Pass.ExtractKernels.DistributeNests

Methods

lookupType :: VName -> DistNestT m Type

lookupInfo :: VName -> DistNestT m (NameInfo Kernels)

askScope :: DistNestT m (Scope Kernels)

asksScope :: (Scope Kernels -> a) -> DistNestT m a

Monad m => Monad (DistNestT m)

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)

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)

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)

Defined in Futhark.Pass.ExtractKernels.DistributeNests

Methods

logMsg :: ToLog a => a -> DistNestT m ()

addLog :: Log -> DistNestT m ()

MonadFreshNames m => MonadFreshNames (DistNestT m)

Defined in Futhark.Pass.ExtractKernels.DistributeNests

Methods

getNameSource :: DistNestT m VNameSource

putNameSource :: VNameSource -> DistNestT m ()

Monad m => MonadReader (DistEnv m) (DistNestT m)

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

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

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

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

postKernelsStms :: PostKernels -> KernelsStms

addStmsToKernel :: KernelsStms -> DistAcc -> DistAcc

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

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

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

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

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