Skip to content

Commit

Permalink
fix test
Browse files Browse the repository at this point in the history
  • Loading branch information
wsmoses committed Jun 7, 2024
1 parent 8bee8b1 commit 5f0ca77
Showing 1 changed file with 6 additions and 3 deletions.
9 changes: 6 additions & 3 deletions test/runtests.jl
Original file line number Diff line number Diff line change
@@ -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")

0 comments on commit 5f0ca77

Please sign in to comment.