Gate BijectorsEnzymeExt precompilation

TuringLang · Oct 23, 2024 · 75e3799 · 75e3799
1 parent 9a19a37
commit 75e3799
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/ext/BijectorsEnzymeExt.jl b/ext/BijectorsEnzymeExt.jl
@@ -8,7 +8,21 @@ else
     using ..Bijectors: find_alpha
 end
 
-@import_rrule typeof(find_alpha) Real Real Real
-@import_frule typeof(find_alpha) Real Real Real
+# Julia 1.11.1 caused a change in the ordering of precompilation for extensions.
+# https://github.com/TuringLang/Bijectors.jl/issues/332
+# See https://github.com/JuliaLang/julia/issues/56204
+@static if v"1.11.1" <= VERSION < v"1.12"
+    function __init__()
+        if !Base.generating_output()
+            Base.eval(quote
+                @import_rrule typeof(find_alpha) Real Real Real
+                @import_frule typeof(find_alpha) Real Real Real
+            end)
+        end
+    end
+else
+    @import_rrule typeof(find_alpha) Real Real Real
+    @import_frule typeof(find_alpha) Real Real Real
+end
 
 end