Merge pull request #1281 from FluxML/bc/precompile

Add precompilation via SnoopPrecompile

Author: ToucheSir

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"
 

src/Zygote.jl

@@ -79,4 +79,7 @@ macro profile(ex)
   end
 end
 
+using SnoopPrecompile
+@precompile_all_calls precompile()
+
 end # module