Skip to content

Commit 69af5c4

Browse files
committed
Fix svd_vals
1 parent 422c34a commit 69af5c4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ext/MatrixAlgebraKitEnzymeExt/MatrixAlgebraKitEnzymeExt.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -409,7 +409,7 @@ function EnzymeRules.augmented_primal(
409409
cache_S = nothing
410410
cache_A = copy(A.val)
411411
U, nS, Vᴴ = svd_compact!(A.val, alg.val)
412-
copy!(S, diagview(nS))
412+
copy!(S.val, diagview(nS))
413413
primal = EnzymeRules.needs_primal(config) ? S.val : nothing
414414
shadow = EnzymeRules.needs_shadow(config) ? S.dval : nothing
415415
return EnzymeRules.AugmentedReturn(primal, shadow, (cache_A, cache_S, U, Vᴴ))

0 commit comments

Comments
 (0)