Futhark.Pass.ExplicitAllocations

A generic transformation for adding memory allocations to a Futhark program. Specialised by specific representations in submodules.

Documentation

explicitAllocationsGeneric :: Allocable fromrep torep inner => (Op fromrep -> AllocM fromrep torep (Op torep)) -> (Exp torep -> AllocM fromrep torep [ExpHint]) -> Pass fromrep torep

explicitAllocationsInStmsGeneric :: (MonadFreshNames m, HasScope torep m, Allocable fromrep torep inner) => (Op fromrep -> AllocM fromrep torep (Op torep)) -> (Exp torep -> AllocM fromrep torep [ExpHint]) -> Stms fromrep -> m (Stms torep)

data ExpHint

Constructors

NoHint

Hint IxFun Space

defaultExpHints :: (Monad m, ASTRep rep) => Exp rep -> m [ExpHint]

type Allocable fromrep torep inner = (PrettyRep fromrep, PrettyRep torep, Mem torep inner, LetDec torep ~ LetDecMem, FParamInfo fromrep ~ DeclType, LParamInfo fromrep ~ Type, BranchType fromrep ~ ExtType, RetType fromrep ~ DeclExtType, BodyDec fromrep ~ (), BodyDec torep ~ (), ExpDec torep ~ (), SizeSubst inner, BuilderOps torep)

data AllocM fromrep torep a

Monad for adding allocations to an entire program.

Instances

ASTRep torep => LocalScope torep (AllocM fromrep torep)

Defined in Futhark.Pass.ExplicitAllocations

Methods

localScope :: Scope torep -> AllocM fromrep torep a -> AllocM fromrep torep a

ASTRep torep => HasScope torep (AllocM fromrep torep)

Defined in Futhark.Pass.ExplicitAllocations

Methods

lookupType :: VName -> AllocM fromrep torep Type

lookupInfo :: VName -> AllocM fromrep torep (NameInfo torep)

askScope :: AllocM fromrep torep (Scope torep)

asksScope :: (Scope torep -> a) -> AllocM fromrep torep a

Monad (AllocM fromrep torep)

Defined in Futhark.Pass.ExplicitAllocations

Methods

(>>=) :: AllocM fromrep torep a -> (a -> AllocM fromrep torep b) -> AllocM fromrep torep b

(>>) :: AllocM fromrep torep a -> AllocM fromrep torep b -> AllocM fromrep torep b

return :: a -> AllocM fromrep torep a

Functor (AllocM fromrep torep)

Defined in Futhark.Pass.ExplicitAllocations

Methods

fmap :: (a -> b) -> AllocM fromrep torep a -> AllocM fromrep torep b

(<$) :: a -> AllocM fromrep torep b -> AllocM fromrep torep a

Applicative (AllocM fromrep torep)

Defined in Futhark.Pass.ExplicitAllocations

Methods

pure :: a -> AllocM fromrep torep a

(<*>) :: AllocM fromrep torep (a -> b) -> AllocM fromrep torep a -> AllocM fromrep torep b

liftA2 :: (a -> b -> c) -> AllocM fromrep torep a -> AllocM fromrep torep b -> AllocM fromrep torep c

(*>) :: AllocM fromrep torep a -> AllocM fromrep torep b -> AllocM fromrep torep b

(<*) :: AllocM fromrep torep a -> AllocM fromrep torep b -> AllocM fromrep torep a

MonadFreshNames (AllocM fromrep torep)

Defined in Futhark.Pass.ExplicitAllocations

Methods

getNameSource :: AllocM fromrep torep VNameSource

putNameSource :: VNameSource -> AllocM fromrep torep ()

Allocable fromrep torep inner => MonadBuilder (AllocM fromrep torep)

Defined in Futhark.Pass.ExplicitAllocations

Associated Types

type Rep (AllocM fromrep torep)

Methods

mkExpDecM :: Pat (Rep (AllocM fromrep torep)) -> Exp (Rep (AllocM fromrep torep)) -> AllocM fromrep torep (ExpDec (Rep (AllocM fromrep torep)))

mkBodyM :: Stms (Rep (AllocM fromrep torep)) -> Result -> AllocM fromrep torep (Body (Rep (AllocM fromrep torep)))

mkLetNamesM :: [VName] -> Exp (Rep (AllocM fromrep torep)) -> AllocM fromrep torep (Stm (Rep (AllocM fromrep torep)))

addStm :: Stm (Rep (AllocM fromrep torep)) -> AllocM fromrep torep ()

addStms :: Stms (Rep (AllocM fromrep torep)) -> AllocM fromrep torep ()

collectStms :: AllocM fromrep torep a -> AllocM fromrep torep (a, Stms (Rep (AllocM fromrep torep)))

certifying :: Certs -> AllocM fromrep torep a -> AllocM fromrep torep a

MonadReader (AllocEnv fromrep torep) (AllocM fromrep torep)

Defined in Futhark.Pass.ExplicitAllocations

Methods

ask :: AllocM fromrep torep (AllocEnv fromrep torep)

local :: (AllocEnv fromrep torep -> AllocEnv fromrep torep) -> AllocM fromrep torep a -> AllocM fromrep torep a

reader :: (AllocEnv fromrep torep -> a) -> AllocM fromrep torep a

type Rep (AllocM fromrep torep)

Defined in Futhark.Pass.ExplicitAllocations

type Rep (AllocM fromrep torep) = torep

data AllocEnv fromrep torep

Constructors

AllocEnv

Fields

• chunkMap :: ChunkMap
• aggressiveReuse :: Bool

  Aggressively try to reuse memory in do-loops - should be True inside kernels, False outside.

• allocSpace :: Space

  When allocating memory, put it in this memory space. This is primarily used to ensure that group-wide statements store their results in local memory.

• envConsts :: Set VName

  The set of names that are known to be constants at kernel compile time.

• allocInOp :: Op fromrep -> AllocM fromrep torep (Op torep)
• envExpHints :: Exp torep -> AllocM fromrep torep [ExpHint]

Instances

MonadReader (AllocEnv fromrep torep) (AllocM fromrep torep)

Defined in Futhark.Pass.ExplicitAllocations

Methods

ask :: AllocM fromrep torep (AllocEnv fromrep torep)

local :: (AllocEnv fromrep torep -> AllocEnv fromrep torep) -> AllocM fromrep torep a -> AllocM fromrep torep a

reader :: (AllocEnv fromrep torep -> a) -> AllocM fromrep torep a

class SizeSubst op where

Minimal complete definition

opSizeSubst

Methods

opSizeSubst :: PatT dec -> op -> ChunkMap

opIsConst :: op -> Bool

Instances

SizeSubst ()

Defined in Futhark.Pass.ExplicitAllocations

Methods

opSizeSubst :: PatT dec -> () -> ChunkMap

opIsConst :: () -> Bool

SizeSubst op => SizeSubst (MemOp op)

Defined in Futhark.Pass.ExplicitAllocations

Methods

opSizeSubst :: PatT dec -> MemOp op -> ChunkMap

opIsConst :: MemOp op -> Bool

SizeSubst (SegOp lvl rep)

Defined in Futhark.Pass.ExplicitAllocations.SegOp

Methods

opSizeSubst :: PatT dec -> SegOp lvl rep -> ChunkMap

opIsConst :: SegOp lvl rep -> Bool

SizeSubst (MCOp rep op)

Defined in Futhark.Pass.ExplicitAllocations.MC

Methods

opSizeSubst :: PatT dec -> MCOp rep op -> ChunkMap

opIsConst :: MCOp rep op -> Bool

SizeSubst (HostOp rep op)

Defined in Futhark.Pass.ExplicitAllocations.GPU

Methods

opSizeSubst :: PatT dec -> HostOp rep op -> ChunkMap

opIsConst :: HostOp rep op -> Bool

allocInStms :: Allocable fromrep torep inner => Stms fromrep -> AllocM fromrep torep a -> AllocM fromrep torep a

allocForArray :: Allocable fromrep torep inner => Type -> Space -> AllocM fromrep torep VName

Allocate memory for a value of the given type.

simplifiable :: (SimplifiableRep rep, ExpDec rep ~ (), BodyDec rep ~ (), Mem rep inner) => (OpWithWisdom inner -> UsageTable) -> (OpWithWisdom inner -> SimpleM rep (OpWithWisdom inner, Stms (Wise rep))) -> SimpleOps rep

arraySizeInBytesExp :: Type -> PrimExp VName

mkLetNamesB' :: (LetDec (Rep m) ~ LetDecMem, Mem (Rep m) inner, MonadBuilder m, ExpDec (Rep m) ~ ()) => ExpDec (Rep m) -> [VName] -> Exp (Rep m) -> m (Stm (Rep m))

mkLetNamesB'' :: (BuilderOps rep, Mem rep inner, LetDec rep ~ LetDecMem, OpReturns (OpWithWisdom inner), ExpDec rep ~ (), Rep m ~ Wise rep, HasScope (Wise rep) m, MonadBuilder m, CanBeWise inner) => [VName] -> Exp (Wise rep) -> m (Stm (Wise rep))

dimAllocationSize :: ChunkMap -> SubExp -> SubExp

The subexpression giving the number of elements we should allocate space for. See ChunkMap comment.

type ChunkMap = Map VName SubExp

A mapping from chunk names to their maximum size. XXX FIXME HACK: This is part of a hack to add loop-invariant allocations to reduce kernels, because memory expansion does not use range analysis yet (it should).

Module re-exports

module Control.Monad.Reader

module Futhark.MonadFreshNames

module Futhark.Pass

module Futhark.Tools