Add using ChainRulesCore to BijectorsEnzymeExt

TuringLang · Oct 24, 2024 · 3e01d94 · 3e01d94
1 parent 9a19a37
commit 3e01d94
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ext/BijectorsEnzymeExt.jl b/ext/BijectorsEnzymeExt.jl
@@ -3,6 +3,7 @@ module BijectorsEnzymeExt
 if isdefined(Base, :get_extension)
     using Enzyme: @import_frule, @import_rrule
     using Bijectors: find_alpha
+    using ChainRulesCore
 else
     using ..Enzyme: @import_frule, @import_rrule
     using ..Bijectors: find_alpha