diff --git a/tensorflow/python/ops/parallel_for/math_test.py b/tensorflow/python/ops/parallel_for/math_test.py index 933ce765cdb..367f40d3411 100644 --- a/tensorflow/python/ops/parallel_for/math_test.py +++ b/tensorflow/python/ops/parallel_for/math_test.py @@ -82,6 +82,11 @@ class MathTest(PForTestCase, parameterized.TestCase): self._test_unary_cwise_ops(complex_ops, True) def test_unary_cwise_real_ops_1(self): + if test.is_built_with_rocm(): + # TODO(rocm): + # This fails on ROCm...see JIRA ticket 236756 + self.skipTest('Fails on ROCM') + real_ops = [ lambda x: math_ops.acosh(1 + math_ops.square(x)), math_ops.abs, diff --git a/tensorflow/python/ops/ragged/ragged_dispatch_test.py b/tensorflow/python/ops/ragged/ragged_dispatch_test.py index 0237624aa45..7a1d7c1882a 100644 --- a/tensorflow/python/ops/ragged/ragged_dispatch_test.py +++ b/tensorflow/python/ops/ragged/ragged_dispatch_test.py @@ -139,6 +139,11 @@ class RaggedDispatchTest(test_util.TensorFlowTestCase, parameterized.TestCase): ] ) # pyformat: disable def testUnaryElementwiseOp(self, x, op=math_ops.abs, **extra_args): + if test_util.IsBuiltWithROCm(): + # TODO(rocm): + # This fails on ROCm...see JIRA ticket 236756 + self.skipTest('Fails on ROCM') + result = op(x, **extra_args) # Run the wrapped op on the dense values, for comparison.