Futhark.Optimise.Simplify

Documentation

simplifyProg :: SimplifiableRep rep => SimpleOps rep -> RuleBook (Wise rep) -> HoistBlockers rep -> Prog rep -> PassM (Prog rep)

Simplify the given program. Even if the output differs from the output, meaningful simplification may not have taken place - the order of bindings may simply have been rearranged.

simplifySomething :: (MonadFreshNames m, SimplifiableRep rep) => (a -> SimpleM rep b) -> (b -> a) -> SimpleOps rep -> RuleBook (Wise rep) -> HoistBlockers rep -> SymbolTable (Wise rep) -> a -> m a

Run a simplification operation to convergence.

simplifyFun :: (MonadFreshNames m, SimplifiableRep rep) => SimpleOps rep -> RuleBook (Wise rep) -> HoistBlockers rep -> SymbolTable (Wise rep) -> FunDef rep -> m (FunDef rep)

Simplify the given function. Even if the output differs from the output, meaningful simplification may not have taken place - the order of bindings may simply have been rearranged. Runs in a loop until convergence.

simplifyLambda :: (MonadFreshNames m, HasScope rep m, SimplifiableRep rep) => SimpleOps rep -> RuleBook (Wise rep) -> HoistBlockers rep -> Lambda rep -> m (Lambda rep)

Simplify just a single Lambda.

simplifyStms :: (MonadFreshNames m, SimplifiableRep rep) => SimpleOps rep -> RuleBook (Wise rep) -> HoistBlockers rep -> Scope rep -> Stms rep -> m (SymbolTable (Wise rep), Stms rep)

Simplify a sequence of Stms.

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

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

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

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

noExtraHoistBlockers :: HoistBlockers rep

neverHoist :: HoistBlockers rep

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))

data HoistBlockers rep

data RuleBook rep

A collection of both top-down and bottom-up rules.

Instances

Semigroup (RuleBook rep)

Defined in Futhark.Optimise.Simplify.Rule

Methods

(<>) :: RuleBook rep -> RuleBook rep -> RuleBook rep

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

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

Monoid (RuleBook rep)

Defined in Futhark.Optimise.Simplify.Rule

Methods

mempty :: RuleBook rep

mappend :: RuleBook rep -> RuleBook rep -> RuleBook rep

mconcat :: [RuleBook rep] -> RuleBook rep