Revert "Remove rule"

This reverts commit 5b3cc7d.
JuliaGaussianProcesses · Jan 10, 2024 · afb213e · afb213e
1 parent cc55ec9
commit afb213e
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/util/chainrules.jl b/src/util/chainrules.jl
@@ -28,6 +28,13 @@ Zygote.accum(a::Tuple, b::Tuple, c::Tuple) = map(Zygote.accum, a, b, c)
 #                                 StaticArrays                                 #
 # ---------------------------------------------------------------------------- #
 
+function rrule(::Type{T}, x::Tuple) where {T<:SArray}
+    SArray_rrule(Δ) = begin
+        (NoTangent(), Tangent{typeof(x)}(unthunk(Δ).data...))
+    end
+    return T(x), SArray_rrule
+end
+
 function rrule(::RuleConfig{>:HasReverseMode}, ::Type{SArray{S, T, N, L}}, x::NTuple{L, T}) where {S, T, N, L}
     SArray_rrule(::AbstractZero) = NoTangent(), NoTangent()
     SArray_rrule(Δ::NamedTuple{(:data,)}) = NoTangent(), Δ.data