Skip conv2d subtest for Rocm6.3

ROCm · Sep 10, 2024 · 94b31ff · 94b31ff
1 parent d7d242f
commit 94b31ff
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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],