Skip to content

Commit 84c5196

Browse files
add some batch tests
1 parent 9273a20 commit 84c5196

File tree

1 file changed

+15
-1
lines changed

1 file changed

+15
-1
lines changed

test/enzyme.jl

Lines changed: 15 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,4 +27,18 @@ db12 = FiniteDiff.finite_difference_gradient(x->f(A,x, b2), copy(b1))
2727

2828
@test dA dA2
2929
@test db1 db12
30-
@test db2 == zeros(4)
30+
@test db2 == zeros(4)
31+
32+
A = rand(n, n);
33+
dA = zeros(n, n);
34+
dA2 = zeros(n, n);
35+
b1 = rand(n);
36+
db1 = zeros(n);
37+
db12 = zeros(n);
38+
39+
b2 = rand(n);
40+
db2 = zeros(n);
41+
db22 = zeros(n);
42+
43+
@test_broken Enzyme.autodiff(Reverse, f, Duplicated(copy(A), dA), BatchDuplicated(copy(b1), (db1, db12)), BatchDuplicated(copy(b2), (db2, db22)))
44+
@test_broken Enzyme.autodiff(Reverse, f, BatchDuplicated(copy(A), (dA, dA2)), Duplicated(copy(b1), db1), Duplicated(copy(b2), db2))

0 commit comments

Comments
 (0)