Missing unthunk

Author: kshyatt

ext/TensorKitChainRulesCoreExt/linalg.jl

@@ -80,7 +80,7 @@ function ChainRulesCore.rrule(
 end
 
 function ChainRulesCore.rrule(::typeof(tr), A::AbstractTensorMap)
-    tr_pullback(Δtr) = NoTangent(), scale!!(id!(similar(A)), Δtr)
+    tr_pullback(Δtr) = NoTangent(), scale!!(id!(similar(A)), unthunk(Δtr))
     return tr(A), tr_pullback
 end