Skip some tests on NHWC layout when cudnn version is old
@@ -184,8 +184,11 @@ class TestConvolutionForward(unittest.TestCase):
def test_call(self):
if self.layout == libcudnn.CUDNN_TENSOR_NHWC:
+ version = libcudnn.getVersion()
if self.groups > 1:
return unittest.SkipTest()
+ if self.dilate > 1 and version < 7300:
+ return unittest.SkipTest()
if self.err is None:
self.call()
self.assertTrue((self.y == 0).all())
