Implements (extended) λPure and λRc proposed in the article "Counting Immutable Beans", Sebastian Ullrich and Leonardo de Moura.

The Lean to IR transformation produces λPure code, and this part is implemented in C++. The procedures described in the paper above are implemented in Lean.

@[reducible, inline]

abbrev Lean.IR.FunId : Type

Function identifier

Equations

• Lean.IR.FunId = Lean.Name

@[reducible, inline]

abbrev Lean.IR.Index : Type

Equations

• Lean.IR.Index = Nat

structure Lean.IR.VarId : Type

Variable identifier

• idx : Index

instance Lean.IR.instInhabitedVarId : Inhabited VarId

Equations

• Lean.IR.instInhabitedVarId = { default := { idx := default } }

instance Lean.IR.instReprVarId : Repr VarId

Equations

• Lean.IR.instReprVarId = { reprPrec := Lean.IR.reprVarId✝ }

structure Lean.IR.JoinPointId : Type

Join point identifier

• idx : Index

instance Lean.IR.instInhabitedJoinPointId : Inhabited JoinPointId

Equations

• Lean.IR.instInhabitedJoinPointId = { default := { idx := default } }

instance Lean.IR.instReprJoinPointId : Repr JoinPointId

Equations

• Lean.IR.instReprJoinPointId = { reprPrec := Lean.IR.reprJoinPointId✝ }

@[reducible, inline]

abbrev Lean.IR.Index.lt (a b : Index) : Bool

Equations

• a.lt b = decide (a < b)

instance Lean.IR.instBEqVarId : BEq VarId

Equations

• Lean.IR.instBEqVarId = { beq := fun (a b : Lean.IR.VarId) => a.idx == b.idx }

instance Lean.IR.instToStringVarId : ToString VarId

Equations

• Lean.IR.instToStringVarId = { toString := fun (a : Lean.IR.VarId) => "x_" ++ toString a.idx }

instance Lean.IR.instToFormatVarId : ToFormat VarId

Equations

• Lean.IR.instToFormatVarId = { format := fun (a : Lean.IR.VarId) => Std.Format.text (toString a) }

instance Lean.IR.instHashableVarId : Hashable VarId

Equations

• Lean.IR.instHashableVarId = { hash := fun (a : Lean.IR.VarId) => hash a.idx }

instance Lean.IR.instBEqJoinPointId : BEq JoinPointId

Equations

• Lean.IR.instBEqJoinPointId = { beq := fun (a b : Lean.IR.JoinPointId) => a.idx == b.idx }

instance Lean.IR.instToStringJoinPointId : ToString JoinPointId

Equations

• Lean.IR.instToStringJoinPointId = { toString := fun (a : Lean.IR.JoinPointId) => "block_" ++ toString a.idx }

instance Lean.IR.instToFormatJoinPointId : ToFormat JoinPointId

Equations

• Lean.IR.instToFormatJoinPointId = { format := fun (a : Lean.IR.JoinPointId) => Std.Format.text (toString a) }

instance Lean.IR.instHashableJoinPointId : Hashable JoinPointId

Equations

• Lean.IR.instHashableJoinPointId = { hash := fun (a : Lean.IR.JoinPointId) => hash a.idx }

@[reducible, inline]

abbrev Lean.IR.MData : Type

Equations

• Lean.IR.MData = Lean.KVMap

@[reducible, inline]

abbrev Lean.IR.MData.empty : MData

Equations

• Lean.IR.MData.empty = { entries := [] }

inductive Lean.IR.IRType : Type

Low Level IR types. Most are self explanatory.

• usize represents the C++ size_t Type. We have it here because it is 32-bit in 32-bit machines, and 64-bit in 64-bit machines, and we want the C++ backend for our Compiler to generate platform independent code.

• irrelevant for Lean types, propositions and proofs.

• object a pointer to a value in the heap.

• tobject a pointer to a value in the heap or tagged pointer (i.e., the least significant bit is 1) storing a scalar value.

• struct and union are used to return small values (e.g., Option, Prod, Except) on the stack.

Remark: the RC operations for tobject are slightly more expensive because we first need to test whether the tobject is really a pointer or not.

Remark: the Lean runtime assumes that sizeof(void*) == sizeof(sizeT). Lean cannot be compiled on old platforms where this is not True.

Since values of type struct and union are only used to return values, We assume they must be used/consumed "linearly". We use the term "linear" here to mean "exactly once" in each execution. That is, given x : S, where S is a struct, then one of the following must hold in each (execution) branch. 1- x occurs only at a single ret x instruction. That is, it is being consumed by being returned. 2- x occurs only at a single ctor. That is, it is being "consumed" by being stored into another struct/union. 3- We extract (aka project) every single field of x exactly once. That is, we are consuming x by consuming each of one of its components. Minor refinement: we don't need to consume scalar fields or struct/union fields that do not contain object fields.

• float : IRType
• uint8 : IRType
• uint16 : IRType
• uint32 : IRType
• uint64 : IRType
• usize : IRType
• irrelevant : IRType
• object : IRType
• tobject : IRType
• float32 : IRType
• struct (leanTypeName : Option Name) (types : Array IRType) : IRType
• union (leanTypeName : Name) (types : Array IRType) : IRType

instance Lean.IR.instInhabitedIRType : Inhabited IRType

Equations

• Lean.IR.instInhabitedIRType = { default := Lean.IR.IRType.float }

instance Lean.IR.instReprIRType : Repr IRType

Equations

• Lean.IR.instReprIRType = { reprPrec := Lean.IR.reprIRType✝ }

partial def Lean.IR.IRType.beq : IRType → IRType → Bool

instance Lean.IR.IRType.instBEq : BEq IRType

Equations

• Lean.IR.IRType.instBEq = { beq := Lean.IR.IRType.beq }

def Lean.IR.IRType.isScalar : IRType → Bool

Equations

• Lean.IR.IRType.float.isScalar = true
• Lean.IR.IRType.float32.isScalar = true
• Lean.IR.IRType.uint8.isScalar = true
• Lean.IR.IRType.uint16.isScalar = true
• Lean.IR.IRType.uint32.isScalar = true
• Lean.IR.IRType.uint64.isScalar = true
• Lean.IR.IRType.usize.isScalar = true
• x✝.isScalar = false

def Lean.IR.IRType.isObj : IRType → Bool

Equations

• Lean.IR.IRType.object.isObj = true
• Lean.IR.IRType.tobject.isObj = true
• x✝.isObj = false

def Lean.IR.IRType.isIrrelevant : IRType → Bool

Equations

• Lean.IR.IRType.irrelevant.isIrrelevant = true
• x✝.isIrrelevant = false

def Lean.IR.IRType.isStruct : IRType → Bool

Equations

• (Lean.IR.IRType.struct leanTypeName types).isStruct = true
• x✝.isStruct = false

def Lean.IR.IRType.isUnion : IRType → Bool

Equations

• (Lean.IR.IRType.union leanTypeName types).isUnion = true
• x✝.isUnion = false

inductive Lean.IR.Arg : Type

Arguments to applications, constructors, etc. We use irrelevant for Lean types, propositions and proofs that have been erased. Recall that for a Function f, we also generate f._rarg which does not take irrelevant arguments. However, f._rarg is only safe to be used in full applications.

• var (id : VarId) : Arg
• irrelevant : Arg

instance Lean.IR.instInhabitedArg : Inhabited Arg

Equations

• Lean.IR.instInhabitedArg = { default := Lean.IR.Arg.var default }

def Lean.IR.Arg.beq : Arg → Arg → Bool

Equations

• (Lean.IR.Arg.var x_2).beq (Lean.IR.Arg.var y) = (x_2 == y)
• Lean.IR.Arg.irrelevant.beq Lean.IR.Arg.irrelevant = true
• x✝¹.beq x✝ = false

instance Lean.IR.instBEqArg : BEq Arg

Equations

• Lean.IR.instBEqArg = { beq := Lean.IR.Arg.beq }

@[export lean_ir_mk_var_arg]

def Lean.IR.mkVarArg (id : VarId) : Arg

Equations

• Lean.IR.mkVarArg id = Lean.IR.Arg.var id

inductive Lean.IR.LitVal : Type

• num (v : Nat) : LitVal
• str (v : String) : LitVal

def Lean.IR.LitVal.beq : LitVal → LitVal → Bool

Equations

• (Lean.IR.LitVal.num v₁).beq (Lean.IR.LitVal.num v₂) = (v₁ == v₂)
• (Lean.IR.LitVal.str v₁).beq (Lean.IR.LitVal.str v₂) = (v₁ == v₂)
• x✝¹.beq x✝ = false

instance Lean.IR.instBEqLitVal : BEq LitVal

Equations

• Lean.IR.instBEqLitVal = { beq := Lean.IR.LitVal.beq }

structure Lean.IR.CtorInfo : Type

Constructor information.

• name is the Name of the Constructor in Lean.
• cidx is the Constructor index (aka tag).
• size is the number of arguments of type object/tobject.
• usize is the number of arguments of type usize.
• ssize is the number of bytes used to store scalar values.

Recall that a Constructor object contains a header, then a sequence of pointers to other Lean objects, a sequence of USize (i.e., size_t) scalar values, and a sequence of other scalar values.

• name : Name
• cidx : Nat
• size : Nat
• usize : Nat
• ssize : Nat

instance Lean.IR.instReprCtorInfo : Repr CtorInfo

Equations

• Lean.IR.instReprCtorInfo = { reprPrec := Lean.IR.reprCtorInfo✝ }

def Lean.IR.CtorInfo.beq : CtorInfo → CtorInfo → Bool

Equations

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

instance Lean.IR.instBEqCtorInfo : BEq CtorInfo

Equations

• Lean.IR.instBEqCtorInfo = { beq := Lean.IR.CtorInfo.beq }

def Lean.IR.CtorInfo.isRef (info : CtorInfo) : Bool

Equations

• info.isRef = (decide (info.size > 0) || decide (info.usize > 0) || decide (info.ssize > 0))

def Lean.IR.CtorInfo.isScalar (info : CtorInfo) : Bool

Equations

• info.isScalar = !info.isRef

inductive Lean.IR.Expr : Type

• ctor (i : CtorInfo) (ys : Array Arg) : Expr

  We use ctor mainly for constructing Lean object/tobject values lean_ctor_object in the runtime. This instruction is also used to creat struct and union return values. For union, only i.cidx is relevant. For struct, i is irrelevant.

• reset (n : Nat) (x : VarId) : Expr
• reuse (x : VarId) (i : CtorInfo) (updtHeader : Bool) (ys : Array Arg) : Expr

  reuse x in ctor_i ys instruction in the paper.

• proj (i : Nat) (x : VarId) : Expr

  Extract the tobject value at Position sizeof(void*)*i from x. We also use proj for extracting fields from struct return values, and casting union return values.

• uproj (i : Nat) (x : VarId) : Expr

  Extract the Usize value at Position sizeof(void*)*i from x.

• sproj (n offset : Nat) (x : VarId) : Expr

  Extract the scalar value at Position sizeof(void*)*n + offset from x.

• fap (c : FunId) (ys : Array Arg) : Expr

  Full application.

• pap (c : FunId) (ys : Array Arg) : Expr

  Partial application that creates a pap value (aka closure in our nonstandard terminology).

• ap (x : VarId) (ys : Array Arg) : Expr

  Application. x must be a pap value.

• box (ty : IRType) (x : VarId) : Expr

  Given x : ty where ty is a scalar type, this operation returns a value of Type tobject. For small scalar values, the Result is a tagged pointer, and no memory allocation is performed.

• unbox (x : VarId) : Expr

  Given x : [t]object, obtain the scalar value.

• lit (v : LitVal) : Expr
• isShared (x : VarId) : Expr

  Return 1 : uint8 Iff RC(x) > 1

@[export lean_ir_mk_ctor_expr]

def Lean.IR.mkCtorExpr (n : Name) (cidx size usize ssize : Nat) (ys : Array Arg) : Expr

Equations

• Lean.IR.mkCtorExpr n cidx size usize ssize ys = Lean.IR.Expr.ctor { name := n, cidx := cidx, size := size, usize := usize, ssize := ssize } ys

@[export lean_ir_mk_proj_expr]

def Lean.IR.mkProjExpr (i : Nat) (x : VarId) : Expr

Equations

• Lean.IR.mkProjExpr i x = Lean.IR.Expr.proj i x

@[export lean_ir_mk_uproj_expr]

def Lean.IR.mkUProjExpr (i : Nat) (x : VarId) : Expr

Equations

• Lean.IR.mkUProjExpr i x = Lean.IR.Expr.uproj i x

@[export lean_ir_mk_sproj_expr]

def Lean.IR.mkSProjExpr (n offset : Nat) (x : VarId) : Expr

Equations

• Lean.IR.mkSProjExpr n offset x = Lean.IR.Expr.sproj n offset x

@[export lean_ir_mk_fapp_expr]

def Lean.IR.mkFAppExpr (c : FunId) (ys : Array Arg) : Expr

Equations

• Lean.IR.mkFAppExpr c ys = Lean.IR.Expr.fap c ys

@[export lean_ir_mk_papp_expr]

def Lean.IR.mkPAppExpr (c : FunId) (ys : Array Arg) : Expr

Equations

• Lean.IR.mkPAppExpr c ys = Lean.IR.Expr.pap c ys

@[export lean_ir_mk_app_expr]

def Lean.IR.mkAppExpr (x : VarId) (ys : Array Arg) : Expr

Equations

• Lean.IR.mkAppExpr x ys = Lean.IR.Expr.ap x ys

@[export lean_ir_mk_num_expr]

def Lean.IR.mkNumExpr (v : Nat) : Expr

Equations

• Lean.IR.mkNumExpr v = Lean.IR.Expr.lit (Lean.IR.LitVal.num v)

@[export lean_ir_mk_str_expr]

def Lean.IR.mkStrExpr (v : String) : Expr

Equations

• Lean.IR.mkStrExpr v = Lean.IR.Expr.lit (Lean.IR.LitVal.str v)

structure Lean.IR.Param : Type

• x : VarId
• borrow : Bool
• ty : IRType

instance Lean.IR.instInhabitedParam : Inhabited Param

Equations

• Lean.IR.instInhabitedParam = { default := { x := default, borrow := default, ty := default } }

instance Lean.IR.instReprParam : Repr Param

Equations

• Lean.IR.instReprParam = { reprPrec := Lean.IR.reprParam✝ }

@[export lean_ir_mk_param]

def Lean.IR.mkParam (x : VarId) (borrow : Bool) (ty : IRType) : Param

Equations

• Lean.IR.mkParam x borrow ty = { x := x, borrow := borrow, ty := ty }

inductive Lean.IR.AltCore (FnBody : Type) : Type

• ctor {FnBody : Type} (info : CtorInfo) (b : FnBody) : AltCore FnBody
• default {FnBody : Type} (b : FnBody) : AltCore FnBody

inductive Lean.IR.FnBody : Type

• vdecl (x : VarId) (ty : IRType) (e : Expr) (b : FnBody) : FnBody

  let x : ty := e; b

• jdecl (j : JoinPointId) (xs : Array Param) (v b : FnBody) : FnBody

  Join point Declaration block_j (xs) := e; b

• set (x : VarId) (i : Nat) (y : Arg) (b : FnBody) : FnBody

  Store y at Position sizeof(void*)*i in x. x must be a Constructor object and RC(x) must be 1. This operation is not part of λPure is only used during optimization.

• setTag (x : VarId) (cidx : Nat) (b : FnBody) : FnBody
• uset (x : VarId) (i : Nat) (y : VarId) (b : FnBody) : FnBody

  Store y : Usize at Position sizeof(void*)*i in x. x must be a Constructor object and RC(x) must be 1.

• sset (x : VarId) (i offset : Nat) (y : VarId) (ty : IRType) (b : FnBody) : FnBody

  Store y : ty at Position sizeof(void*)*i + offset in x. x must be a Constructor object and RC(x) must be 1. ty must not be object, tobject, irrelevant nor Usize.

• inc (x : VarId) (n : Nat) (c persistent : Bool) (b : FnBody) : FnBody

  RC increment for object. If c == true, then inc must check whether x is a tagged pointer or not. If persistent == true then x is statically known to be a persistent object.

• dec (x : VarId) (n : Nat) (c persistent : Bool) (b : FnBody) : FnBody

  RC decrement for object. If c == true, then inc must check whether x is a tagged pointer or not. If persistent == true then x is statically known to be a persistent object.

• del (x : VarId) (b : FnBody) : FnBody
• mdata (d : MData) (b : FnBody) : FnBody
• case (tid : Name) (x : VarId) (xType : IRType) (cs : Array (AltCore FnBody)) : FnBody
• ret (x : Arg) : FnBody
• jmp (j : JoinPointId) (ys : Array Arg) : FnBody

  Jump to join point j

• unreachable : FnBody

instance Lean.IR.instInhabitedFnBody : Inhabited FnBody

Equations

• Lean.IR.instInhabitedFnBody = { default := Lean.IR.FnBody.unreachable }

@[reducible, inline]

abbrev Lean.IR.FnBody.nil : FnBody

Equations

• Lean.IR.FnBody.nil = Lean.IR.FnBody.unreachable

@[export lean_ir_mk_vdecl]

def Lean.IR.mkVDecl (x : VarId) (ty : IRType) (e : Expr) (b : FnBody) : FnBody

Equations

• Lean.IR.mkVDecl x ty e b = Lean.IR.FnBody.vdecl x ty e b

@[export lean_ir_mk_jdecl]

def Lean.IR.mkJDecl (j : JoinPointId) (xs : Array Param) (v b : FnBody) : FnBody

Equations

• Lean.IR.mkJDecl j xs v b = Lean.IR.FnBody.jdecl j xs v b

@[export lean_ir_mk_uset]

def Lean.IR.mkUSet (x : VarId) (i : Nat) (y : VarId) (b : FnBody) : FnBody

Equations

• Lean.IR.mkUSet x i y b = Lean.IR.FnBody.uset x i y b

@[export lean_ir_mk_sset]

def Lean.IR.mkSSet (x : VarId) (i offset : Nat) (y : VarId) (ty : IRType) (b : FnBody) : FnBody

Equations

• Lean.IR.mkSSet x i offset y ty b = Lean.IR.FnBody.sset x i offset y ty b

@[export lean_ir_mk_case]

def Lean.IR.mkCase (tid : Name) (x : VarId) (cs : Array (AltCore FnBody)) : FnBody

Equations

• Lean.IR.mkCase tid x cs = Lean.IR.FnBody.case tid x Lean.IR.IRType.object cs

@[export lean_ir_mk_ret]

def Lean.IR.mkRet (x : Arg) : FnBody

Equations

• Lean.IR.mkRet x = Lean.IR.FnBody.ret x

@[export lean_ir_mk_jmp]

def Lean.IR.mkJmp (j : JoinPointId) (ys : Array Arg) : FnBody

Equations

• Lean.IR.mkJmp j ys = Lean.IR.FnBody.jmp j ys

@[export lean_ir_mk_unreachable]

def Lean.IR.mkUnreachable : Unit → FnBody

Equations

• Lean.IR.mkUnreachable x✝ = Lean.IR.FnBody.unreachable

@[reducible, inline]

abbrev Lean.IR.Alt : Type

Equations

• Lean.IR.Alt = Lean.IR.AltCore Lean.IR.FnBody

@[reducible, match_pattern, inline]

abbrev Lean.IR.Alt.ctor (info : CtorInfo) (b : FnBody) : AltCore FnBody

Equations

• Lean.IR.Alt.ctor = Lean.IR.AltCore.ctor

@[reducible, match_pattern, inline]

abbrev Lean.IR.Alt.default (b : FnBody) : AltCore FnBody

Equations

• Lean.IR.Alt.default = Lean.IR.AltCore.default

instance Lean.IR.instInhabitedAlt : Inhabited Alt

Equations

• Lean.IR.instInhabitedAlt = { default := Lean.IR.Alt.default default }

def Lean.IR.FnBody.isTerminal : FnBody → Bool

Equations

• (Lean.IR.FnBody.case tid x_1 xType cs).isTerminal = true
• (Lean.IR.FnBody.ret x_1).isTerminal = true
• (Lean.IR.FnBody.jmp j ys).isTerminal = true
• Lean.IR.FnBody.unreachable.isTerminal = true
• x✝.isTerminal = false

def Lean.IR.FnBody.body : FnBody → FnBody

Equations

• (Lean.IR.FnBody.vdecl x_1 ty e b).body = b
• (Lean.IR.FnBody.jdecl j xs v b).body = b
• (Lean.IR.FnBody.set x_1 i y b).body = b
• (Lean.IR.FnBody.uset x_1 i y b).body = b
• (Lean.IR.FnBody.sset x_1 i offset y ty b).body = b
• (Lean.IR.FnBody.setTag x_1 cidx b).body = b
• (Lean.IR.FnBody.inc x_1 n c persistent b).body = b
• (Lean.IR.FnBody.dec x_1 n c persistent b).body = b
• (Lean.IR.FnBody.del x_1 b).body = b
• (Lean.IR.FnBody.mdata d b).body = b
• x✝.body = x✝

def Lean.IR.FnBody.setBody : FnBody → FnBody → FnBody

Equations

• (Lean.IR.FnBody.vdecl x_2 t v b).setBody x✝ = Lean.IR.FnBody.vdecl x_2 t v x✝
• (Lean.IR.FnBody.jdecl j xs v b).setBody x✝ = Lean.IR.FnBody.jdecl j xs v x✝
• (Lean.IR.FnBody.set x_2 i y b).setBody x✝ = Lean.IR.FnBody.set x_2 i y x✝
• (Lean.IR.FnBody.uset x_2 i y b).setBody x✝ = Lean.IR.FnBody.uset x_2 i y x✝
• (Lean.IR.FnBody.sset x_2 i o y t b).setBody x✝ = Lean.IR.FnBody.sset x_2 i o y t x✝
• (Lean.IR.FnBody.setTag x_2 i b).setBody x✝ = Lean.IR.FnBody.setTag x_2 i x✝
• (Lean.IR.FnBody.inc x_2 n c p b).setBody x✝ = Lean.IR.FnBody.inc x_2 n c p x✝
• (Lean.IR.FnBody.dec x_2 n c p b).setBody x✝ = Lean.IR.FnBody.dec x_2 n c p x✝
• (Lean.IR.FnBody.del x_2 b).setBody x✝ = Lean.IR.FnBody.del x_2 x✝
• (Lean.IR.FnBody.mdata d b).setBody x✝ = Lean.IR.FnBody.mdata d x✝
• x✝¹.setBody x✝ = x✝¹

@[inline]

def Lean.IR.FnBody.resetBody (b : FnBody) : FnBody

Equations

• b.resetBody = b.setBody Lean.IR.FnBody.nil

@[inline]

def Lean.IR.FnBody.split (b : FnBody) : FnBody × FnBody

If b is a non terminal, then return a pair (c, b') s.t. b == c <;> b', and c.body == FnBody.nil

Equations

• b.split = (b.resetBody, b.body)

def Lean.IR.AltCore.body : Alt → FnBody

Equations

• (Lean.IR.AltCore.ctor info b).body = b
• (Lean.IR.AltCore.default b).body = b

def Lean.IR.AltCore.setBody : Alt → FnBody → Alt

Equations

• (Lean.IR.AltCore.ctor c b).setBody x✝ = Lean.IR.Alt.ctor c x✝
• (Lean.IR.AltCore.default b).setBody x✝ = Lean.IR.Alt.default x✝

@[inline]

def Lean.IR.AltCore.modifyBody (f : FnBody → FnBody) : AltCore FnBody → Alt

Equations

• Lean.IR.AltCore.modifyBody f (Lean.IR.AltCore.ctor c b) = Lean.IR.Alt.ctor c (f b)
• Lean.IR.AltCore.modifyBody f (Lean.IR.AltCore.default b) = Lean.IR.Alt.default (f b)

@[inline]

def Lean.IR.AltCore.mmodifyBody {m : Type → Type} [Monad m] (f : FnBody → m FnBody) : AltCore FnBody → m Alt

Equations

• Lean.IR.AltCore.mmodifyBody f (Lean.IR.AltCore.ctor c b) = Lean.IR.Alt.ctor c <$> f b
• Lean.IR.AltCore.mmodifyBody f (Lean.IR.AltCore.default b) = Lean.IR.Alt.default <$> f b

def Lean.IR.Alt.isDefault : Alt → Bool

Equations

• Lean.IR.Alt.isDefault (Lean.IR.AltCore.ctor info b) = false
• Lean.IR.Alt.isDefault (Lean.IR.AltCore.default b) = true

def Lean.IR.push (bs : Array FnBody) (b : FnBody) : Array FnBody

Equations

• Lean.IR.push bs b = bs.push b.resetBody

partial def Lean.IR.flattenAux (b : FnBody) (r : Array FnBody) : Array FnBody × FnBody

def Lean.IR.FnBody.flatten (b : FnBody) : Array FnBody × FnBody

Equations

• b.flatten = Lean.IR.flattenAux b #[]

partial def Lean.IR.reshapeAux (a : Array FnBody) (i : Nat) (b : FnBody) : FnBody

def Lean.IR.reshape (bs : Array FnBody) (term : FnBody) : FnBody

Equations

• Lean.IR.reshape bs term = Lean.IR.reshapeAux bs bs.size term

@[inline]

def Lean.IR.modifyJPs (bs : Array FnBody) (f : FnBody → FnBody) : Array FnBody

Equations

• Lean.IR.modifyJPs bs f = Array.map (fun (b : Lean.IR.FnBody) => match b with | Lean.IR.FnBody.jdecl j xs v k => Lean.IR.FnBody.jdecl j xs (f v) k | other => other) bs

@[inline]

def Lean.IR.mmodifyJPs {m : Type → Type} [Monad m] (bs : Array FnBody) (f : FnBody → m FnBody) : m (Array FnBody)

Equations

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

@[export lean_ir_mk_alt]

def Lean.IR.mkAlt (n : Name) (cidx size usize ssize : Nat) (b : FnBody) : Alt

Equations

• Lean.IR.mkAlt n cidx size usize ssize b = Lean.IR.Alt.ctor { name := n, cidx := cidx, size := size, usize := usize, ssize := ssize } b

structure Lean.IR.DeclInfo : Type

Extra information associated with a declaration.

• sorryDep? : Option Name

  If some <blame>, then declaration depends on <blame> which uses a sorry axiom.

inductive Lean.IR.Decl : Type

• fdecl (f : FunId) (xs : Array Param) (type : IRType) (body : FnBody) (info : DeclInfo) : Decl
• extern (f : FunId) (xs : Array Param) (type : IRType) (ext : ExternAttrData) : Decl

instance Lean.IR.instInhabitedDecl : Inhabited Decl

Equations

• Lean.IR.instInhabitedDecl = { default := Lean.IR.Decl.extern default default default default }

def Lean.IR.Decl.name : Decl → FunId

Equations

• (Lean.IR.Decl.fdecl f xs type body info).name = f
• (Lean.IR.Decl.extern f xs type ext).name = f

def Lean.IR.Decl.params : Decl → Array Param

Equations

• (Lean.IR.Decl.fdecl f xs type body info).params = xs
• (Lean.IR.Decl.extern f xs type ext).params = xs

def Lean.IR.Decl.resultType : Decl → IRType

Equations

• (Lean.IR.Decl.fdecl f xs type body info).resultType = type
• (Lean.IR.Decl.extern f xs type ext).resultType = type

def Lean.IR.Decl.isExtern : Decl → Bool

Equations

• (Lean.IR.Decl.extern f xs type ext).isExtern = true
• x✝.isExtern = false

def Lean.IR.Decl.getInfo : Decl → DeclInfo

Equations

• (Lean.IR.Decl.fdecl f xs type body info).getInfo = info
• x✝.getInfo = { sorryDep? := none }

def Lean.IR.Decl.updateBody! (d : Decl) (bNew : FnBody) : Decl

Equations

• (Lean.IR.Decl.fdecl f xs type body info).updateBody! bNew = Lean.IR.Decl.fdecl f xs type bNew info
• d.updateBody! bNew = panicWithPosWithDecl "Lean.Compiler.IR.Basic" "Lean.IR.Decl.updateBody!" 433 9 "expected definition"

@[export lean_ir_mk_decl]

def Lean.IR.mkDecl (f : FunId) (xs : Array Param) (ty : IRType) (b : FnBody) : Decl

Equations

• Lean.IR.mkDecl f xs ty b = Lean.IR.Decl.fdecl f xs ty b { sorryDep? := none }

@[export lean_ir_mk_extern_decl]

def Lean.IR.mkExternDecl (f : FunId) (xs : Array Param) (ty : IRType) (e : ExternAttrData) : Decl

Equations

• Lean.IR.mkExternDecl f xs ty e = Lean.IR.Decl.extern f xs ty e

@[export lean_ir_mk_dummy_extern_decl]

def Lean.IR.mkDummyExternDecl (f : FunId) (xs : Array Param) (ty : IRType) : Decl

Equations

• Lean.IR.mkDummyExternDecl f xs ty = Lean.IR.Decl.fdecl f xs ty Lean.IR.FnBody.unreachable { sorryDep? := none }

@[reducible, inline]

abbrev Lean.IR.IndexSet : Type

Set of variable and join point names

Equations

• Lean.IR.IndexSet = Lean.RBTree Lean.IR.Index compare

instance Lean.IR.instInhabitedIndexSet : Inhabited IndexSet

Equations

• Lean.IR.instInhabitedIndexSet = { default := ∅ }

def Lean.IR.mkIndexSet (idx : Index) : IndexSet

Equations

• Lean.IR.mkIndexSet idx = Lean.RBTree.empty.insert idx

inductive Lean.IR.LocalContextEntry : Type

• param : IRType → LocalContextEntry
• localVar : IRType → Expr → LocalContextEntry
• joinPoint : Array Param → FnBody → LocalContextEntry

@[reducible, inline]

abbrev Lean.IR.LocalContext : Type

Equations

• Lean.IR.LocalContext = Lean.RBMap Lean.IR.Index Lean.IR.LocalContextEntry compare

def Lean.IR.LocalContext.addLocal (ctx : LocalContext) (x : VarId) (t : IRType) (v : Expr) : LocalContext

Equations

• ctx.addLocal x t v = Lean.RBMap.insert ctx x.idx (Lean.IR.LocalContextEntry.localVar t v)

def Lean.IR.LocalContext.addJP (ctx : LocalContext) (j : JoinPointId) (xs : Array Param) (b : FnBody) : LocalContext

Equations

• ctx.addJP j xs b = Lean.RBMap.insert ctx j.idx (Lean.IR.LocalContextEntry.joinPoint xs b)

def Lean.IR.LocalContext.addParam (ctx : LocalContext) (p : Param) : LocalContext

Equations

• ctx.addParam p = Lean.RBMap.insert ctx p.x.idx (Lean.IR.LocalContextEntry.param p.ty)

def Lean.IR.LocalContext.addParams (ctx : LocalContext) (ps : Array Param) : LocalContext

Equations

• ctx.addParams ps = Array.foldl Lean.IR.LocalContext.addParam ctx ps

def Lean.IR.LocalContext.isJP (ctx : LocalContext) (idx : Index) : Bool

Equations

• ctx.isJP idx = match Lean.RBMap.find? ctx idx with | some (Lean.IR.LocalContextEntry.joinPoint a a_1) => true | x => false

def Lean.IR.LocalContext.getJPBody (ctx : LocalContext) (j : JoinPointId) : Option FnBody

Equations

• ctx.getJPBody j = match Lean.RBMap.find? ctx j.idx with | some (Lean.IR.LocalContextEntry.joinPoint a b) => some b | x => none

def Lean.IR.LocalContext.getJPParams (ctx : LocalContext) (j : JoinPointId) : Option (Array Param)

Equations

• ctx.getJPParams j = match Lean.RBMap.find? ctx j.idx with | some (Lean.IR.LocalContextEntry.joinPoint ys a) => some ys | x => none

def Lean.IR.LocalContext.isParam (ctx : LocalContext) (idx : Index) : Bool

Equations

• ctx.isParam idx = match Lean.RBMap.find? ctx idx with | some (Lean.IR.LocalContextEntry.param a) => true | x => false

def Lean.IR.LocalContext.isLocalVar (ctx : LocalContext) (idx : Index) : Bool

Equations

• ctx.isLocalVar idx = match Lean.RBMap.find? ctx idx with | some (Lean.IR.LocalContextEntry.localVar a a_1) => true | x => false

def Lean.IR.LocalContext.contains (ctx : LocalContext) (idx : Index) : Bool

Equations

• ctx.contains idx = Lean.RBMap.contains ctx idx

def Lean.IR.LocalContext.eraseJoinPointDecl (ctx : LocalContext) (j : JoinPointId) : LocalContext

Equations

• ctx.eraseJoinPointDecl j = Lean.RBMap.erase ctx j.idx

def Lean.IR.LocalContext.getType (ctx : LocalContext) (x : VarId) : Option IRType

Equations

• ctx.getType x = match Lean.RBMap.find? ctx x.idx with | some (Lean.IR.LocalContextEntry.param t) => some t | some (Lean.IR.LocalContextEntry.localVar t a) => some t | x => none

def Lean.IR.LocalContext.getValue (ctx : LocalContext) (x : VarId) : Option Expr

Equations

• ctx.getValue x = match Lean.RBMap.find? ctx x.idx with | some (Lean.IR.LocalContextEntry.localVar a v) => some v | x => none

@[reducible, inline]

abbrev Lean.IR.IndexRenaming : Type

Equations

• Lean.IR.IndexRenaming = Lean.RBMap Lean.IR.Index Lean.IR.Index compare

class Lean.IR.AlphaEqv (α : Type) : Type

• aeqv : IndexRenaming → α → α → Bool

def Lean.IR.VarId.alphaEqv (ρ : IndexRenaming) (v₁ v₂ : VarId) : Bool

Equations

• Lean.IR.VarId.alphaEqv ρ v₁ v₂ = match Lean.RBMap.find? ρ v₁.idx with | some v => v == v₂.idx | none => v₁ == v₂

instance Lean.IR.instAlphaEqvVarId : AlphaEqv VarId

Equations

• Lean.IR.instAlphaEqvVarId = { aeqv := Lean.IR.VarId.alphaEqv }

def Lean.IR.Arg.alphaEqv (ρ : IndexRenaming) : Arg → Arg → Bool

Equations

• Lean.IR.Arg.alphaEqv ρ (Lean.IR.Arg.var x_2) (Lean.IR.Arg.var y) = Lean.IR.aeqv ρ x_2 y
• Lean.IR.Arg.alphaEqv ρ Lean.IR.Arg.irrelevant Lean.IR.Arg.irrelevant = true
• Lean.IR.Arg.alphaEqv ρ x✝¹ x✝ = false

instance Lean.IR.instAlphaEqvArg : AlphaEqv Arg

Equations

• Lean.IR.instAlphaEqvArg = { aeqv := Lean.IR.Arg.alphaEqv }

def Lean.IR.args.alphaEqv (ρ : IndexRenaming) (args₁ args₂ : Array Arg) : Bool

Equations

• Lean.IR.args.alphaEqv ρ args₁ args₂ = args₁.isEqv args₂ fun (a b : Lean.IR.Arg) => Lean.IR.aeqv ρ a b

instance Lean.IR.instAlphaEqvArrayArg : AlphaEqv (Array Arg)

Equations

• Lean.IR.instAlphaEqvArrayArg = { aeqv := Lean.IR.args.alphaEqv }

def Lean.IR.Expr.alphaEqv (ρ : IndexRenaming) : Expr → Expr → Bool

instance Lean.IR.instAlphaEqvExpr : AlphaEqv Expr

Equations

• Lean.IR.instAlphaEqvExpr = { aeqv := Lean.IR.Expr.alphaEqv }

def Lean.IR.addVarRename (ρ : IndexRenaming) (x₁ x₂ : Nat) : IndexRenaming

Equations

• Lean.IR.addVarRename ρ x₁ x₂ = if (x₁ == x₂) = true then ρ else Lean.RBMap.insert ρ x₁ x₂

def Lean.IR.addParamRename (ρ : IndexRenaming) (p₁ p₂ : Param) : Option IndexRenaming

Equations

• Lean.IR.addParamRename ρ p₁ p₂ = if (p₁.ty == p₂.ty && decide (p₁.borrow = p₂.borrow)) = true then some (Lean.IR.addVarRename ρ p₁.x.idx p₂.x.idx) else none

def Lean.IR.addParamsRename (ρ : IndexRenaming) (ps₁ ps₂ : Array Param) : Option IndexRenaming

Equations

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

partial def Lean.IR.FnBody.alphaEqv : IndexRenaming → FnBody → FnBody → Bool

def Lean.IR.FnBody.beq (b₁ b₂ : FnBody) : Bool

Equations

• b₁.beq b₂ = Lean.IR.FnBody.alphaEqv ∅ b₁ b₂

instance Lean.IR.instBEqFnBody : BEq FnBody

Equations

• Lean.IR.instBEqFnBody = { beq := Lean.IR.FnBody.beq }

@[reducible, inline]

abbrev Lean.IR.VarIdSet : Type

Equations

• Lean.IR.VarIdSet = Lean.RBTree Lean.IR.VarId fun (x y : Lean.IR.VarId) => compare x.idx y.idx

instance Lean.IR.instInhabitedVarIdSet : Inhabited VarIdSet

Equations

• Lean.IR.instInhabitedVarIdSet = { default := ∅ }

def Lean.IR.mkIf (x : VarId) (t e : FnBody) : FnBody

Equations

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

def Lean.IR.getUnboxOpName (t : IRType) : String

Equations

• Lean.IR.getUnboxOpName Lean.IR.IRType.usize = "lean_unbox_usize"
• Lean.IR.getUnboxOpName Lean.IR.IRType.uint32 = "lean_unbox_uint32"
• Lean.IR.getUnboxOpName Lean.IR.IRType.uint64 = "lean_unbox_uint64"
• Lean.IR.getUnboxOpName Lean.IR.IRType.float = "lean_unbox_float"
• Lean.IR.getUnboxOpName Lean.IR.IRType.float32 = "lean_unbox_float32"
• Lean.IR.getUnboxOpName t = "lean_unbox"