unbreak gpu conv test (#541)

* unbreak gpu conv test * cleanup
FluxML · Sep 27, 2023 · d0a03f2 · d0a03f2
1 parent 8da76bd
commit d0a03f2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/ext_cuda/conv.jl b/test/ext_cuda/conv.jl
@@ -101,7 +101,7 @@ using NNlib: DenseConvDims
             @testset "scale-beta" begin
                 gputest((y, x, w) -> act.(NNlib.conv!(copy(y), x, w, cdims; beta=T(2.0))), y, x, w, checkgrad=false, broken=false)
                 gputest((w, x, y) -> act.(NNlib.∇conv_filter!(copy(w), x, y, cdims; beta=T(2.0))), w, x, y, checkgrad=false, broken=false) 
-                gputest((x, y, w) -> act.(NNlib.∇conv_data!(copy(x), y, w, cdims; beta=T(2.0))), x, y, w, checkgrad=false, broken=true) 
+                gputest((x, y, w) -> act.(NNlib.∇conv_data!(copy(x), y, w, cdims; beta=T(2.0))), x, y, w, checkgrad=false, broken=false) 
 
                 if T <: Complex
                     gputest((y, x, w) -> act.(NNlib.conv!(copy(y), x, w, cdims; beta=T(2.0))), y, real(x), w, checkgrad=false)