Adapt SnoopPrecompile to work for new julia API.

Starting in JuliaLang/julia#47615, we are changing the API for snoopi_deep, to allow for thread safe snooping in one thread while compiling in another. This commit changes SnoopPrecompile to work with the new API if it's available. Co-Authored-By: Pete Vilter <[email protected]>
timholy · Nov 22, 2022 · 40a9f6c · 40a9f6c
1 parent ad792af
commit 40a9f6c
Showing 1 changed file with 23 additions and 9 deletions.
diff --git a/SnoopPrecompile/src/SnoopPrecompile.jl b/SnoopPrecompile/src/SnoopPrecompile.jl
@@ -5,6 +5,7 @@ export @precompile_all_calls, @precompile_setup
 const verbose = Ref(false)    # if true, prints all the precompiles
 const have_inference_tracking = isdefined(Core.Compiler, :__set_measure_typeinf)
 const have_force_compile = isdefined(Base, :Experimental) && isdefined(Base.Experimental, Symbol("#@force_compile"))
+const have_clear_and_fetch_timings = isdefined(Core.Compiler.Timings, :clear_and_fetch_timings)
 
 function precompile_roots(roots)
     @assert have_inference_tracking
@@ -59,16 +60,29 @@ macro precompile_all_calls(ex::Expr)
         end
     end
     if have_inference_tracking
-        ex = quote
-            Core.Compiler.Timings.reset_timings()
-            Core.Compiler.__set_measure_typeinf(true)
-            try
-                $ex
-            finally
-                Core.Compiler.__set_measure_typeinf(false)
-                Core.Compiler.Timings.close_current_timer()
+        if have_clear_and_fetch_timings
+            # use new thread-safe timings API if it's available in this version of Julia
+            ex = quote
+                Core.Compiler.__set_measure_typeinf(true)
+                try
+                    $ex
+                finally
+                    Core.Compiler.__set_measure_typeinf(false)
+                end
+                $SnoopPrecompile.precompile_roots(Core.Compiler.Timings.clear_and_fetch_timings())
+            end
+        else
+            ex = quote
+                Core.Compiler.Timings.reset_timings()
+                Core.Compiler.__set_measure_typeinf(true)
+                try
+                    $ex
+                finally
+                    Core.Compiler.__set_measure_typeinf(false)
+                    Core.Compiler.Timings.close_current_timer()
+                end
+                $SnoopPrecompile.precompile_roots(Core.Compiler.Timings._timings[1].children)
             end
-            $SnoopPrecompile.precompile_roots(Core.Compiler.Timings._timings[1].children)
         end
     end
     return esc(quote