experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Commeting out subtests that are failing due to JIRA ticket 236756, and also removing the no_rocm tag from the tests that contain those subtests

Browse Source
This commit is contained in:
Deven Desai 2020-08-12 23:05:32 +00:00
parent f5a822d201
commit ae9e3bd2fb
2 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/python/ops/parallel_for/math_test.py
View File

@ -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,

5
tensorflow/python/ops/ragged/ragged_dispatch_test.py
View File

@ -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.