make Infiltrator available within Core.Compiler.EscapeAnalysis

JuliaDebug · Oct 4, 2024 · b5a4d58 · b5a4d58
1 parent db31af9
commit b5a4d58
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/Infiltrator.jl b/src/Infiltrator.jl
@@ -62,6 +62,10 @@ 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)
+    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))