The following commit adds a new unit-test which is failing on the ROC…

…m platform 50f8897 ``` //tensorflow/python/keras/distribute:dataset_creator_model_fit_test_gpu FAILED in 3 out of 3 in 112.5s ``` This commit adds a `no_rocm` tag to temporarily disable that unit-test on ROCm.
tensorflow · Mar 22, 2021 · 11e2175 · 11e2175
1 parent 65788c7
commit 11e2175
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tensorflow/python/keras/distribute/BUILD b/tensorflow/python/keras/distribute/BUILD
@@ -898,6 +898,7 @@ distribute_py_test(
     shard_count = 50,
     tags = [
         "multi_gpu",
+        "no_rocm",
         "no_tfrt",  # TODO(b/180537361): Reenable TFRT after the issue is resolved.
     ],
     deps = [