Update Tapir to use cached rules where possible (#464)

* Use get_tapir_interpreter * Bump patch * Revert version bump * Bump compat --------- Co-authored-by: Guillaume Dalle <22795598+gdalle@users.noreply.github.com>
JuliaDiff · Sep 12, 2024 · 41bade9 · 41bade9
1 parent 60e7815
commit 41bade9
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/DifferentiationInterface/Project.toml b/DifferentiationInterface/Project.toml
@@ -61,7 +61,7 @@ SparseArrays = "<0.0.1,1"
 SparseConnectivityTracer = "0.5.0,0.6"
 SparseMatrixColorings = "0.4.0"
 Symbolics = "5.27.1, 6"
-Tapir = "0.2.4"
+Tapir = "0.2.48"
 Tracker = "0.2.33"
 Zygote = "0.6.69"
 julia = "1.6"

diff --git a/...tiationInterface/ext/DifferentiationInterfaceTapirExt/DifferentiationInterfaceTapirExt.jl b/...tiationInterface/ext/DifferentiationInterfaceTapirExt/DifferentiationInterfaceTapirExt.jl
@@ -19,7 +19,7 @@ using Tapir:
     fdata,
     rdata,
     __value_and_pullback!!,
-    TapirInterpreter
+    get_tapir_interpreter
 
 DI.check_available(::AutoTapir) = true
 

diff --git a/DifferentiationInterface/ext/DifferentiationInterfaceTapirExt/onearg.jl b/DifferentiationInterface/ext/DifferentiationInterfaceTapirExt/onearg.jl
@@ -6,7 +6,7 @@ end
 function DI.prepare_pullback(f, backend::AutoTapir, x, ty::Tangents)
     y = f(x)
     rrule = build_rrule(
-        TapirInterpreter(),
+        get_tapir_interpreter(),
         Tuple{typeof(f),typeof(x)};
         safety_on=backend.safe_mode,
         silence_safety_messages=false,

diff --git a/DifferentiationInterface/ext/DifferentiationInterfaceTapirExt/twoarg.jl b/DifferentiationInterface/ext/DifferentiationInterfaceTapirExt/twoarg.jl
@@ -4,7 +4,7 @@ end
 
 function DI.prepare_pullback(f!, y, backend::AutoTapir, x, ty::Tangents)
     rrule = build_rrule(
-        TapirInterpreter(),
+        get_tapir_interpreter(),
         Tuple{typeof(f!),typeof(y),typeof(x)};
         safety_on=backend.safe_mode,
         silence_safety_messages=false,