commit 1a26ef7993e8c840125d3b694c6e95cf0bc89830
author Deven Desai <deven.desai.amd@gmail.com> Thu Mar 25 20:49:35 2021 +0000
committer Deven Desai <deven.desai.amd@gmail.com> Fri Apr 02 18:28:12 2021 +0000
tree 92b1938e50ab24c46357fe91b9114c407fd21592
parent c8ed381a3f50e986d2986c596ca3f38e1d74f981

Adding no_rocm tag to tests that seem to have done flaky with ROCm 4.1

these tests

```
//tensorflow/python/keras/distribute:saved_model_save_load_test_gpu
//tensorflow/python/kernel_tests:tensordot_op_test_gpu
```

See details here - https://github.com/ROCmSoftwarePlatform/tensorflow-upstream/pull/1301

tensorflow/python/keras/distribute/BUILD

diff --git a/tensorflow/python/keras/distribute/BUILD b/tensorflow/python/keras/distribute/BUILD
index e15062c..76daa1c 100644
--- a/tensorflow/python/keras/distribute/BUILD
+++ b/tensorflow/python/keras/distribute/BUILD
@@ -819,6 +819,7 @@
     tags = [
         "multi_and_single_gpu",
         "no_cuda_asan",  # times out
+        "no_rocm",
     ],
     xla_tags = [
         "no_cuda_asan",  # times out

tensorflow/python/kernel_tests/BUILD

diff --git a/tensorflow/python/kernel_tests/BUILD b/tensorflow/python/kernel_tests/BUILD
index c586425..4d621ea 100644
--- a/tensorflow/python/kernel_tests/BUILD
+++ b/tensorflow/python/kernel_tests/BUILD
@@ -3586,6 +3586,7 @@
     size = "medium",
     srcs = ["tensordot_op_test.py"],
     shard_count = 20,
+    tags = ["no_rocm"],
     xla_enable_strict_auto_jit = False,  # b/161856380
     deps = [
         "//tensorflow/python:array_ops",