Futhark.Pass.ExtractKernels.DistributeNests

Documentation

data MapLoop

Constructors

MapLoop (Pat Type) (StmAux ()) SubExp (Lambda SOACS) [VName]

mapLoopStm :: MapLoop -> Stm SOACS

bodyContainsParallelism :: Body SOACS -> Bool

lambdaContainsParallelism :: Lambda SOACS -> Bool

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

histKernel :: (MonadBuilder m, DistRep (Rep m)) => (Lambda SOACS -> m (Lambda (Rep m))) -> SegOpLevel (Rep m) -> Pat Type -> [(VName, SubExp)] -> [KernelInput] -> Certs -> SubExp -> [HistOp SOACS] -> Lambda (Rep m) -> [VName] -> m (Stms (Rep m))

data DistEnv rep m

Constructors

DistEnv

Fields

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

Instances

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

Defined in Futhark.Pass.ExtractKernels.DistributeNests

Methods

ask :: DistNestT rep m (DistEnv rep m)

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

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

data DistAcc rep

Constructors

DistAcc

Fields

• distTargets :: Targets
• distStms :: Stms rep

runDistNestT :: (MonadLogger m, DistRep rep) => DistEnv rep m -> DistNestT rep m (DistAcc rep) -> m (Stms rep)

data DistNestT rep m a

Instances

(Monad m, ASTRep rep) => HasScope (rep :: k) (DistNestT rep m)

Defined in Futhark.Pass.ExtractKernels.DistributeNests

Methods

lookupType :: VName -> DistNestT rep m Type

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

askScope :: DistNestT rep m (Scope rep)

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

(Monad m, ASTRep rep) => LocalScope (rep :: k) (DistNestT rep m)

Defined in Futhark.Pass.ExtractKernels.DistributeNests

Methods

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

Applicative m => Applicative (DistNestT rep m)

Defined in Futhark.Pass.ExtractKernels.DistributeNests

Methods

pure :: a -> DistNestT rep m a

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

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

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

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

Functor m => Functor (DistNestT rep m)

Defined in Futhark.Pass.ExtractKernels.DistributeNests

Methods

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

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

Monad m => Monad (DistNestT rep m)

Defined in Futhark.Pass.ExtractKernels.DistributeNests

Methods

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

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

return :: a -> DistNestT rep m a

MonadFreshNames m => MonadFreshNames (DistNestT rep m)

Defined in Futhark.Pass.ExtractKernels.DistributeNests

Methods

getNameSource :: DistNestT rep m VNameSource

putNameSource :: VNameSource -> DistNestT rep m ()

Monad m => MonadLogger (DistNestT rep m)

Defined in Futhark.Pass.ExtractKernels.DistributeNests

Methods

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

addLog :: Log -> DistNestT rep m ()

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

Defined in Futhark.Pass.ExtractKernels.DistributeNests

Methods

ask :: DistNestT rep m (DistEnv rep m)

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

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

liftInner :: (LocalScope rep m, DistRep rep) => m a -> DistNestT rep m a

distributeMap :: (MonadFreshNames m, LocalScope rep m, DistRep rep) => MapLoop -> DistAcc rep -> DistNestT rep m (DistAcc rep)

distribute :: (MonadFreshNames m, LocalScope rep m, DistRep rep) => DistAcc rep -> DistNestT rep m (DistAcc rep)

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

distributeMapBodyStms :: (MonadFreshNames m, LocalScope rep m, DistRep rep) => DistAcc rep -> Stms SOACS -> DistNestT rep m (DistAcc rep)

addStmsToAcc :: Stms rep -> DistAcc rep -> DistAcc rep

addStmToAcc :: (MonadFreshNames m, DistRep rep) => Stm SOACS -> DistAcc rep -> DistNestT rep m (DistAcc rep)

permutationAndMissing :: Pat Type -> Result -> Maybe ([Int], [PatElem Type])

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

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

inNesting :: (Monad m, DistRep rep) => KernelNest -> DistNestT rep m a -> DistNestT rep m a