Skip to content

Commit d0a03f2

Browse files
unbreak gpu conv test (#541)
* unbreak gpu conv test * cleanup
1 parent 8da76bd commit d0a03f2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/ext_cuda/conv.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,7 @@ using NNlib: DenseConvDims
101101
@testset "scale-beta" begin
102102
gputest((y, x, w) -> act.(NNlib.conv!(copy(y), x, w, cdims; beta=T(2.0))), y, x, w, checkgrad=false, broken=false)
103103
gputest((w, x, y) -> act.(NNlib.∇conv_filter!(copy(w), x, y, cdims; beta=T(2.0))), w, x, y, checkgrad=false, broken=false)
104-
gputest((x, y, w) -> act.(NNlib.∇conv_data!(copy(x), y, w, cdims; beta=T(2.0))), x, y, w, checkgrad=false, broken=true)
104+
gputest((x, y, w) -> act.(NNlib.∇conv_data!(copy(x), y, w, cdims; beta=T(2.0))), x, y, w, checkgrad=false, broken=false)
105105

106106
if T <: Complex
107107
gputest((y, x, w) -> act.(NNlib.conv!(copy(y), x, w, cdims; beta=T(2.0))), y, real(x), w, checkgrad=false)

0 commit comments

Comments
 (0)