113 lines
4.4 KiB
Text
113 lines
4.4 KiB
Text
/-
|
|
Copyright (c) 2019 Microsoft Corporation. All rights reserved.
|
|
Released under Apache 2.0 license as described in the file LICENSE.
|
|
Authors: Leonardo de Moura
|
|
-/
|
|
prelude
|
|
import init.lean.compiler.ir.compilerm
|
|
|
|
namespace Lean
|
|
namespace IR
|
|
|
|
namespace Checker
|
|
|
|
abbrev M := ExceptT String (ReaderT LocalContext Id)
|
|
|
|
def checkVar (x : VarId) : M Unit :=
|
|
do ctx ← read,
|
|
unless (ctx.isLocalVar x.idx || ctx.isParam x.idx) $ throw ("unknown variable '" ++ toString x ++ "'")
|
|
|
|
def checkJP (j : JoinPointId) : M Unit :=
|
|
do ctx ← read,
|
|
unless (ctx.isJP j.idx) $ throw ("unknown join point '" ++ toString j ++ "'")
|
|
|
|
def checkArg (a : Arg) : M Unit :=
|
|
match a with
|
|
| Arg.var x := checkVar x
|
|
| other := pure ()
|
|
|
|
def checkArgs (as : Array Arg) : M Unit :=
|
|
as.mfor checkArg
|
|
|
|
@[inline] def checkType (ty : IRType) (p : IRType → Bool) : M Unit :=
|
|
unless (p ty) $ throw ("unexpected type")
|
|
|
|
def checkObjType (ty : IRType) : M Unit := checkType ty IRType.isObj
|
|
|
|
def checkScalarType (ty : IRType) : M Unit := checkType ty IRType.isScalar
|
|
|
|
@[inline] def checkVarType (x : VarId) (p : IRType → Bool) : M Unit :=
|
|
do ctx ← read,
|
|
match ctx.getType x with
|
|
| some ty := checkType ty p
|
|
| none := throw ("unknown variable '" ++ toString x ++ "'")
|
|
|
|
def checkObjVar (x : VarId) : M Unit :=
|
|
checkVarType x IRType.isObj
|
|
|
|
def checkScalarVar (x : VarId) : M Unit :=
|
|
checkVarType x IRType.isScalar
|
|
|
|
def checkExpr (ty : IRType) : Expr → M Unit
|
|
| (Expr.pap _ ys) := checkObjType ty *> checkArgs ys -- partial applications should always produce a closure object
|
|
| (Expr.ap x ys) := checkObjVar x *> checkArgs ys
|
|
| (Expr.fap c ys) := checkArgs ys
|
|
| (Expr.ctor c ys) := when c.isRef (checkObjType ty) *> checkArgs ys
|
|
| (Expr.reset x) := checkObjVar x *> checkObjType ty
|
|
| (Expr.reuse x i u ys) := checkObjVar x *> checkArgs ys *> checkObjType ty
|
|
| (Expr.box xty x) := checkObjType ty *> checkScalarVar x *> checkVarType x (==xty)
|
|
| (Expr.unbox x) := checkScalarType ty *> checkObjVar x
|
|
| (Expr.proj _ x) := checkObjVar x *> checkObjType ty
|
|
| (Expr.uproj _ x) := checkObjVar x *> checkType ty (==IRType.usize)
|
|
| (Expr.sproj _ _ x) := checkObjVar x *> checkScalarType ty
|
|
| (Expr.isShared x) := checkObjVar x *> checkType ty (==IRType.uint8)
|
|
| (Expr.isTaggedPtr x) := checkObjVar x *> checkType ty (==IRType.uint8)
|
|
| (Expr.lit (LitVal.str _)) := checkObjType ty
|
|
| (Expr.lit _) := pure ()
|
|
|
|
@[inline] def withParams (ps : Array Param) (k : M Unit) : M Unit :=
|
|
do ctx ← read,
|
|
ctx ← ps.mfoldl (λ (ctx : LocalContext) p, do
|
|
when (ctx.contains p.x.idx) $ throw ("invalid parameter declaration, shadowing is not allowed"),
|
|
pure $ ctx.addParam p) ctx,
|
|
adaptReader (λ _, ctx) k
|
|
|
|
partial def checkFnBody : FnBody → M Unit
|
|
| (FnBody.vdecl x t v b) := do
|
|
checkExpr t v,
|
|
ctx ← read,
|
|
when (ctx.contains x.idx) $ throw ("invalid variable declaration, shadowing is not allowed"),
|
|
adaptReader (λ ctx : LocalContext, ctx.addLocal x t v) (checkFnBody b)
|
|
| (FnBody.jdecl j ys v b) := do
|
|
withParams ys (checkFnBody v),
|
|
ctx ← read,
|
|
when (ctx.contains j.idx) $ throw ("invalid join point declaration, shadowing is not allowed"),
|
|
adaptReader (λ ctx : LocalContext, ctx.addJP j ys v) (checkFnBody b)
|
|
| (FnBody.set x _ y b) := checkVar x *> checkVar y *> checkFnBody b
|
|
| (FnBody.uset x _ y b) := checkVar x *> checkVar y *> checkFnBody b
|
|
| (FnBody.sset x _ _ y _ b) := checkVar x *> checkVar y *> checkFnBody b
|
|
| (FnBody.release x _ b) := checkVar x *> checkFnBody b
|
|
| (FnBody.inc x _ _ b) := checkVar x *> checkFnBody b
|
|
| (FnBody.dec x _ _ b) := checkVar x *> checkFnBody b
|
|
| (FnBody.mdata _ b) := checkFnBody b
|
|
| (FnBody.jmp j ys) := checkJP j *> checkArgs ys
|
|
| (FnBody.ret x) := checkArg x
|
|
| (FnBody.case _ x alts) := checkVar x *> alts.mfor (λ alt, checkFnBody alt.body)
|
|
| (FnBody.unreachable) := pure ()
|
|
|
|
def checkDecl : Decl → M Unit
|
|
| (Decl.fdecl f xs t b) := withParams xs (checkFnBody b)
|
|
| (Decl.extern f xs t _) := withParams xs (pure ())
|
|
|
|
end Checker
|
|
|
|
def checkDecl (decl : Decl) : CompilerM Unit :=
|
|
match Checker.checkDecl decl {} with
|
|
| Except.error msg := throw ("IR check failed at '" ++ toString decl.name ++ "', error: " ++ msg)
|
|
| other := pure ()
|
|
|
|
def checkDecls (decls : Array Decl) : CompilerM Unit :=
|
|
decls.mfor checkDecl
|
|
|
|
end IR
|
|
end Lean
|