Lean.Compiler.IR.NormIds

source

@[reducible, inline]

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

Type

Equations

Lean.IR.UniqueIds.M = StateT Lean.IR.IndexSet Id

Instances For

source

def Lean.IR.UniqueIds.checkId (id : Lean.IR.Index) :

Lean.IR.UniqueIds.M Bool

Equations

Lean.IR.UniqueIds.checkId id = modifyGet fun (s : Lean.IR.IndexSet) => if Lean.RBTree.contains s id = true then (false, s) else (true, Lean.RBTree.insert s id)

Instances For

source

def Lean.IR.UniqueIds.checkParams (ps : Array Lean.IR.Param) :

Lean.IR.UniqueIds.M Bool

Equations

Lean.IR.UniqueIds.checkParams ps = Array.allM (fun (p : Lean.IR.Param) => Lean.IR.UniqueIds.checkId p.x.idx) ps

Instances For

source

partial def Lean.IR.UniqueIds.checkFnBody :

Lean.IR.FnBody → Lean.IR.UniqueIds.M Bool

source

def Lean.IR.UniqueIds.checkDecl :

Lean.IR.Decl → Lean.IR.UniqueIds.M Bool

Equations

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

Instances For

source

def Lean.IR.Decl.uniqueIds (d : Lean.IR.Decl) :

Bool

Return true if variable, parameter and join point ids are unique

Equations

d.uniqueIds = StateT.run' (Lean.IR.UniqueIds.checkDecl d) ∅

Instances For

source

@[reducible, inline]

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

Type

Equations

Lean.IR.NormalizeIds.M = ReaderT Lean.IR.IndexRenaming Id

Instances For

source

def Lean.IR.NormalizeIds.normIndex (x : Lean.IR.Index) :

Lean.IR.NormalizeIds.M Lean.IR.Index

Equations

Lean.IR.NormalizeIds.normIndex x m = match Lean.RBMap.find? m x with | some y => y | none => x

Instances For

source

def Lean.IR.NormalizeIds.normVar (x : Lean.IR.VarId) :

Lean.IR.NormalizeIds.M Lean.IR.VarId

Equations

Lean.IR.NormalizeIds.normVar x = Lean.IR.VarId.mk <$> Lean.IR.NormalizeIds.normIndex x.idx

Instances For

source

def Lean.IR.NormalizeIds.normJP (x : Lean.IR.JoinPointId) :

Lean.IR.NormalizeIds.M Lean.IR.JoinPointId

Equations

Lean.IR.NormalizeIds.normJP x = Lean.IR.JoinPointId.mk <$> Lean.IR.NormalizeIds.normIndex x.idx

Instances For

source

def Lean.IR.NormalizeIds.normArg :

Lean.IR.Arg → Lean.IR.NormalizeIds.M Lean.IR.Arg

Equations

Lean.IR.NormalizeIds.normArg x = match x with | Lean.IR.Arg.var x => Lean.IR.Arg.var <$> Lean.IR.NormalizeIds.normVar x | other => pure other

Instances For

source

def Lean.IR.NormalizeIds.normArgs (as : Array Lean.IR.Arg) :

Lean.IR.NormalizeIds.M (Array Lean.IR.Arg)

Equations

Lean.IR.NormalizeIds.normArgs as m = Array.map (fun (a : Lean.IR.Arg) => Lean.IR.NormalizeIds.normArg a m) as

Instances For

source

def Lean.IR.NormalizeIds.normExpr :

Lean.IR.Expr → Lean.IR.NormalizeIds.M Lean.IR.Expr

Equations

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

Instances For

source

@[reducible, inline]

abbrev Lean.IR.NormalizeIds.N (α : Type) :

Type

Equations

Lean.IR.NormalizeIds.N = ReaderT Lean.IR.IndexRenaming (StateM Nat)

Instances For

source

@[inline]

def Lean.IR.NormalizeIds.withVar {α : Type} (x : Lean.IR.VarId) (k : Lean.IR.VarId → Lean.IR.NormalizeIds.N α) :

Lean.IR.NormalizeIds.N α

Equations

Lean.IR.NormalizeIds.withVar x k m = do let n ← getModify fun (n : Nat) => n + 1 k { idx := n } (Lean.RBMap.insert m x.idx n)

Instances For

source

@[inline]

def Lean.IR.NormalizeIds.withJP {α : Type} (x : Lean.IR.JoinPointId) (k : Lean.IR.JoinPointId → Lean.IR.NormalizeIds.N α) :

Lean.IR.NormalizeIds.N α

Equations

Lean.IR.NormalizeIds.withJP x k m = do let n ← getModify fun (n : Nat) => n + 1 k { idx := n } (Lean.RBMap.insert m x.idx n)

Instances For

source

@[inline]

def Lean.IR.NormalizeIds.withParams {α : Type} (ps : Array Lean.IR.Param) (k : Array Lean.IR.Param → Lean.IR.NormalizeIds.N α) :

Lean.IR.NormalizeIds.N α

Equations

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

Instances For

source

instance Lean.IR.NormalizeIds.instMonadLiftMN :

MonadLift Lean.IR.NormalizeIds.M Lean.IR.NormalizeIds.N

Equations

Lean.IR.NormalizeIds.instMonadLiftMN = { monadLift := fun {α : Type} (x : Lean.IR.NormalizeIds.M α) (m : Lean.IR.IndexRenaming) => pure (x m) }

source

partial def Lean.IR.NormalizeIds.normFnBody :

Lean.IR.FnBody → Lean.IR.NormalizeIds.N Lean.IR.FnBody

source

def Lean.IR.NormalizeIds.normDecl (d : Lean.IR.Decl) :

Lean.IR.NormalizeIds.N Lean.IR.Decl

Equations

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

Instances For

source

def Lean.IR.Decl.normalizeIds (d : Lean.IR.Decl) :

Lean.IR.Decl

Create a declaration equivalent to d s.t. d.normalizeIds.uniqueIds == true

Equations

d.normalizeIds = StateT.run' (Lean.IR.NormalizeIds.normDecl d ∅) 1

Instances For

Apply a function f : VarId → VarId to variable occurrences. The following functions assume the IR code does not have variable shadowing.

source

@[inline]

def Lean.IR.MapVars.mapArg (f : Lean.IR.VarId → Lean.IR.VarId) :

Lean.IR.Arg → Lean.IR.Arg

Equations

Lean.IR.MapVars.mapArg f x = match x with | Lean.IR.Arg.var x => Lean.IR.Arg.var (f x) | a => a

Instances For

source

def Lean.IR.MapVars.mapArgs (f : Lean.IR.VarId → Lean.IR.VarId) (as : Array Lean.IR.Arg) :

Array Lean.IR.Arg

Equations

Lean.IR.MapVars.mapArgs f as = Array.map (Lean.IR.MapVars.mapArg f) as

Instances For

source

def Lean.IR.MapVars.mapExpr (f : Lean.IR.VarId → Lean.IR.VarId) :

Lean.IR.Expr → Lean.IR.Expr

Equations

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

Instances For

source

partial def Lean.IR.MapVars.mapFnBody (f : Lean.IR.VarId → Lean.IR.VarId) :

Lean.IR.FnBody → Lean.IR.FnBody

source

@[inline]

def Lean.IR.FnBody.mapVars (f : Lean.IR.VarId → Lean.IR.VarId) (b : Lean.IR.FnBody) :

Lean.IR.FnBody

Equations

Lean.IR.FnBody.mapVars f b = Lean.IR.MapVars.mapFnBody f b

Instances For

source

def Lean.IR.FnBody.replaceVar (x : Lean.IR.VarId) (y : Lean.IR.VarId) (b : Lean.IR.FnBody) :

Lean.IR.FnBody

Replace x with y in b. This function assumes b does not shadow x

Equations

Lean.IR.FnBody.replaceVar x y b = Lean.IR.FnBody.mapVars (fun (z : Lean.IR.VarId) => if (x == z) = true then y else z) b

Instances For