Inline expanded frule and rrule in BijectorsEnzymeExt

TuringLang · Oct 28, 2024 · 5aa04c0 · 5aa04c0
1 parent 18b3b1c
commit 5aa04c0
Show file tree

Hide file tree

Showing 2 changed files with 596 additions and 9 deletions.
diff --git a/Project.toml b/Project.toml
@@ -26,21 +26,22 @@ Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
 [weakdeps]
 DistributionsAD = "ced4e74d-a319-5a8a-b0ac-84af2272839c"
 Enzyme = "7da242da-08ed-463a-9acd-ee780be4f1d9"
+EnzymeCore = "f151be2c-9106-41f4-ab19-57ee4f262869"
 ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
 LazyArrays = "5078a376-72f3-5289-bfd5-ec5146d43c02"
 ReverseDiff = "37e2e3b7-166d-5795-8a7a-e32c996b4267"
-Tracker = "9f7883ad-71c0-57eb-9f7f-b5c9e6d3789c"
 Tapir = "07d77754-e150-4737-8c94-cd238a1fb45b"
+Tracker = "9f7883ad-71c0-57eb-9f7f-b5c9e6d3789c"
 Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 
 [extensions]
 BijectorsDistributionsADExt = "DistributionsAD"
-BijectorsEnzymeExt = "Enzyme"
+BijectorsEnzymeExt = ["Enzyme", "EnzymeCore", "ChainRulesCore"]
 BijectorsForwardDiffExt = "ForwardDiff"
 BijectorsLazyArraysExt = "LazyArrays"
 BijectorsReverseDiffExt = "ReverseDiff"
-BijectorsTrackerExt = "Tracker"
 BijectorsTapirExt = "Tapir"
+BijectorsTrackerExt = "Tracker"
 BijectorsZygoteExt = "Zygote"
 
 [compat]
@@ -53,6 +54,7 @@ Distributions = "0.25.33"
 DistributionsAD = "0.6"
 DocStringExtensions = "0.9"
 Enzyme = "0.12.22"
+EnzymeCore = "0.7.8"
 ForwardDiff = "0.10"
 Functors = "0.1, 0.2, 0.3, 0.4"
 InverseFunctions = "0.1"