Skip to content

Commit c9e637b

Browse files
kshyattJutho
andauthored
Update ext/MatrixAlgebraKitEnzymeExt/MatrixAlgebraKitEnzymeExt.jl
Co-authored-by: Jutho <Jutho@users.noreply.github.com>
1 parent a87c115 commit c9e637b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

ext/MatrixAlgebraKitEnzymeExt/MatrixAlgebraKitEnzymeExt.jl

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -270,8 +270,9 @@ function EnzymeRules.reverse(
270270
) where {RT}
271271
cache_A, cache_USVᴴ, dUSVᴴ, ind = cache
272272
Aval = something(cache_A, A.val)
273+
USVᴴval = something(cache_USVᴴ, USVᴴ.val)
273274
if !isa(A, Const)
274-
svd_pullback!(A.dval, Aval, cache_USVᴴ, dUSVᴴ, ind)
275+
svd_pullback!(A.dval, Aval, USVᴴval, dUSVᴴ, ind)
275276
end
276277
!isa(USVᴴ, Const) && make_zero!(USVᴴ.dval)
277278
return (nothing, nothing, nothing)

0 commit comments

Comments
 (0)