max/lean4-htt
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 2

feat: add getSpecializationArgs?

Browse source
This commit is contained in:
Leonardo de Moura 2022-09-07 16:08:45 -07:00
parent de0be1d820
commit 19f5fe6f42
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/Lean/Compiler/Specialize.lean
View file

@ -51,13 +51,15 @@ builtin_initialize specializeAttr : ParametricAttribute (Array Nat) ←
partial def hasSpecializeAttribute (env : Environment) (n : Name) : Bool :=
match specializeAttr.getParam? env n with
| some _ => true
| none => if n.isInternal then hasSpecializeAttribute env n.getPrefix else false
| none => if n.isInternal then hasSpecializeAttribute env n.getPrefix else false -- TODO: remove recursion after we move to new compiler
@[export lean_has_nospecialize_attribute]
partial def hasNospecializeAttribute (env : Environment) (n : Name) : Bool :=
nospecializeAttr.hasTag env n ||
(n.isInternal && hasNospecializeAttribute env n.getPrefix)
(n.isInternal && hasNospecializeAttribute env n.getPrefix) -- TODO: remove recursion after we move to new compiler
def getSpecializationArgs? (env : Environment) (n : Name) : Option (Array Nat) :=
specializeAttr.getParam? env n
/- TODO: the rest of the file is for the old / current code generator. We should remove it as soon as we move to the new one. -/