fixup! fixup! fixup! fixup! make_zero! for immutable

EnzymeAD · Oct 10, 2024 · 0a4f468 · 0a4f468
1 parent a3def6c
commit 0a4f468
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/src/make_zero.jl b/src/make_zero.jl
@@ -387,7 +387,8 @@ end
     # For make_zero!(NamedTuple) we want to recurse and zero out
     # the storage
     if !Base.ismutabletype(T)
-        return make_zero_immutable!(prev, seen)
+        make_zero_immutable!(prev, seen)
+        return nothing
     end
 
     for i = 1:nf

diff --git a/test/abi.jl b/test/abi.jl
@@ -503,11 +503,9 @@ end
     rand!(dparams.q)
     rand!(dparams.p)
 
-    dparams2 = make_zero!(dparams)
+    make_zero!(dparams)
     @test all(==(0), dparams.q)
     @test all(==(0), dparams.p)
-
-    @test dparams2 == dparams
 end