Specialize capture to work with StaticArrays

JuliaDiff · Oct 14, 2020 · 622e50f · 622e50f
1 parent fd62c7c
commit 622e50f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/tracked.jl b/src/tracked.jl
@@ -223,6 +223,8 @@ unseed!(x::AbstractArray, i) = unseed!(x[i])
 capture(t::TrackedReal) = ifelse(hastape(t), t, value(t))
 capture(t::TrackedArray) = t
 capture(t::AbstractArray) = istracked(t) ?  map!(capture, similar(t), t) : copy(t)
+# `StaticArray`s don't support mutation unless the eltype is a bits type (`isbitstype`).
+capture(t::SA) where SA <: StaticArray = istracked(t) ? SA(map(capture, t)) : copy(t)
 
 ########################
 # Conversion/Promotion #