Skip to content

Commit 0387838

Browse files
Tracker oops
1 parent fb2cacb commit 0387838

File tree

3 files changed

+10
-7
lines changed

3 files changed

+10
-7
lines changed

Project.toml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,10 +22,10 @@ Tables = "bd369af6-aec1-5ad0-b16a-f7cc5008161c"
2222
ZygoteRules = "700de1a5-db45-46bc-99cf-38207098b444"
2323

2424
[weakdeps]
25-
ReverseDiff = "37e2e3b7-166d-5795-8a7a-e32c996b4267"
25+
Tracker = "9f7883ad-71c0-57eb-9f7f-b5c9e6d3789c"
2626

2727
[extensions]
28-
RecursiveArrayToolsReverseDiffExt = "ReverseDiff"
28+
RecursiveArrayToolsTrackerExt = "Tracker"
2929

3030
[compat]
3131
Adapt = "3"
Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,7 @@
1-
module RecursiveArrayToolsReverseDiffExt
1+
module RecursiveArrayToolsTrackerExt
2+
3+
import RecursiveArrayTools
4+
import Tracker
25

36
function RecursiveArrayTools.recursivecopy!(b::AbstractArray{T, N},
47
a::AbstractArray{T2, N}) where {

src/RecursiveArrayTools.jl

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -42,10 +42,10 @@ function ChainRulesCore.rrule(T::Type{<:GPUArraysCore.AbstractGPUArray},
4242
T(xs), ȳ -> (NoTangent(), ȳ)
4343
end
4444

45-
import Requires
46-
function __init__()
47-
@static if !isdefined(Base, :get_extension)
48-
Requires.@require ReverseDiff="37e2e3b7-166d-5795-8a7a-e32c996b4267" begin include("../ext/RecursiveArrayToolsReverseDiffExt.jl") end
45+
@static if !isdefined(Base, :get_extension)
46+
import Requires
47+
function __init__()
48+
Requires.@require Tracker="9f7883ad-71c0-57eb-9f7f-b5c9e6d3789c" begin include("../ext/RecursiveArrayToolsTrackerExt.jl") end
4949
end
5050
end
5151

0 commit comments

Comments
 (0)