Merge pull request #47980 from ROCmSoftwarePlatform:google_upstream_r…

…ocm_csb_fix_210322 PiperOrigin-RevId: 367266081 Change-Id: If9650240583a388f65e04a9e6c6ff2de2d502673
tensorflow · Apr 7, 2021 · 37c5dfb · 37c5dfb
2 parents 30664a4 + 43124b6
commit 37c5dfb
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/tensorflow/python/keras/distribute/BUILD b/tensorflow/python/keras/distribute/BUILD
@@ -897,6 +897,7 @@ distribute_py_test(
     tags = [
         "multi_gpu",
         "no_oss",  # TODO(b/183640564): Reenable
+        "no_rocm",
         "notsan",  # TODO(b/184542721)
     ],
     deps = [

diff --git a/tensorflow/python/keras/optimizer_v2/BUILD b/tensorflow/python/keras/optimizer_v2/BUILD
@@ -138,6 +138,7 @@ cuda_py_test(
     srcs = ["adamax_test.py"],
     shard_count = 4,
     # TODO(b/168527439): invalid resource variable reference on GPU for TFRT.
+    tags = ["no_rocm"],
     deps = [
         ":optimizer_v2",
         "//tensorflow/python:client_testlib",

diff --git a/tensorflow/python/training/BUILD b/tensorflow/python/training/BUILD
@@ -1104,6 +1104,7 @@ cuda_py_test(
     size = "medium",
     srcs = ["adam_test.py"],
     python_version = "PY3",
+    tags = ["no_rocm"],
     deps = [
         ":adam",
         "//tensorflow/python:array_ops",