From 732aa661223c969327b96e5c1d0b073bcc958673 Mon Sep 17 00:00:00 2001 From: Harsha HS Date: Tue, 10 Sep 2024 09:19:39 -0500 Subject: [PATCH] Skip conv2d tests on Rocm6.3 --- tensorflow/compiler/tests/conv2d_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/compiler/tests/conv2d_test.py b/tensorflow/compiler/tests/conv2d_test.py index 2afd996ed49f6b..082e15c2c36004 100644 --- a/tensorflow/compiler/tests/conv2d_test.py +++ b/tensorflow/compiler/tests/conv2d_test.py @@ -544,7 +544,7 @@ def testConv2D2x2FilterStride2Same(self, data_format): def testConv2D2x2Depth3ValidBackpropInputStride1x1Dilation2x1( self, data_format): if test.is_built_with_rocm(): - self.skipTest('only ROCm 6.2 will skip this subtest') + self.skipTest('ROCm 6.2/6.3 will skip this subtest') self._VerifyValues( input_sizes=[1, 3, 6, 1], filter_sizes=[2, 2, 1, 1], @@ -604,7 +604,7 @@ def testConv2D2x2Depth3ValidBackpropInputDilation2x1(self, data_format): def testConv2DKernelSizeMatchesInputSizeBackpropInputDilation2x2( self, data_format): if test.is_built_with_rocm(): - self.skipTest('only ROCm 6.2 will skip this subtest') + self.skipTest('only ROCm 6.2 will skip this subtest') self._VerifyValues( input_sizes=[1, 3, 3, 1], filter_sizes=[2, 2, 1, 2],