Pick up a PreallocationTools.jl dependency

This drops LabelledArrays.jl out of the DiffEq dependency tree.

Author: ChrisRackauckas

Project.toml

@@ -10,12 +10,14 @@ ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 MacroTools = "1914dd2f-81c6-5fcd-8719-6d5c9610ff09"
 StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
+PreallocationTools = "d236fae5-4411-538c-8e31-a6e3d9e00b46"
 
 [compat]
 ArrayInterface = "2.8, 3.0, 4, 5, 6"
 ArrayInterfaceStaticArrays = "0.1"
 ChainRulesCore = "1"
 MacroTools = "0.5"
+PreallocationTools = "0.4"
 StaticArrays = "0.10, 0.11, 0.12, 1.0"
 julia = "1.6"
 

src/LabelledArrays.jl

@@ -63,6 +63,16 @@ function ArrayInterface.ismutable(::Type{<:LArray{T, N, Syms}}) where {T, N, Sym
 end
 ArrayInterface.can_setindex(::Type{<:SLArray}) = false
 
+function PreallocationTools.get_tmp(dc::PreallocationTools.DiffCache,
+                 u::LArray{T, N, D, Syms}) where {T, N, D, Syms}
+                 nelem = div(sizeof(T), sizeof(eltype(dc.dual_du))) * length(dc.du)
+    if nelem > length(dc.dual_du)
+        PreallocationTools.enlargedualcache!(dc, nelem)
+    end
+    _x = ArrayInterfaceCore.restructure(dc.du, reinterpret(T, view(dc.dual_du, 1:nelem)))
+    LabelledArrays.LArray{T, N, D, Syms}(_x)
+end
+
 export SLArray, LArray, SLVector, LVector, @SLVector, @LArray, @LVector, @SLArray
 
 export @SLSliced, @LSliced