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, Allocator torep (AllocM fromrep torep)) => (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) => (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 = (PrettyRep fromrep, PrettyRep torep, Mem torep, FParamInfo fromrep ~ DeclType, LParamInfo fromrep ~ Type, BranchType fromrep ~ ExtType, RetType fromrep ~ DeclExtType, BodyDec fromrep ~ (), BodyDec torep ~ (), ExpDec torep ~ (), SizeSubst (Op torep), BinderOps torep)

class (MonadFreshNames m, LocalScope rep m, Mem rep) => Allocator rep m where

Minimal complete definition

askDefaultSpace, askConsts

Methods

addAllocStm :: AllocStm -> m ()

default addAllocStm :: (Allocable fromrep rep, m ~ AllocM fromrep rep) => AllocStm -> m ()

askDefaultSpace :: m Space

dimAllocationSize :: SubExp -> m SubExp

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

default dimAllocationSize :: m ~ AllocM fromrep rep => SubExp -> m SubExp

askConsts :: m (Set VName)

Get those names that are known to be constants at run-time.

expHints :: Exp rep -> m [ExpHint]

Instances

Allocable fromrep torep => Allocator torep (AllocM fromrep torep)

Defined in Futhark.Pass.ExplicitAllocations

Methods

addAllocStm :: AllocStm -> AllocM fromrep torep ()

askDefaultSpace :: AllocM fromrep torep Space

dimAllocationSize :: SubExp -> AllocM fromrep torep SubExp

askConsts :: AllocM fromrep torep (Set VName)

expHints :: Exp torep -> AllocM fromrep torep [ExpHint]

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

Allocable fromrep torep => Allocator torep (AllocM fromrep torep)

Defined in Futhark.Pass.ExplicitAllocations

Methods

addAllocStm :: AllocStm -> AllocM fromrep torep ()

askDefaultSpace :: AllocM fromrep torep Space

dimAllocationSize :: SubExp -> AllocM fromrep torep SubExp

askConsts :: AllocM fromrep torep (Set VName)

expHints :: Exp torep -> AllocM fromrep torep [ExpHint]

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, Allocator torep (AllocM fromrep torep)) => MonadBinder (AllocM fromrep torep)

Defined in Futhark.Pass.ExplicitAllocations

Associated Types

type Rep (AllocM fromrep torep)

Methods

mkExpDecM :: Pattern (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 :: Certificates -> 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 :: PatternT dec -> op -> ChunkMap

opIsConst :: op -> Bool

Instances

SizeSubst ()

Defined in Futhark.Pass.ExplicitAllocations

Methods

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

opIsConst :: () -> Bool

SizeSubst op => SizeSubst (MemOp op)

Defined in Futhark.Pass.ExplicitAllocations

Methods

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

opIsConst :: MemOp op -> Bool

SizeSubst (SegOp lvl rep)

Defined in Futhark.Pass.ExplicitAllocations.SegOp

Methods

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

opIsConst :: SegOp lvl rep -> Bool

SizeSubst (MCOp rep op)

Defined in Futhark.Pass.ExplicitAllocations.MC

Methods

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

opIsConst :: MCOp rep op -> Bool

SizeSubst (HostOp rep op)

Defined in Futhark.Pass.ExplicitAllocations.GPU

Methods

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

opIsConst :: HostOp rep op -> Bool

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

allocForArray :: Allocator rep m => Type -> Space -> m VName

Allocate memory for a value of the given type.

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

arraySizeInBytesExp :: Type -> PrimExp VName

mkLetNamesB' :: (Op (Rep m) ~ MemOp inner, MonadBinder m, ExpDec (Rep m) ~ (), Allocator (Rep m) (PatAllocM (Rep m))) => ExpDec (Rep m) -> [VName] -> Exp (Rep m) -> m (Stm (Rep m))

mkLetNamesB'' :: (Op (Rep m) ~ MemOp inner, ExpDec rep ~ (), HasScope (Wise rep) m, Allocator rep (PatAllocM rep), MonadBinder m, CanBeWise (Op rep)) => [VName] -> Exp (Wise rep) -> m (Stm (Wise rep))

Module re-exports

module Control.Monad.Reader

module Futhark.MonadFreshNames

module Futhark.Pass

module Futhark.Tools