Futhark.Optimise.Simplify.Engine

Perform general rule-based simplification based on data dependency information. This module will:

• Perform common-subexpression elimination (CSE).
• Hoist expressions out of loops (including lambdas) and branches. This is done as aggressively as possible.
• Apply simplification rules (see Futhark.Optimise.Simplification.Rules).

If you just want to run the simplifier as simply as possible, you may prefer to use the Futhark.Optimise.Simplify module.

Monadic interface

data SimpleM rep a

Instances

Monad (SimpleM rep)

Defined in Futhark.Optimise.Simplify.Engine

Methods

(>>=) :: SimpleM rep a -> (a -> SimpleM rep b) -> SimpleM rep b

(>>) :: SimpleM rep a -> SimpleM rep b -> SimpleM rep b

return :: a -> SimpleM rep a

Functor (SimpleM rep)

Defined in Futhark.Optimise.Simplify.Engine

Methods

fmap :: (a -> b) -> SimpleM rep a -> SimpleM rep b

(<$) :: a -> SimpleM rep b -> SimpleM rep a

Applicative (SimpleM rep)

Defined in Futhark.Optimise.Simplify.Engine

Methods

pure :: a -> SimpleM rep a

(<*>) :: SimpleM rep (a -> b) -> SimpleM rep a -> SimpleM rep b

liftA2 :: (a -> b -> c) -> SimpleM rep a -> SimpleM rep b -> SimpleM rep c

(*>) :: SimpleM rep a -> SimpleM rep b -> SimpleM rep b

(<*) :: SimpleM rep a -> SimpleM rep b -> SimpleM rep a

MonadFreshNames (SimpleM rep)

Defined in Futhark.Optimise.Simplify.Engine

Methods

getNameSource :: SimpleM rep VNameSource

putNameSource :: VNameSource -> SimpleM rep ()

SimplifiableRep rep => LocalScope (Wise rep) (SimpleM rep)

Defined in Futhark.Optimise.Simplify.Engine

Methods

localScope :: Scope (Wise rep) -> SimpleM rep a -> SimpleM rep a

SimplifiableRep rep => HasScope (Wise rep) (SimpleM rep)

Defined in Futhark.Optimise.Simplify.Engine

Methods

lookupType :: VName -> SimpleM rep Type

lookupInfo :: VName -> SimpleM rep (NameInfo (Wise rep))

askScope :: SimpleM rep (Scope (Wise rep))

asksScope :: (Scope (Wise rep) -> a) -> SimpleM rep a

MonadReader (SimpleOps rep, Env rep) (SimpleM rep)

Defined in Futhark.Optimise.Simplify.Engine

Methods

ask :: SimpleM rep (SimpleOps rep, Env rep)

local :: ((SimpleOps rep, Env rep) -> (SimpleOps rep, Env rep)) -> SimpleM rep a -> SimpleM rep a

reader :: ((SimpleOps rep, Env rep) -> a) -> SimpleM rep a

MonadState (VNameSource, Bool, Certs) (SimpleM rep)

Defined in Futhark.Optimise.Simplify.Engine

Methods

get :: SimpleM rep (VNameSource, Bool, Certs)

put :: (VNameSource, Bool, Certs) -> SimpleM rep ()

state :: ((VNameSource, Bool, Certs) -> (a, (VNameSource, Bool, Certs))) -> SimpleM rep a

runSimpleM :: SimpleM rep a -> SimpleOps rep -> Env rep -> VNameSource -> ((a, Bool), VNameSource)

data SimpleOps rep

Constructors

SimpleOps

Fields

• mkExpDecS :: SymbolTable (Wise rep) -> Pat (Wise rep) -> Exp (Wise rep) -> SimpleM rep (ExpDec (Wise rep))
• mkBodyS :: SymbolTable (Wise rep) -> Stms (Wise rep) -> Result -> SimpleM rep (Body (Wise rep))
• protectHoistedOpS :: Protect (Builder (Wise rep))

  Make a hoisted Op safe. The SubExp is a boolean that is true when the value of the statement will actually be used.

• opUsageS :: Op (Wise rep) -> UsageTable
• simplifyOpS :: SimplifyOp rep (Op rep)

Instances

MonadReader (SimpleOps rep, Env rep) (SimpleM rep)

Defined in Futhark.Optimise.Simplify.Engine

Methods

ask :: SimpleM rep (SimpleOps rep, Env rep)

local :: ((SimpleOps rep, Env rep) -> (SimpleOps rep, Env rep)) -> SimpleM rep a -> SimpleM rep a

reader :: ((SimpleOps rep, Env rep) -> a) -> SimpleM rep a

type SimplifyOp rep op = op -> SimpleM rep (OpWithWisdom op, Stms (Wise rep))

bindableSimpleOps :: (SimplifiableRep rep, Buildable rep) => SimplifyOp rep (Op rep) -> SimpleOps rep

data Env rep

Instances

MonadReader (SimpleOps rep, Env rep) (SimpleM rep)

Defined in Futhark.Optimise.Simplify.Engine

Methods

ask :: SimpleM rep (SimpleOps rep, Env rep)

local :: ((SimpleOps rep, Env rep) -> (SimpleOps rep, Env rep)) -> SimpleM rep a -> SimpleM rep a

reader :: ((SimpleOps rep, Env rep) -> a) -> SimpleM rep a

emptyEnv :: RuleBook (Wise rep) -> HoistBlockers rep -> Env rep

data HoistBlockers rep

Constructors

HoistBlockers

Fields

• blockHoistPar :: BlockPred (Wise rep)

  Blocker for hoisting out of parallel loops.

• blockHoistSeq :: BlockPred (Wise rep)

  Blocker for hoisting out of sequential loops.

• blockHoistBranch :: BlockPred (Wise rep)

  Blocker for hoisting out of branches.

• isAllocation :: Stm (Wise rep) -> Bool

neverBlocks :: BlockPred rep

noExtraHoistBlockers :: HoistBlockers rep

neverHoist :: HoistBlockers rep

type BlockPred rep = SymbolTable rep -> UsageTable -> Stm rep -> Bool

orIf :: BlockPred rep -> BlockPred rep -> BlockPred rep

hasFree :: ASTRep rep => Names -> BlockPred rep

isConsumed :: BlockPred rep

isFalse :: Bool -> BlockPred rep

isOp :: BlockPred rep

isNotSafe :: ASTRep rep => BlockPred rep

asksEngineEnv :: (Env rep -> a) -> SimpleM rep a

askVtable :: SimpleM rep (SymbolTable (Wise rep))

localVtable :: (SymbolTable (Wise rep) -> SymbolTable (Wise rep)) -> SimpleM rep a -> SimpleM rep a

Building blocks

type SimplifiableRep rep = (ASTRep rep, Simplifiable (LetDec rep), Simplifiable (FParamInfo rep), Simplifiable (LParamInfo rep), Simplifiable (RetType rep), Simplifiable (BranchType rep), CanBeWise (Op rep), IndexOp (OpWithWisdom (Op rep)), BuilderOps (Wise rep), IsOp (Op rep))

class Simplifiable e where

Methods

simplify :: SimplifiableRep rep => e -> SimpleM rep e

Instances

Simplifiable Int

Defined in Futhark.Optimise.Simplify.Engine

Methods

simplify :: SimplifiableRep rep => Int -> SimpleM rep Int

Simplifiable ()

Defined in Futhark.Optimise.Simplify.Engine

Methods

simplify :: SimplifiableRep rep => () -> SimpleM rep ()

Simplifiable PrimType

Defined in Futhark.Optimise.Simplify.Engine

Methods

simplify :: SimplifiableRep rep => PrimType -> SimpleM rep PrimType

Simplifiable VName

Defined in Futhark.Optimise.Simplify.Engine

Methods

simplify :: SimplifiableRep rep => VName -> SimpleM rep VName

Simplifiable SubExp

Defined in Futhark.Optimise.Simplify.Engine

Methods

simplify :: SimplifiableRep rep => SubExp -> SimpleM rep SubExp

Simplifiable Certs

Defined in Futhark.Optimise.Simplify.Engine

Methods

simplify :: SimplifiableRep rep => Certs -> SimpleM rep Certs

Simplifiable Space

Defined in Futhark.Optimise.Simplify.Engine

Methods

simplify :: SimplifiableRep rep => Space -> SimpleM rep Space

Simplifiable ExtSize

Defined in Futhark.Optimise.Simplify.Engine

Methods

simplify :: SimplifiableRep rep => ExtSize -> SimpleM rep ExtSize

Simplifiable SubExpRes

Defined in Futhark.Optimise.Simplify.Engine

Methods

simplify :: SimplifiableRep rep => SubExpRes -> SimpleM rep SubExpRes

Simplifiable MemReturn

Defined in Futhark.IR.Mem

Methods

simplify :: SimplifiableRep rep => MemReturn -> SimpleM rep MemReturn

Simplifiable MemBind

Defined in Futhark.IR.Mem

Methods

simplify :: SimplifiableRep rep => MemBind -> SimpleM rep MemBind

Simplifiable SegSpace

Defined in Futhark.IR.SegOp

Methods

simplify :: SimplifiableRep rep => SegSpace -> SimpleM rep SegSpace

Simplifiable KernelResult

Defined in Futhark.IR.SegOp

Methods

simplify :: SimplifiableRep rep => KernelResult -> SimpleM rep KernelResult

Simplifiable SplitOrdering

Defined in Futhark.IR.SegOp

Methods

simplify :: SimplifiableRep rep => SplitOrdering -> SimpleM rep SplitOrdering

Simplifiable SegLevel

Defined in Futhark.IR.GPU.Op

Methods

simplify :: SimplifiableRep rep => SegLevel -> SimpleM rep SegLevel

Simplifiable a => Simplifiable [a]

Defined in Futhark.Optimise.Simplify.Engine

Methods

simplify :: SimplifiableRep rep => [a] -> SimpleM rep [a]

Simplifiable [FunReturns]

Defined in Futhark.IR.Mem

Methods

simplify :: SimplifiableRep rep => [FunReturns] -> SimpleM rep [FunReturns]

Simplifiable a => Simplifiable (Maybe a)

Defined in Futhark.Optimise.Simplify.Engine

Methods

simplify :: SimplifiableRep rep => Maybe a -> SimpleM rep (Maybe a)

Simplifiable d => Simplifiable (Slice d)

Defined in Futhark.Optimise.Simplify.Engine

Methods

simplify :: SimplifiableRep rep => Slice d -> SimpleM rep (Slice d)

Simplifiable d => Simplifiable (DimIndex d)

Defined in Futhark.Optimise.Simplify.Engine

Methods

simplify :: SimplifiableRep rep => DimIndex d -> SimpleM rep (DimIndex d)

Simplifiable d => Simplifiable (ShapeBase d)

Defined in Futhark.Optimise.Simplify.Engine

Methods

simplify :: SimplifiableRep rep => ShapeBase d -> SimpleM rep (ShapeBase d)

(Simplifiable a, Simplifiable b) => Simplifiable (a, b)

Defined in Futhark.Optimise.Simplify.Engine

Methods

simplify :: SimplifiableRep rep => (a, b) -> SimpleM rep (a, b)

Simplifiable shape => Simplifiable (TypeBase shape u)

Defined in Futhark.Optimise.Simplify.Engine

Methods

simplify :: SimplifiableRep rep => TypeBase shape u -> SimpleM rep (TypeBase shape u)

(Simplifiable a, Simplifiable b, Simplifiable c) => Simplifiable (a, b, c)

Defined in Futhark.Optimise.Simplify.Engine

Methods

simplify :: SimplifiableRep rep => (a, b, c) -> SimpleM rep (a, b, c)

(Simplifiable d, Simplifiable ret) => Simplifiable (MemInfo d u ret)

Defined in Futhark.IR.Mem

Methods

simplify :: SimplifiableRep rep => MemInfo d u ret -> SimpleM rep (MemInfo d u ret)

simplifyStms :: SimplifiableRep rep => Stms rep -> SimpleM rep (a, Stms (Wise rep)) -> SimpleM rep (a, Stms (Wise rep))

simplifyFun :: SimplifiableRep rep => FunDef rep -> SimpleM rep (FunDef (Wise rep))

simplifyLambda :: SimplifiableRep rep => Lambda rep -> SimpleM rep (Lambda (Wise rep), Stms (Wise rep))

simplifyLambdaNoHoisting :: SimplifiableRep rep => Lambda rep -> SimpleM rep (Lambda (Wise rep))

bindLParams :: SimplifiableRep rep => [LParam (Wise rep)] -> SimpleM rep a -> SimpleM rep a

simplifyBody :: SimplifiableRep rep => [Diet] -> Body rep -> SimpleM rep (SimplifiedBody rep Result)

Simplify a single body. The [Diet] only covers the value elements, because the context cannot be consumed.

type SimplifiedBody rep a = ((a, UsageTable), Stms (Wise rep))

data SymbolTable rep

Instances

Semigroup (SymbolTable rep)

Defined in Futhark.Analysis.SymbolTable

Methods

(<>) :: SymbolTable rep -> SymbolTable rep -> SymbolTable rep

sconcat :: NonEmpty (SymbolTable rep) -> SymbolTable rep

stimes :: Integral b => b -> SymbolTable rep -> SymbolTable rep

Monoid (SymbolTable rep)

Defined in Futhark.Analysis.SymbolTable

Methods

mempty :: SymbolTable rep

mappend :: SymbolTable rep -> SymbolTable rep -> SymbolTable rep

mconcat :: [SymbolTable rep] -> SymbolTable rep

hoistStms :: SimplifiableRep rep => RuleBook (Wise rep) -> BlockPred (Wise rep) -> SymbolTable (Wise rep) -> UsageTable -> Stms (Wise rep) -> SimpleM rep (Stms (Wise rep), Stms (Wise rep))

blockIf :: SimplifiableRep rep => BlockPred (Wise rep) -> SimpleM rep (SimplifiedBody rep a) -> SimpleM rep ((Stms (Wise rep), a), Stms (Wise rep))

enterLoop :: SimpleM rep a -> SimpleM rep a

Indicate in the symbol table that we have descended into a loop.

module Futhark.Optimise.Simplify.Rep