80 lines
2.7 KiB
Text
80 lines
2.7 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.Basic
|
||
import Init.Lean.Compiler.IR.Format
|
||
import Init.Lean.Compiler.IR.CompilerM
|
||
import Init.Lean.Compiler.IR.PushProj
|
||
import Init.Lean.Compiler.IR.ElimDeadVars
|
||
import Init.Lean.Compiler.IR.SimpCase
|
||
import Init.Lean.Compiler.IR.ResetReuse
|
||
import Init.Lean.Compiler.IR.NormIds
|
||
import Init.Lean.Compiler.IR.Checker
|
||
import Init.Lean.Compiler.IR.Borrow
|
||
import Init.Lean.Compiler.IR.Boxing
|
||
import Init.Lean.Compiler.IR.RC
|
||
import Init.Lean.Compiler.IR.ExpandResetReuse
|
||
import Init.Lean.Compiler.IR.UnboxResult
|
||
import Init.Lean.Compiler.IR.ElimDeadBranches
|
||
import Init.Lean.Compiler.IR.EmitC
|
||
|
||
namespace Lean
|
||
namespace IR
|
||
|
||
private def compileAux (decls : Array Decl) : CompilerM Unit :=
|
||
do logDecls `init decls;
|
||
checkDecls decls;
|
||
decls ← elimDeadBranches decls;
|
||
logDecls `elim_dead_branches decls;
|
||
let decls := decls.map Decl.pushProj;
|
||
logDecls `push_proj decls;
|
||
let decls := decls.map Decl.insertResetReuse;
|
||
logDecls `reset_reuse decls;
|
||
let decls := decls.map Decl.elimDead;
|
||
logDecls `elim_dead decls;
|
||
let decls := decls.map Decl.simpCase;
|
||
logDecls `simp_case decls;
|
||
let decls := decls.map Decl.normalizeIds;
|
||
decls ← inferBorrow decls;
|
||
logDecls `borrow decls;
|
||
decls ← explicitBoxing decls;
|
||
logDecls `boxing decls;
|
||
decls ← explicitRC decls;
|
||
logDecls `rc decls;
|
||
let decls := decls.map Decl.expandResetReuse;
|
||
logDecls `expand_reset_reuse decls;
|
||
let decls := decls.map Decl.pushProj;
|
||
logDecls `push_proj decls;
|
||
logDecls `result decls;
|
||
checkDecls decls;
|
||
addDecls decls;
|
||
pure ()
|
||
|
||
@[export lean_ir_compile]
|
||
def compile (env : Environment) (opts : Options) (decls : Array Decl) : Log × (Except String Environment) :=
|
||
match (compileAux decls opts).run { env := env } with
|
||
| EStateM.Result.ok _ s => (s.log, Except.ok s.env)
|
||
| EStateM.Result.error msg s => (s.log, Except.error msg)
|
||
|
||
def addBoxedVersionAux (decl : Decl) : CompilerM Unit :=
|
||
do env ← getEnv;
|
||
if !ExplicitBoxing.requiresBoxedVersion env decl then pure ()
|
||
else do
|
||
let decl := ExplicitBoxing.mkBoxedVersion decl;
|
||
let decls : Array Decl := #[decl];
|
||
decls ← explicitRC decls;
|
||
decls.forM $ fun decl => modifyEnv $ fun env => addDeclAux env decl;
|
||
pure ()
|
||
|
||
-- Remark: we are ignoring the `Log` here. This should be fine.
|
||
@[export lean_ir_add_boxed_version]
|
||
def addBoxedVersion (env : Environment) (decl : Decl) : Except String Environment :=
|
||
match (addBoxedVersionAux decl Options.empty).run { env := env } with
|
||
| EStateM.Result.ok _ s => Except.ok s.env
|
||
| EStateM.Result.error msg s => Except.error msg
|
||
|
||
end IR
|
||
end Lean
|