From 27c104a4d3cd7e955c14a7c24577963f2d8fafa1 Mon Sep 17 00:00:00 2001
From: Avik Pal <avikpal@mit.edu>
Date: Thu, 14 Sep 2023 19:01:51 -0400
Subject: [PATCH] depthwise conv doesn't have GPU dispatches

---
 test/jvp.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/jvp.jl b/test/jvp.jl
index 5e6cf665..0f1e35f1 100644
--- 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