Skip to content

Commit 836b1ba

Browse files
author
Michael Abbott
committed
a test without Zygote.Params
1 parent e40d0bd commit 836b1ba

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

test/runtests.jl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -155,5 +155,6 @@ end
155155
grad_slicemap = Zygote.gradient(() -> sum(slicemap(F(w), x, dims=1)), Zygote.Params([w,x]))
156156
@test grad_slicemap[x] gradx
157157
@test_broken grad_slicemap[w] gradw # wrong numbers
158+
@test gradw Zygote.gradient(w -> sum(slicemap(F(w), x, dims=1)), w)[1]
158159

159160
end

0 commit comments

Comments
 (0)