structure Lean.Compiler.LCNF.PullFunDecls.ToPull : Type

Local function declaration and join point being pulled.

• isFun : Bool
• decl : FunDecl
• used : FVarIdSet

instance Lean.Compiler.LCNF.PullFunDecls.instInhabitedToPull : Inhabited ToPull

Equations

• Lean.Compiler.LCNF.PullFunDecls.instInhabitedToPull = { default := { isFun := default, decl := default, used := default } }

@[reducible, inline]

abbrev Lean.Compiler.LCNF.PullFunDecls.PullM (α : Type) : Type

The PullM state contains the local function declarations and join points being pulled.

Equations

• Lean.Compiler.LCNF.PullFunDecls.PullM = StateRefT' IO.RealWorld (List Lean.Compiler.LCNF.PullFunDecls.ToPull) Lean.Compiler.LCNF.CompilerM

def Lean.Compiler.LCNF.PullFunDecls.findFVarDirectDeps (fvarId : FVarId) : PullM (List ToPull)

Extract from the state any local function declarations that depends on the given free variable. The idea is that we have to stop pulling these declarations because they depend on fvarId.

Equations

• One or more equations did not get rendered due to their size.

def Lean.Compiler.LCNF.PullFunDecls.findFVarDirectDeps.go (fvarId : FVarId) (as keep dep : List ToPull) : CoreM (List ToPull × List ToPull)

Equations

• One or more equations did not get rendered due to their size.
• Lean.Compiler.LCNF.PullFunDecls.findFVarDirectDeps.go fvarId [] keep dep = pure (keep, dep)

partial def Lean.Compiler.LCNF.PullFunDecls.findFVarDepsFixpoint (todo : List ToPull) (acc : Array ToPull := #[]) : PullM (Array ToPull)

def Lean.Compiler.LCNF.PullFunDecls.findFVarDeps (fvarId : FVarId) : PullM (Array ToPull)

Equations

• Lean.Compiler.LCNF.PullFunDecls.findFVarDeps fvarId = do let ps ← Lean.Compiler.LCNF.PullFunDecls.findFVarDirectDeps fvarId Lean.Compiler.LCNF.PullFunDecls.findFVarDepsFixpoint ps

def Lean.Compiler.LCNF.PullFunDecls.findParamsDeps (params : Array Param) : PullM (Array ToPull)

Similar to findFVarDeps. Extract from the state any local function declarations that depends on the given parameters.

Equations

• One or more equations did not get rendered due to their size.

def Lean.Compiler.LCNF.PullFunDecls.ToPull.attach (p : ToPull) (k : Code) : Code

Construct the code fun p.decl k or jp p.decl k.

Equations

• p.attach k = if p.isFun = true then Lean.Compiler.LCNF.Code.fun p.decl k else Lean.Compiler.LCNF.Code.jp p.decl k

def Lean.Compiler.LCNF.PullFunDecls.attach (ps : Array ToPull) (k : Code) : Code

Attach the given array of local function declarations and join points to k.

Equations

• One or more equations did not get rendered due to their size.

def Lean.Compiler.LCNF.PullFunDecls.attach.go (ps : Array ToPull) : StateM (Code × Array Bool) Unit

Equations

• One or more equations did not get rendered due to their size.

def Lean.Compiler.LCNF.PullFunDecls.attach.visited (i : Nat) : StateM (Code × Array Bool) Bool

Equations

• Lean.Compiler.LCNF.PullFunDecls.attach.visited i = do let __do_lift ← get pure __do_lift.snd[i]!

partial def Lean.Compiler.LCNF.PullFunDecls.attach.visit (ps : Array ToPull) (i : Nat) : StateM (Code × Array Bool) Unit

def Lean.Compiler.LCNF.PullFunDecls.attachFVarDeps (fvarId : FVarId) (k : Code) : PullM Code

Extract from the state any local function declarations that depends on the given free variable, and attach to code k.

Equations

• Lean.Compiler.LCNF.PullFunDecls.attachFVarDeps fvarId k = do let ps ← Lean.Compiler.LCNF.PullFunDecls.findFVarDeps fvarId pure (Lean.Compiler.LCNF.PullFunDecls.attach ps k)

def Lean.Compiler.LCNF.PullFunDecls.attachParamsDeps (params : Array Param) (k : Code) : PullM Code

Similar to attachFVarDeps. Extract from the state any local function declarations that depends on the given parameters, and attach to code k.

Equations

• Lean.Compiler.LCNF.PullFunDecls.attachParamsDeps params k = do let ps ← Lean.Compiler.LCNF.PullFunDecls.findParamsDeps params pure (Lean.Compiler.LCNF.PullFunDecls.attach ps k)

def Lean.Compiler.LCNF.PullFunDecls.attachJps (k : Code) : PullM Code

Equations

• One or more equations did not get rendered due to their size.

partial def Lean.Compiler.LCNF.PullFunDecls.addToPull (isFun : Bool) (decl : FunDecl) : PullM Unit

Add local function declaration (or join point if isFun = false) to the state.

partial def Lean.Compiler.LCNF.PullFunDecls.pull (code : Code) : PullM Code

Pull local function declarations and join points in code. The state contains the declarations being pulled.

def Lean.Compiler.LCNF.Decl.pullFunDecls (decl : Decl) : CompilerM Decl

Pull local function declarations and join points in the given declaration.

Equations

• One or more equations did not get rendered due to their size.

def Lean.Compiler.LCNF.pullFunDecls : Pass

Equations

• Lean.Compiler.LCNF.pullFunDecls = Lean.Compiler.LCNF.Pass.mkPerDeclaration `pullFunDecls Lean.Compiler.LCNF.Decl.pullFunDecls Lean.Compiler.LCNF.Phase.base