Merge pull request #128 from aviatesk/avi/EA

make Infiltrator available within `Core.Compiler.EscapeAnalysis`
JuliaDebug · Jan 7, 2025 · f0ae6b9 · f0ae6b9
2 parents db31af9 + 3f0f6bb
commit f0ae6b9
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/Infiltrator.jl b/src/Infiltrator.jl
@@ -62,6 +62,14 @@ macro infiltrate(cond = true)
     # XXX Dict isn't available in Core.Compiler, so make it available now
     Core.eval(Core.Compiler, :(using .Main: Dict))
     Core.eval(Core.Compiler, :(setindex!(x::Dict, args...) = Main.setindex!(x, args...)))
+  elseif @static isdefined(Core.Compiler, :EscapeAnalysis) &&
+     __module__ === Core.Compiler.EscapeAnalysis && !isdefined(Core.Compiler.EscapeAnalysis, :Dict)
+    if !isdefined(Core.Compiler, :Dict)
+      Core.eval(Core.Compiler, :(using .Main: Dict))
+      Core.eval(Core.Compiler, :(setindex!(x::Dict, args...) = Main.setindex!(x, args...)))
+    end
+    Core.eval(Core.Compiler.EscapeAnalysis, :(using .Main: Dict))
+    Core.eval(Core.Compiler.EscapeAnalysis, :(setindex!(x::Dict, args...) = Main.setindex!(x, args...)))
   end
   quote
     if $(esc(cond))