Type inference for LCNF

@[reducible, inline]

abbrev Lean.Compiler.LCNF.InferType.InferTypeM (α : Type) : Type

We use a regular local context to store temporary local declarations created during type inference.

Equations

• Lean.Compiler.LCNF.InferType.InferTypeM = ReaderT Lean.LocalContext Lean.Compiler.LCNF.CompilerM

def Lean.Compiler.LCNF.InferType.getBinderName (fvarId : FVarId) : InferTypeM Name

Equations

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

def Lean.Compiler.LCNF.InferType.getType (fvarId : FVarId) : InferTypeM Expr

Equations

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

def Lean.Compiler.LCNF.InferType.mkForallFVars (xs : Array Expr) (type : Expr) : InferTypeM Expr

Equations

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

def Lean.Compiler.LCNF.InferType.mkForallParams (params : Array Param) (type : Expr) : InferTypeM Expr

Equations

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

@[inline]

def Lean.Compiler.LCNF.InferType.withLocalDecl {α : Type} (binderName : Name) (type : Expr) (binderInfo : BinderInfo) (k : Expr → InferTypeM α) : InferTypeM α

Equations

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

def Lean.Compiler.LCNF.InferType.inferConstType (declName : Name) (us : List Level) : CompilerM Expr

Equations

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

def Lean.Compiler.LCNF.InferType.inferLitValueType (value : LitValue) : Expr

Equations

• Lean.Compiler.LCNF.InferType.inferLitValueType (Lean.Compiler.LCNF.LitValue.nat val) = Lean.mkConst `Nat
• Lean.Compiler.LCNF.InferType.inferLitValueType (Lean.Compiler.LCNF.LitValue.str val) = Lean.mkConst `String
• Lean.Compiler.LCNF.InferType.inferLitValueType (Lean.Compiler.LCNF.LitValue.uint8 val) = Lean.mkConst `UInt8
• Lean.Compiler.LCNF.InferType.inferLitValueType (Lean.Compiler.LCNF.LitValue.uint16 val) = Lean.mkConst `UInt16
• Lean.Compiler.LCNF.InferType.inferLitValueType (Lean.Compiler.LCNF.LitValue.uint32 val) = Lean.mkConst `UInt32
• Lean.Compiler.LCNF.InferType.inferLitValueType (Lean.Compiler.LCNF.LitValue.uint64 val) = Lean.mkConst `UInt64
• Lean.Compiler.LCNF.InferType.inferLitValueType (Lean.Compiler.LCNF.LitValue.usize val) = Lean.mkConst `USize

def Lean.Compiler.LCNF.InferType.inferArgType (arg : Arg) : InferTypeM Expr

Equations

• Lean.Compiler.LCNF.InferType.inferArgType Lean.Compiler.LCNF.Arg.erased = pure Lean.Compiler.LCNF.erasedExpr
• Lean.Compiler.LCNF.InferType.inferArgType (Lean.Compiler.LCNF.Arg.type e) = Lean.Compiler.LCNF.InferType.inferType e
• Lean.Compiler.LCNF.InferType.inferArgType (Lean.Compiler.LCNF.Arg.fvar fvarId) = liftM (Lean.Compiler.LCNF.getType fvarId)

partial def Lean.Compiler.LCNF.InferType.inferType (e : Expr) : InferTypeM Expr

def Lean.Compiler.LCNF.InferType.inferLetValueType (e : LetValue) : InferTypeM Expr

Equations

• One or more equations did not get rendered due to their size.
• Lean.Compiler.LCNF.InferType.inferLetValueType Lean.Compiler.LCNF.LetValue.erased = pure Lean.Compiler.LCNF.erasedExpr
• Lean.Compiler.LCNF.InferType.inferLetValueType (Lean.Compiler.LCNF.LetValue.lit v) = pure (Lean.Compiler.LCNF.InferType.inferLitValueType v)
• Lean.Compiler.LCNF.InferType.inferLetValueType (Lean.Compiler.LCNF.LetValue.proj structName idx fvarId) = Lean.Compiler.LCNF.InferType.inferProjType structName idx fvarId

def Lean.Compiler.LCNF.InferType.inferAppTypeCore (fType : Expr) (args : Array Arg) : InferTypeM Expr

Equations

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

partial def Lean.Compiler.LCNF.InferType.inferAppType (e : Expr) : InferTypeM Expr

def Lean.Compiler.LCNF.InferType.inferProjType (structName : Name) (idx : Nat) (s : FVarId) : InferTypeM Expr

Equations

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

partial def Lean.Compiler.LCNF.InferType.getLevel? (type : Expr) : InferTypeM (Option Level)

partial def Lean.Compiler.LCNF.InferType.inferForallType (e : Expr) : InferTypeM Expr

partial def Lean.Compiler.LCNF.InferType.inferForallType.go (e : Expr) (fvars : Array Expr) : InferTypeM Expr

partial def Lean.Compiler.LCNF.InferType.inferLambdaType (e : Expr) : InferTypeM Expr

partial def Lean.Compiler.LCNF.InferType.inferLambdaType.go (e : Expr) (fvars all : Array Expr) : InferTypeM Expr

def Lean.Compiler.LCNF.inferType (e : Expr) : CompilerM Expr

Equations

• Lean.Compiler.LCNF.inferType e = ReaderT.run (Lean.Compiler.LCNF.InferType.inferType e) { }

def Lean.Compiler.LCNF.inferAppType (fnType : Expr) (args : Array Arg) : CompilerM Expr

Equations

• Lean.Compiler.LCNF.inferAppType fnType args = ReaderT.run (Lean.Compiler.LCNF.InferType.inferAppTypeCore fnType args) { }

def Lean.Compiler.LCNF.getLevel (type : Expr) : CompilerM Level

Equations

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

def Lean.Compiler.LCNF.Arg.inferType (arg : Arg) : CompilerM Expr

Equations

• arg.inferType = ReaderT.run (Lean.Compiler.LCNF.InferType.inferArgType arg) { }

def Lean.Compiler.LCNF.LetValue.inferType (e : LetValue) : CompilerM Expr

Equations

• e.inferType = ReaderT.run (Lean.Compiler.LCNF.InferType.inferLetValueType e) { }

def Lean.Compiler.LCNF.Code.inferType (code : Code) : CompilerM Expr

Equations

• (Lean.Compiler.LCNF.Code.let decl k).inferType = k.inferType
• (Lean.Compiler.LCNF.Code.fun decl k).inferType = k.inferType
• (Lean.Compiler.LCNF.Code.jp decl k).inferType = k.inferType
• (Lean.Compiler.LCNF.Code.return fvarId).inferType = Lean.Compiler.LCNF.getType fvarId
• (Lean.Compiler.LCNF.Code.jmp fvarId args).inferType = do let __do_lift ← Lean.Compiler.LCNF.getType fvarId ReaderT.run (Lean.Compiler.LCNF.InferType.inferAppTypeCore __do_lift args) { }
• (Lean.Compiler.LCNF.Code.unreach type).inferType = pure type
• (Lean.Compiler.LCNF.Code.cases c).inferType = pure c.resultType

def Lean.Compiler.LCNF.Code.inferParamType (params : Array Param) (code : Code) : CompilerM Expr

Equations

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

def Lean.Compiler.LCNF.Alt.inferType (alt : Alt) : CompilerM Expr

Equations

• alt.inferType = alt.getCode.inferType

def Lean.Compiler.LCNF.mkAuxLetDecl (e : LetValue) (prefixName : Name := `_x) : CompilerM LetDecl

Equations

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

def Lean.Compiler.LCNF.mkForallParams (params : Array Param) (type : Expr) : CompilerM Expr

Equations

• Lean.Compiler.LCNF.mkForallParams params type = ReaderT.run (Lean.Compiler.LCNF.InferType.mkForallParams params type) { }

def Lean.Compiler.LCNF.mkAuxFunDecl (params : Array Param) (code : Code) (prefixName : Name := `_f) : CompilerM FunDecl

Equations

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

def Lean.Compiler.LCNF.mkAuxJpDecl (params : Array Param) (code : Code) (prefixName : Name := `_jp) : CompilerM FunDecl

Equations

• Lean.Compiler.LCNF.mkAuxJpDecl params code prefixName = Lean.Compiler.LCNF.mkAuxFunDecl params code prefixName

def Lean.Compiler.LCNF.mkAuxJpDecl' (param : Param) (code : Code) (prefixName : Name := `_jp) : CompilerM FunDecl

Equations

• Lean.Compiler.LCNF.mkAuxJpDecl' param code prefixName = Lean.Compiler.LCNF.mkAuxFunDecl #[param] code prefixName

def Lean.Compiler.LCNF.mkCasesResultType (alts : Array Alt) : CompilerM Expr

Equations

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

def Lean.Compiler.LCNF.isErasedCompatible (type : Expr) (predVars : Array Bool := #[]) : CompilerM Bool

Return true if type should be erased. See item 1 in the note above where x ◾ ◾ is a proposition and should be erased when the universe level parameter is set to 0.

Remark: predVars is a bitmask that indicates whether de-bruijn variables are predicates or not. That is, #i is a predicate if predVars[predVars.size - i - 1] = true

Equations

• Lean.Compiler.LCNF.isErasedCompatible type predVars = Lean.Compiler.LCNF.isErasedCompatible.go type predVars

partial def Lean.Compiler.LCNF.isErasedCompatible.go (type : Expr) (predVars : Array Bool) : CompilerM Bool

partial def Lean.Compiler.LCNF.eqvTypes (a b : Expr) : Bool

Return true if the given LCNF are equivalent. List Nat and (fun x => List x) Nat are both equivalent.