Futhark.Pass.ExplicitAllocations

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

Documentation

explicitAllocationsGeneric :: (Allocable fromlore tolore, Allocator tolore (AllocM fromlore tolore)) => (Op fromlore -> AllocM fromlore tolore (Op tolore)) -> (Exp tolore -> AllocM fromlore tolore [ExpHint]) -> Pass fromlore tolore

explicitAllocationsInStmsGeneric :: (MonadFreshNames m, HasScope tolore m, Allocable fromlore tolore) => (Op fromlore -> AllocM fromlore tolore (Op tolore)) -> (Exp tolore -> AllocM fromlore tolore [ExpHint]) -> Stms fromlore -> m (Stms tolore)

data ExpHint

Constructors

NoHint

Hint IxFun Space

defaultExpHints :: (Monad m, ASTLore lore) => Exp lore -> m [ExpHint]

type Allocable fromlore tolore = (PrettyLore fromlore, PrettyLore tolore, Mem tolore, FParamInfo fromlore ~ DeclType, LParamInfo fromlore ~ Type, BranchType fromlore ~ ExtType, RetType fromlore ~ DeclExtType, BodyDec fromlore ~ (), BodyDec tolore ~ (), ExpDec tolore ~ (), SizeSubst (Op tolore), BinderOps tolore)

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

Minimal complete definition

askDefaultSpace, askConsts

Methods

addAllocStm :: AllocStm -> m ()

default addAllocStm :: (Allocable fromlore lore, m ~ AllocM fromlore lore) => 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 fromlore lore => SubExp -> m SubExp

askConsts :: m (Set VName)

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

expHints :: Exp lore -> m [ExpHint]

Instances

Allocable fromlore tolore => Allocator tolore (AllocM fromlore tolore)

Defined in Futhark.Pass.ExplicitAllocations

Methods

addAllocStm :: AllocStm -> AllocM fromlore tolore ()

askDefaultSpace :: AllocM fromlore tolore Space

dimAllocationSize :: SubExp -> AllocM fromlore tolore SubExp

askConsts :: AllocM fromlore tolore (Set VName)

expHints :: Exp tolore -> AllocM fromlore tolore [ExpHint]

data AllocM fromlore tolore a

Monad for adding allocations to an entire program.

Instances

ASTLore tolore => LocalScope tolore (AllocM fromlore tolore)

Defined in Futhark.Pass.ExplicitAllocations

Methods

localScope :: Scope tolore -> AllocM fromlore tolore a -> AllocM fromlore tolore a

ASTLore tolore => HasScope tolore (AllocM fromlore tolore)

Defined in Futhark.Pass.ExplicitAllocations

Methods

lookupType :: VName -> AllocM fromlore tolore Type

lookupInfo :: VName -> AllocM fromlore tolore (NameInfo tolore)

askScope :: AllocM fromlore tolore (Scope tolore)

asksScope :: (Scope tolore -> a) -> AllocM fromlore tolore a

Allocable fromlore tolore => Allocator tolore (AllocM fromlore tolore)

Defined in Futhark.Pass.ExplicitAllocations

Methods

addAllocStm :: AllocStm -> AllocM fromlore tolore ()

askDefaultSpace :: AllocM fromlore tolore Space

dimAllocationSize :: SubExp -> AllocM fromlore tolore SubExp

askConsts :: AllocM fromlore tolore (Set VName)

expHints :: Exp tolore -> AllocM fromlore tolore [ExpHint]

Monad (AllocM fromlore tolore)

Defined in Futhark.Pass.ExplicitAllocations

Methods

(>>=) :: AllocM fromlore tolore a -> (a -> AllocM fromlore tolore b) -> AllocM fromlore tolore b

(>>) :: AllocM fromlore tolore a -> AllocM fromlore tolore b -> AllocM fromlore tolore b

return :: a -> AllocM fromlore tolore a

Functor (AllocM fromlore tolore)

Defined in Futhark.Pass.ExplicitAllocations

Methods

fmap :: (a -> b) -> AllocM fromlore tolore a -> AllocM fromlore tolore b

(<$) :: a -> AllocM fromlore tolore b -> AllocM fromlore tolore a

Applicative (AllocM fromlore tolore)

Defined in Futhark.Pass.ExplicitAllocations

Methods

pure :: a -> AllocM fromlore tolore a

(<*>) :: AllocM fromlore tolore (a -> b) -> AllocM fromlore tolore a -> AllocM fromlore tolore b

liftA2 :: (a -> b -> c) -> AllocM fromlore tolore a -> AllocM fromlore tolore b -> AllocM fromlore tolore c

(*>) :: AllocM fromlore tolore a -> AllocM fromlore tolore b -> AllocM fromlore tolore b

(<*) :: AllocM fromlore tolore a -> AllocM fromlore tolore b -> AllocM fromlore tolore a

MonadFreshNames (AllocM fromlore tolore)

Defined in Futhark.Pass.ExplicitAllocations

Methods

getNameSource :: AllocM fromlore tolore VNameSource

putNameSource :: VNameSource -> AllocM fromlore tolore ()

(Allocable fromlore tolore, Allocator tolore (AllocM fromlore tolore)) => MonadBinder (AllocM fromlore tolore)

Defined in Futhark.Pass.ExplicitAllocations

Associated Types

type Lore (AllocM fromlore tolore)

Methods

mkExpDecM :: Pattern (Lore (AllocM fromlore tolore)) -> Exp (Lore (AllocM fromlore tolore)) -> AllocM fromlore tolore (ExpDec (Lore (AllocM fromlore tolore)))

mkBodyM :: Stms (Lore (AllocM fromlore tolore)) -> Result -> AllocM fromlore tolore (Body (Lore (AllocM fromlore tolore)))

mkLetNamesM :: [VName] -> Exp (Lore (AllocM fromlore tolore)) -> AllocM fromlore tolore (Stm (Lore (AllocM fromlore tolore)))

addStm :: Stm (Lore (AllocM fromlore tolore)) -> AllocM fromlore tolore ()

addStms :: Stms (Lore (AllocM fromlore tolore)) -> AllocM fromlore tolore ()

collectStms :: AllocM fromlore tolore a -> AllocM fromlore tolore (a, Stms (Lore (AllocM fromlore tolore)))

certifying :: Certificates -> AllocM fromlore tolore a -> AllocM fromlore tolore a

MonadReader (AllocEnv fromlore tolore) (AllocM fromlore tolore)

Defined in Futhark.Pass.ExplicitAllocations

Methods

ask :: AllocM fromlore tolore (AllocEnv fromlore tolore)

local :: (AllocEnv fromlore tolore -> AllocEnv fromlore tolore) -> AllocM fromlore tolore a -> AllocM fromlore tolore a

reader :: (AllocEnv fromlore tolore -> a) -> AllocM fromlore tolore a

type Lore (AllocM fromlore tolore)

Defined in Futhark.Pass.ExplicitAllocations

type Lore (AllocM fromlore tolore) = tolore

data AllocEnv fromlore tolore

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 fromlore -> AllocM fromlore tolore (Op tolore)
• envExpHints :: Exp tolore -> AllocM fromlore tolore [ExpHint]

Instances

MonadReader (AllocEnv fromlore tolore) (AllocM fromlore tolore)

Defined in Futhark.Pass.ExplicitAllocations

Methods

ask :: AllocM fromlore tolore (AllocEnv fromlore tolore)

local :: (AllocEnv fromlore tolore -> AllocEnv fromlore tolore) -> AllocM fromlore tolore a -> AllocM fromlore tolore a

reader :: (AllocEnv fromlore tolore -> a) -> AllocM fromlore tolore 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 lore)

Defined in Futhark.Pass.ExplicitAllocations.SegOp

Methods

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

opIsConst :: SegOp lvl lore -> Bool

SizeSubst (MCOp lore op)

Defined in Futhark.Pass.ExplicitAllocations.MC

Methods

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

opIsConst :: MCOp lore op -> Bool

SizeSubst (HostOp lore op)

Defined in Futhark.Pass.ExplicitAllocations.Kernels

Methods

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

opIsConst :: HostOp lore op -> Bool

allocInStms :: Allocable fromlore tolore => Stms fromlore -> AllocM fromlore tolore a -> AllocM fromlore tolore a

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

Allocate memory for a value of the given type.

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

arraySizeInBytesExp :: Type -> PrimExp VName

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

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

Module re-exports

module Control.Monad.Reader

module Futhark.MonadFreshNames

module Futhark.Pass

module Futhark.Tools