Helper functions for backend code generators

def Lean.IR.isTailCallTo (g : Lake.Name) (b : Lean.IR.FnBody) : Bool

Return true iff b is of the form let x := g ys; ret x

Equations

• Lean.IR.isTailCallTo g b = match b with | Lean.IR.FnBody.vdecl x ty (Lean.IR.Expr.fap f ys) (Lean.IR.FnBody.ret (Lean.IR.Arg.var y)) => x == y && f == g | x => false

def Lean.IR.usesModuleFrom (env : Lean.Environment) (modulePrefix : Lake.Name) : Bool

Equations

• Lean.IR.usesModuleFrom env modulePrefix = List.any (Array.toList (Lean.Environment.allImportedModuleNames env)) fun (modName : Lake.Name) => Lean.Name.isPrefixOf modulePrefix modName

@[inline, reducible]

abbrev Lean.IR.CollectUsedDecls.M (α : Type) : Type

Equations

• Lean.IR.CollectUsedDecls.M = ReaderT Lean.Environment (StateM Lean.NameSet)

@[inline]

def Lean.IR.CollectUsedDecls.collect (f : Lean.IR.FunId) : Lean.IR.CollectUsedDecls.M Unit

Equations

• Lean.IR.CollectUsedDecls.collect f = modify fun (s : Lean.NameSet) => Lean.NameSet.insert s f

partial def Lean.IR.CollectUsedDecls.collectFnBody : Lean.IR.FnBody → Lean.IR.CollectUsedDecls.M Unit

def Lean.IR.CollectUsedDecls.collectInitDecl (fn : Lake.Name) : Lean.IR.CollectUsedDecls.M Unit

Equations

• Lean.IR.CollectUsedDecls.collectInitDecl fn = do let env ← read match Lean.getInitFnNameFor? env fn with | some initFn => Lean.IR.CollectUsedDecls.collect initFn | x => pure ()

def Lean.IR.CollectUsedDecls.collectDecl : Lean.IR.Decl → Lean.IR.CollectUsedDecls.M Lean.NameSet

Equations

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

def Lean.IR.collectUsedDecls (env : Lean.Environment) (decl : Lean.IR.Decl) (used : optParam Lean.NameSet ∅) : Lean.NameSet

Equations

• Lean.IR.collectUsedDecls env decl used = StateT.run' (Lean.IR.CollectUsedDecls.collectDecl decl env) used

@[inline, reducible]

abbrev Lean.IR.VarTypeMap : Type

Equations

• Lean.IR.VarTypeMap = Lean.HashMap Lean.IR.VarId Lean.IR.IRType

@[inline, reducible]

abbrev Lean.IR.JPParamsMap : Type

Equations

• Lean.IR.JPParamsMap = Lean.HashMap Lean.IR.JoinPointId (Array Lean.IR.Param)

@[inline, reducible]

abbrev Lean.IR.CollectMaps.Collector : Type

Equations

• Lean.IR.CollectMaps.Collector = (Lean.IR.VarTypeMap × Lean.IR.JPParamsMap → Lean.IR.VarTypeMap × Lean.IR.JPParamsMap)

@[inline]

def Lean.IR.CollectMaps.collectVar (x : Lean.IR.VarId) (t : Lean.IR.IRType) : Lean.IR.CollectMaps.Collector

Equations

• Lean.IR.CollectMaps.collectVar x✝ t x = match x with | (vs, js) => (Lean.HashMap.insert vs x✝ t, js)

def Lean.IR.CollectMaps.collectParams (ps : Array Lean.IR.Param) : Lean.IR.CollectMaps.Collector

Equations

• Lean.IR.CollectMaps.collectParams ps s = Array.foldl (fun (s : Lean.IR.VarTypeMap × Lean.IR.JPParamsMap) (p : Lean.IR.Param) => Lean.IR.CollectMaps.collectVar p.x p.ty s) s ps 0

@[inline]

def Lean.IR.CollectMaps.collectJP (j : Lean.IR.JoinPointId) (xs : Array Lean.IR.Param) : Lean.IR.CollectMaps.Collector

Equations

• Lean.IR.CollectMaps.collectJP j xs x = match x with | (vs, js) => (vs, Lean.HashMap.insert js j xs)

partial def Lean.IR.CollectMaps.collectFnBody : Lean.IR.FnBody → Lean.IR.CollectMaps.Collector

collectFnBody assumes the variables in

def Lean.IR.CollectMaps.collectDecl : Lean.IR.Decl → Lean.IR.CollectMaps.Collector

Equations

• Lean.IR.CollectMaps.collectDecl x = match x with | Lean.IR.Decl.fdecl f xs type b info => Lean.IR.CollectMaps.collectParams xs ∘ Lean.IR.CollectMaps.collectFnBody b | x => id

def Lean.IR.mkVarJPMaps (d : Lean.IR.Decl) : Lean.IR.VarTypeMap × Lean.IR.JPParamsMap

Return a pair (v, j), where v is a mapping from variable/parameter to type, and j is a mapping from join point to parameters. This function assumes d has normalized indexes (see normids.lean).

Equations

• Lean.IR.mkVarJPMaps d = Lean.IR.CollectMaps.collectDecl d (∅, ∅)