Skip to content

Commit 5a48e78

Browse files
committed
Add unthunk in tsvd!_pullback
1 parent 4ef17d5 commit 5a48e78

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

ext/TensorKitChainRulesCoreExt/factorizations.jl

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,8 @@ function ChainRulesCore.rrule(::typeof(TensorKit.tsvd!), t::AbstractTensorMap;
2020
Ũ, Σ̃, Ṽ⁺ = U, Σ, V⁺
2121
end
2222

23-
function tsvd!_pullback((ΔU, ΔΣ, ΔV⁺, Δϵ))
23+
function tsvd!_pullback(ΔUSVϵ)
24+
ΔU, ΔΣ, ΔV⁺, = unthunk.(ΔUSVϵ)
2425
Δt = similar(t)
2526
for (c, b) in blocks(Δt)
2627
Uc, Σc, V⁺c = block(U, c), block(Σ, c), block(V⁺, c)

0 commit comments

Comments
 (0)