diff --git a/src/rulesets/LinearAlgebra/factorization.jl b/src/rulesets/LinearAlgebra/factorization.jl index e082d56e9..84caab6ea 100644 --- a/src/rulesets/LinearAlgebra/factorization.jl +++ b/src/rulesets/LinearAlgebra/factorization.jl @@ -7,7 +7,7 @@ using LinearAlgebra.BLAS: gemv, gemv!, gemm!, trsm!, axpy!, ger! function rrule(::typeof(svd), X::AbstractMatrix{<:Real}) F = svd(X) - function svd_pullback(Ȳ::Composite}) + function svd_pullback(Ȳ::Composite) ∂X = @thunk(svd_rev(F, Ȳ.U, Ȳ.S, Ȳ.V)) return (NO_FIELDS, ∂X) end