f2bb86f45c
@Kha @dselsam I was experiencing an insane code explosion with the previous approach. There were too many definitions marked with `@[specialize]`. `Meta.c` was reaching 0.5 million lines of code. We would need a more sophisticated code specializer cache to handle this kind of code. The new approach is much simpler. I don't see any major disadvantages.
33 lines
1.2 KiB
Text
33 lines
1.2 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.AuxRecursor
|
||
import Init.Lean.WHNF
|
||
import Init.Lean.Meta.Basic
|
||
import Init.Lean.Meta.LevelDefEq
|
||
|
||
namespace Lean
|
||
namespace Meta
|
||
|
||
def isAuxDef? (constName : Name) : MetaM Bool :=
|
||
do env ← getEnv; pure (isAuxRecursor env constName || isNoConfusion env constName)
|
||
|
||
@[specialize] def unfoldDefinitionAux {α}
|
||
(e : Expr) (failK : MetaM α) (successK : Expr → MetaM α) : MetaM α :=
|
||
Lean.unfoldDefinitionAux getConstNoEx isAuxDef? whnf inferType isExprDefEq synthPending getLocalDecl
|
||
getExprMVarAssignment e (fun _ => failK) successK
|
||
|
||
partial def whnfImpl : Expr → MetaM Expr
|
||
| e => whnfEasyCases getLocalDecl getExprMVarAssignment e $ fun e => do
|
||
e ← whnfCore getConstNoEx isAuxDef? whnfImpl inferType isExprDefEqAux getLocalDecl getExprMVarAssignment e;
|
||
Lean.unfoldDefinitionAux getConstNoEx isAuxDef? whnf inferType isExprDefEq synthPending getLocalDecl
|
||
getExprMVarAssignment e (fun _ => pure e) whnfImpl
|
||
|
||
@[init] def setWHNFRef : IO Unit :=
|
||
whnfRef.set whnfImpl
|
||
|
||
end Meta
|
||
end Lean
|