Skip to content

Commit 41bade9

Browse files
willtebbuttgdalle
andauthored
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>
1 parent 60e7815 commit 41bade9

4 files changed

Lines changed: 4 additions & 4 deletions

File tree

DifferentiationInterface/Project.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ SparseArrays = "<0.0.1,1"
6161
SparseConnectivityTracer = "0.5.0,0.6"
6262
SparseMatrixColorings = "0.4.0"
6363
Symbolics = "5.27.1, 6"
64-
Tapir = "0.2.4"
64+
Tapir = "0.2.48"
6565
Tracker = "0.2.33"
6666
Zygote = "0.6.69"
6767
julia = "1.6"

DifferentiationInterface/ext/DifferentiationInterfaceTapirExt/DifferentiationInterfaceTapirExt.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ using Tapir:
1919
fdata,
2020
rdata,
2121
__value_and_pullback!!,
22-
TapirInterpreter
22+
get_tapir_interpreter
2323

2424
DI.check_available(::AutoTapir) = true
2525

DifferentiationInterface/ext/DifferentiationInterfaceTapirExt/onearg.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ end
66
function DI.prepare_pullback(f, backend::AutoTapir, x, ty::Tangents)
77
y = f(x)
88
rrule = build_rrule(
9-
TapirInterpreter(),
9+
get_tapir_interpreter(),
1010
Tuple{typeof(f),typeof(x)};
1111
safety_on=backend.safe_mode,
1212
silence_safety_messages=false,

DifferentiationInterface/ext/DifferentiationInterfaceTapirExt/twoarg.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ end
44

55
function DI.prepare_pullback(f!, y, backend::AutoTapir, x, ty::Tangents)
66
rrule = build_rrule(
7-
TapirInterpreter(),
7+
get_tapir_interpreter(),
88
Tuple{typeof(f!),typeof(y),typeof(x)};
99
safety_on=backend.safe_mode,
1010
silence_safety_messages=false,

0 commit comments

Comments
 (0)