next attempt at proper fix

Jutho · Nov 17, 2023 · b7de300 · b7de300
1 parent 21e8f6f
commit b7de300
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ext/TensorKitChainRulesCoreExt.jl b/ext/TensorKitChainRulesCoreExt.jl
@@ -166,7 +166,7 @@ function ChainRulesCore.rrule(::typeof(TensorKit.tsvd!), t::AbstractTensorMap;
             Uc, Σc, Vc = block(U, c), block(Σ, c), block(V, c)
             ΔUc, ΔΣc, ΔVc = block(ΔU, c), block(ΔΣ, c), block(ΔV, c)
             Σdc = view(Σc, diagind(Σc))
-            ΔΣdc = (ΔΣdc isa AbstractZero) ? ΔΣdc : view(ΔΣdc, diagind(ΔΣdc))
+            ΔΣdc = (ΔΣc isa AbstractZero) ? ΔΣc : view(ΔΣc, diagind(ΔΣc))
             copyto!(b, svd_pullback(Uc, Σdc, Vc, ΔUc, ΔΣdc, ΔVc))
         end
         return NoTangent(), Δt