perf: no need to register axioms outside of the module system (#8121)

src/Lean/AddDecl.lean

@@ -148,8 +148,12 @@ where
           let env ← (← getEnv).addDeclAux (← getOptions) decl (← read).cancelTk?
             |> ofExceptKernelException
           setEnv env
-          for n in decl.getTopLevelNames do
-            registerAxiomsForDecl n
+          if env.header.isModule then
+            -- Under the module system, must record axioms in environment extension now when we
+            -- still have access to the declaration's body. We could skip this for non-opaque decls
+            -- but for them, registering simply acts as a cache.
+            for n in decl.getTopLevelNames do
+              registerAxiomsForDecl n
         catch ex =>
           -- avoid follow-up errors by (trying to) add broken decl as axiom
           addAsAxiom

src/Lean/Util/CollectAxioms.lean

@@ -51,6 +51,10 @@ def collectAxioms [Monad m] [MonadEnv m] (constName : Name) : m (Array Name) :=
   let (_, s) := ((CollectAxioms.collect constName).run env).run {}
   pure s.axioms
 
+/--
+Registers axioms used by the declaration in an environment extension so they can be retrieved even
+if some parts of the declaration's become inaccessible.
+-/
 def registerAxiomsForDecl (n : Name) : CoreM Unit := do
   let axioms ← collectAxioms n
   modifyEnv (exportedAxiomsExt.addEntry · (n, axioms))