fix test

EnzymeAD · Jun 7, 2024 · 5f0ca77 · 5f0ca77
1 parent 8bee8b1
commit 5f0ca77
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -1670,9 +1670,12 @@ end
     dout2 = Ref(3.0)
     dout3 = Ref(5.0)
     Enzyme.autodiff(Reverse, batchgf, Const, BatchDuplicatedNoNeed(out, (dout, dout2, dout3)), BatchDuplicated(x, (dx, dx2, dx3)))
-    @test dx[1] ≈ (4.0, 6.0)
-    @test dx2[1] ≈ (3*4.0, 3*6.0)
-    @test dx3[1] ≈ (5*4.0, 5*6.0)
+    @test dx[1][1] ≈ 1.0
+    @test dx[1][2] ≈ 0.0
+    @test dx2[1][1] ≈ 3.0
+    @test dx2[1][2] ≈ 0.0
+    @test dx3[1][1] ≈ 5.0
+    @test dx2[1][2] ≈ 0.0
 end
 
 include("applyiter.jl")