Futhark.Pass.ExtractKernels.DistributeNests

Documentation

data MapLoop

Constructors

MapLoop Pattern (StmAux ()) SubExp Lambda [VName]

mapLoopStm :: MapLoop -> Stm SOACS

bodyContainsParallelism :: Body SOACS -> Bool

lambdaContainsParallelism :: Lambda SOACS -> Bool

determineReduceOp :: MonadBinder m => Lambda SOACS -> [SubExp] -> m (Lambda SOACS, [SubExp], Shape)

histKernel :: (MonadBinder m, DistLore (Lore m)) => (Lambda SOACS -> m (Lambda (Lore m))) -> SegOpLevel (Lore m) -> PatternT Type -> [(VName, SubExp)] -> [KernelInput] -> Certificates -> SubExp -> [HistOp SOACS] -> Lambda (Lore m) -> [VName] -> m (Stms (Lore m))

data DistEnv lore m

Constructors

DistEnv

Fields

• distNest :: Nestings
• distScope :: Scope lore
• distOnTopLevelStms :: Stms SOACS -> DistNestT lore m (Stms lore)
• distOnInnerMap :: MapLoop -> DistAcc lore -> DistNestT lore m (DistAcc lore)
• distOnSOACSStms :: Stm SOACS -> Binder lore (Stms lore)
• distOnSOACSLambda :: Lambda SOACS -> Binder lore (Lambda lore)
• distSegLevel :: MkSegLevel lore m

Instances

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

Defined in Futhark.Pass.ExtractKernels.DistributeNests

Methods

ask :: DistNestT lore m (DistEnv lore m)

local :: (DistEnv lore m -> DistEnv lore m) -> DistNestT lore m a -> DistNestT lore m a

reader :: (DistEnv lore m -> a) -> DistNestT lore m a

data DistAcc lore

Constructors

DistAcc

Fields

• distTargets :: Targets
• distStms :: Stms lore

runDistNestT :: (MonadLogger m, DistLore lore) => DistEnv lore m -> DistNestT lore m (DistAcc lore) -> m (Stms lore)

data DistNestT lore m a

Instances

(Monad m, ASTLore lore) => LocalScope lore (DistNestT lore m)

Defined in Futhark.Pass.ExtractKernels.DistributeNests

Methods

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

(Monad m, ASTLore lore) => HasScope lore (DistNestT lore m)

Defined in Futhark.Pass.ExtractKernels.DistributeNests

Methods

lookupType :: VName -> DistNestT lore m Type

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

askScope :: DistNestT lore m (Scope lore)

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

Monad m => Monad (DistNestT lore m)

Defined in Futhark.Pass.ExtractKernels.DistributeNests

Methods

(>>=) :: DistNestT lore m a -> (a -> DistNestT lore m b) -> DistNestT lore m b

(>>) :: DistNestT lore m a -> DistNestT lore m b -> DistNestT lore m b

return :: a -> DistNestT lore m a

Functor m => Functor (DistNestT lore m)

Defined in Futhark.Pass.ExtractKernels.DistributeNests

Methods

fmap :: (a -> b) -> DistNestT lore m a -> DistNestT lore m b

(<$) :: a -> DistNestT lore m b -> DistNestT lore m a

Applicative m => Applicative (DistNestT lore m)

Defined in Futhark.Pass.ExtractKernels.DistributeNests

Methods

pure :: a -> DistNestT lore m a

(<*>) :: DistNestT lore m (a -> b) -> DistNestT lore m a -> DistNestT lore m b

liftA2 :: (a -> b -> c) -> DistNestT lore m a -> DistNestT lore m b -> DistNestT lore m c

(*>) :: DistNestT lore m a -> DistNestT lore m b -> DistNestT lore m b

(<*) :: DistNestT lore m a -> DistNestT lore m b -> DistNestT lore m a

Monad m => MonadLogger (DistNestT lore m)

Defined in Futhark.Pass.ExtractKernels.DistributeNests

Methods

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

addLog :: Log -> DistNestT lore m ()

MonadFreshNames m => MonadFreshNames (DistNestT lore m)

Defined in Futhark.Pass.ExtractKernels.DistributeNests

Methods

getNameSource :: DistNestT lore m VNameSource

putNameSource :: VNameSource -> DistNestT lore m ()

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

Defined in Futhark.Pass.ExtractKernels.DistributeNests

Methods

ask :: DistNestT lore m (DistEnv lore m)

local :: (DistEnv lore m -> DistEnv lore m) -> DistNestT lore m a -> DistNestT lore m a

reader :: (DistEnv lore m -> a) -> DistNestT lore m a

liftInner :: (LocalScope lore m, DistLore lore) => m a -> DistNestT lore m a

distributeMap :: (MonadFreshNames m, LocalScope lore m, DistLore lore) => MapLoop -> DistAcc lore -> DistNestT lore m (DistAcc lore)

distribute :: (MonadFreshNames m, LocalScope lore m, DistLore lore) => DistAcc lore -> DistNestT lore m (DistAcc lore)

distributeSingleStm :: (MonadFreshNames m, LocalScope lore m, DistLore lore) => DistAcc lore -> Stm SOACS -> DistNestT lore m (Maybe (PostStms lore, Result, KernelNest, DistAcc lore))

distributeMapBodyStms :: (MonadFreshNames m, LocalScope lore m, DistLore lore) => DistAcc lore -> Stms SOACS -> DistNestT lore m (DistAcc lore)

addStmsToAcc :: Stms lore -> DistAcc lore -> DistAcc lore

addStmToAcc :: (MonadFreshNames m, DistLore lore) => Stm SOACS -> DistAcc lore -> DistNestT lore m (DistAcc lore)

permutationAndMissing :: PatternT Type -> [SubExp] -> Maybe ([Int], [PatElemT Type])

addPostStms :: Monad m => PostStms lore -> DistNestT lore m ()

postStm :: Monad m => Stms lore -> DistNestT lore m ()

inNesting :: (Monad m, DistLore lore) => KernelNest -> DistNestT lore m a -> DistNestT lore m a