fixed nhwc cudnn depthwise conv

yangulei · Dec 11, 2021 · 93cd039 · 93cd039
1 parent 6db7172
commit 93cd039
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/python/tvm/relay/op/strategy/cuda.py b/python/tvm/relay/op/strategy/cuda.py
@@ -324,7 +324,10 @@ def conv2d_strategy_cuda(attrs, inputs, out_type, target):
                 plevel=25,
             )
 
-    elif is_depthwise_conv2d(data.shape, layout, kernel.shape, kernel_layout, groups):
+    elif (
+        is_depthwise_conv2d(data.shape, layout, kernel.shape, kernel_layout, groups)
+        and "cudnn" not in target.libs
+    ):
         if layout == "NCHW":
             assert kernel_layout == "OIHW"
             strategy.add_implementation(