Merge pull request #1281 from FluxML/bc/precompile

Add precompilation via SnoopPrecompile
FluxML · Oct 22, 2022 · d39ab59 · d39ab59
2 parents 1f07901 + 5c81bcf
commit d39ab59
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/Project.toml b/Project.toml
@@ -20,6 +20,7 @@ MacroTools = "1914dd2f-81c6-5fcd-8719-6d5c9610ff09"
 NaNMath = "77ba4419-2d1f-58cd-9bb1-8ffee604a2e3"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 Requires = "ae029012-a4dd-5104-9daa-d747884805df"
+SnoopPrecompile = "66db9d55-30c0-4569-8b51-7e840670fc0c"
 SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
 SpecialFunctions = "276daf66-3868-5448-9aa4-cd146d93841b"
 Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
@@ -41,6 +42,7 @@ MacroTools = "0.5"
 NaNMath = "0.3, 1"
 Requires = "1.1"
 SpecialFunctions = "1.6, 2"
+SnoopPrecompile = "1"
 ZygoteRules = "0.2.1"
 julia = "1.6"
 

diff --git a/src/Zygote.jl b/src/Zygote.jl
@@ -79,4 +79,7 @@ macro profile(ex)
   end
 end
 
+using SnoopPrecompile
+@precompile_all_calls precompile()
+
 end # module