depthwise conv doesn't have GPU dispatches

LuxDL · Sep 14, 2023 · 27c104a · 27c104a
1 parent 6a6b2dc
commit 27c104a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/jvp.jl b/test/jvp.jl
@@ -49,7 +49,7 @@ end
     @testset "$(op)(; flipped = $flipped)" for flipped in (true, false),
         op in (depthwiseconv, conv)
 
-        op === depthwiseconv && mode == "AMDGPU" && continue
+        op === depthwiseconv && on_gpu && continue
 
         input_dims = [(2, 4, 2, 1, 3), (4, 4, 1, 3), (4, 4, 3, 2), (4, 1, 3), (4, 3, 2)]
         weight_dims = if op === conv