[UnitTests] pytest.xfail for CuDNN conv2d with asymmetric padding

apache · Aug 3, 2021 · 93a9535 · 93a9535
1 parent 60e3b85
commit 93a9535
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/python/topi/python/test_topi_conv2d_nchw.py b/tests/python/topi/python/test_topi_conv2d_nchw.py
@@ -104,6 +104,10 @@ def test_conv2d_nchw(
         pad_top, pad_left, pad_bottom, pad_right = get_pad_tuple(padding, (kernel, kernel))
         padding_sum = pad_top + pad_left + pad_bottom + pad_right
 
+        has_asymmetric_padding = (pad_top != pad_bottom) or (pad_left != pad_right)
+        if is_cudnn_target and has_asymmetric_padding:
+            pytest.xfail("CuDNN does not support asymmetric padding")
+
         a_np, w_np, b_np, c_np = ref_data
 
         A = te.placeholder(a_np.shape, name="A", dtype=dtype)