Merge pull request #58 from SciML/reversediff

Add ReverseDiff weak dep

Author: ChrisRackauckas

Project.toml

@@ -7,13 +7,21 @@ version = "0.4.8"
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
 ArrayInterfaceCore = "30b0a656-2188-435a-8636-2ec0e6a096e2"
 ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
+Requires = "ae029012-a4dd-5104-9daa-d747884805df"
 
 [compat]
 Adapt = "3"
 ArrayInterfaceCore = "0.1.1"
 ForwardDiff = "0.10.3"
+Requires = "1.0"
 julia = "1.6"
 
+[weakdeps]
+ReverseDiff = "37e2e3b7-166d-5795-8a7a-e32c996b4267"
+
+[extensions]
+PreallocationToolsReverseDiffExt = "ReverseDiff"
+
 [extras]
 LabelledArrays = "2ee39098-c373-598a-b85f-a56591580800"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
@@ -23,6 +31,7 @@ OrdinaryDiffEq = "1dea7af3-3e70-54e6-95c3-0bf5283fa5ed"
 Pkg = "44cfe95a-1eb2-52ea-b672-e2afdf69b78f"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 RecursiveArrayTools = "731186ca-8d62-57ce-b412-fbd966d074cd"
+ReverseDiff = "37e2e3b7-166d-5795-8a7a-e32c996b4267"
 SafeTestsets = "1bc83da4-3b8d-516f-aca4-4fe02f6d838f"
 SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
 Symbolics = "0c5d862f-8b57-4792-8d23-62f2024744c7"

ext/PreallocationToolsReverseDiffExt.jl

@@ -0,0 +1,17 @@
+module PreallocationToolsReverseDiffExt
+
+using PreallocationTools
+import ReverseDiff
+
+# PreallocationTools https://github.com/SciML/PreallocationTools.jl/issues/39
+function Base.getindex(b::PreallocationTools.LazyBufferCache, u::ReverseDiff.TrackedArray)
+    s = b.sizemap(size(u)) # required buffer size
+    T = ReverseDiff.TrackedArray
+    buf = get!(b.bufs, (T, s)) do
+        # declare type since b.bufs dictionary is untyped
+        similar(u, s)
+    end
+    return buf
+end
+
+end

src/PreallocationTools.jl

@@ -214,4 +214,11 @@ end
 export GeneralLazyBufferCache, FixedSizeDiffCache, DiffCache, LazyBufferCache, dualcache
 export get_tmp
 
+import Requires
+@static if !isdefined(Base, :get_extension)
+    function __init__()
+            Requires.@require ReverseDiff="37e2e3b7-166d-5795-8a7a-e32c996b4267" begin include("../ext/PreallocationToolsReverseDiffExt.jl") end
+    end
+end
+
 end