Enable more gpu_pycpp tests

Author: hsharsha

tensorflow/compiler/tests/scatter_nd_op_test.py

@@ -149,8 +149,6 @@ def testSimple3(self):
     expected = np.array([[0., 0.], [11., 12.], [0., 0.]])
     self.assertAllEqual(expected, self._runScatterNd(indices, updates, [3, 2]))
 
-  @test.disable_with_predicate(
-      pred=test.is_built_with_rocm, skip_message="Test fails on ROCm.") #TODO(rocm): weekly sync 25-05-14
   def testVariableRankUpdate(self):
     self._VariableRankTests(_NumpyUpdate, self._runScatterNd)
 

tensorflow/compiler/tests/segment_reduction_ops_test.py

@@ -78,8 +78,6 @@ def _unsortedSegmentMax(self, data, indices, num_segments):
     return self._segmentReduction(math_ops.unsorted_segment_max, data, indices,
                                   num_segments)
 
-  @test.disable_with_predicate(
-      pred=test.is_built_with_rocm, skip_message="Test fails on ROCm.") #TODO(rocm): weekly sync 25-05-14
   def testSegmentSum(self):
     for dtype in self.numeric_types:
       self.assertAllClose(
@@ -88,8 +86,6 @@ def testSegmentSum(self):
               np.array([0, 1, 2, 3, 4, 5], dtype=dtype),
               np.array([0, 0, 2, 3, 3, 3], dtype=np.int32), 4))
 
-  @test.disable_with_predicate(
-      pred=test.is_built_with_rocm, skip_message="Test fails on ROCm.") #TODO(rocm): weekly sync 24-11-05
   def testSegmentProd(self):
     for dtype in self.numeric_types:
       self.assertAllClose(
@@ -98,8 +94,6 @@ def testSegmentProd(self):
               np.array([0, 1, 2, 3, 4, 5], dtype=dtype),
               np.array([0, 0, 2, 3, 3, 3], dtype=np.int32), 4))
 
-  @test.disable_with_predicate(
-      pred=test.is_built_with_rocm, skip_message="Test fails on ROCm.") #TODO(rocm): weekly sync 24-11-05
   def testSegmentProdNumSegmentsLess(self):
     for dtype in self.numeric_types:
       self.assertAllClose(
@@ -108,8 +102,6 @@ def testSegmentProdNumSegmentsLess(self):
               np.array([0, 1, 2, 3, 4, 5], dtype=dtype),
               np.array([0, 0, 2, 3, 3, 3], dtype=np.int32), 3))
 
-  @test.disable_with_predicate(
-      pred=test.is_built_with_rocm, skip_message="Test fails on ROCm.") #TODO(rocm): weekly sync 24-11-05
   def testSegmentProdNumSegmentsMore(self):
     for dtype in self.numeric_types:
       self.assertAllClose(
@@ -194,8 +186,6 @@ def testUnsortedSegmentSum0DIndices1DData(self):
           self._unsortedSegmentSum(
               np.array([0, 1, 2, 3, 4, 5], dtype=dtype), 2, 4))
 
-  @test.disable_with_predicate(
-      pred=test.is_built_with_rocm, skip_message="Test fails on ROCm.") #TODO(rocm): weekly sync 25-05-14
   def testUnsortedSegmentSum1DIndices1DData(self):
     for dtype in self.numeric_types:
       self.assertAllClose(
@@ -204,8 +194,6 @@ def testUnsortedSegmentSum1DIndices1DData(self):
               np.array([0, 1, 2, 3, 4, 5], dtype=dtype),
               np.array([3, 0, 2, 1, 3, 3], dtype=np.int32), 4))
 
-  @test.disable_with_predicate(
-      pred=test.is_built_with_rocm, skip_message="Test fails on ROCm.") #TODO(rocm): weekly sync 25-05-14
   def testUnsortedSegmentSum1DIndices1DDataNegativeIndices(self):
     for dtype in self.numeric_types:
       self.assertAllClose(
@@ -214,8 +202,6 @@ def testUnsortedSegmentSum1DIndices1DDataNegativeIndices(self):
               np.array([0, 1, 2, 3, 4, 5, 6], dtype=dtype),
               np.array([3, -1, 0, 1, 0, -1, 3], dtype=np.int32), 4))
 
-  @test.disable_with_predicate(
-      pred=test.is_built_with_rocm, skip_message="Test fails on ROCm.") #TODO(rocm): weekly sync 25-05-14
   def testUnsortedSegmentSum1DIndices2DDataDisjoint(self):
     for dtype in self.numeric_types:
       data = np.array(
@@ -232,8 +218,6 @@ def testUnsortedSegmentSum1DIndices2DDataDisjoint(self):
                [50, 51, 52, 53], [0, 1, 2, 3], [0, 0, 0, 0]],
               dtype=dtype), y)
 
-  @test.disable_with_predicate(
-      pred=test.is_built_with_rocm, skip_message="Test fails on ROCm.") #TODO(rocm): weekly sync 25-05-14
   def testUnsortedSegmentSum1DIndices2DDataNonDisjoint(self):
     for dtype in self.numeric_types:
       data = np.array(
@@ -249,8 +233,6 @@ def testUnsortedSegmentSum1DIndices2DDataNonDisjoint(self):
                [0, 0, 0, 0]],
               dtype=dtype), y)
 
-  @test.disable_with_predicate(
-      pred=test.is_built_with_rocm, skip_message="Test fails on ROCm.") #TODO(rocm): weekly sync 25-05-14
   def testUnsortedSegmentSum2DIndices3DData(self):
     for dtype in self.numeric_types:
       data = np.array(
@@ -268,8 +250,6 @@ def testUnsortedSegmentSum2DIndices3DData(self):
               ], [0, 0, 0.], [90, 92, 94], [103, 104, 105], [0, 0, 0]],
               dtype=dtype), y)
 
-  @test.disable_with_predicate(
-      pred=test.is_built_with_rocm, skip_message="Test fails on ROCm.") #TODO(rocm): weekly sync 25-05-14
   def testUnsortedSegmentSum1DIndices3DData(self):
     for dtype in self.numeric_types:
       data = np.array(
@@ -298,8 +278,6 @@ def testUnsortedSegmentSumShapeError(self):
                             math_ops.unsorted_segment_sum, data, indices,
                             num_segments))
 
-  @test.disable_with_predicate(
-      pred=test.is_built_with_rocm, skip_message="Test fails on ROCm.") #TODO(rocm): weekly sync 24-11-05
   def testUnsortedSegmentOps1DIndices1DDataNegativeIndices(self):
     """Tests for min, max, and prod ops.
 

tensorflow/compiler/tests/unary_ops_test.py

@@ -215,8 +215,6 @@ def testCos(self):
             math_ops.cos, x, expected=np.cos(x), rtol=tol, atol=1e-5
         )
 
-  @test.disable_with_predicate(
-      pred=test.is_built_with_rocm, skip_message="Test fails on ROCm.") #TODO(rocm): weekly sync 24-11-05 
   def testSigmoidNumericalStability(self):
     for dtype in self.float_types:
       if dtype != np.float16:

tensorflow/python/ops/BUILD

@@ -3513,7 +3513,6 @@ cuda_py_strict_test(
     main = "math_ops_test.py",
     tags = [
         "no_windows_gpu",
-        "cuda-only", #TODO(rocm): weekly sync 240919
     ],
     deps = [
         ":array_ops",