diff --git a/tensorflow/compiler/mlir/glob_lit_test.bzl b/tensorflow/compiler/mlir/glob_lit_test.bzl index 0ce1911d49e..416d1359a91 100644 --- a/tensorflow/compiler/mlir/glob_lit_test.bzl +++ b/tensorflow/compiler/mlir/glob_lit_test.bzl @@ -34,6 +34,7 @@ def glob_lit_tests( native.py_test( name = "glob_lit_tests", srcs = ["@llvm//:lit"], + tags = ["no_rocm"], args = [ "tensorflow/compiler/mlir --config-prefix=runlit", ], diff --git a/tensorflow/compiler/mlir/tensorflow/BUILD b/tensorflow/compiler/mlir/tensorflow/BUILD index 9342a34dd96..05950c7c630 100644 --- a/tensorflow/compiler/mlir/tensorflow/BUILD +++ b/tensorflow/compiler/mlir/tensorflow/BUILD @@ -632,6 +632,7 @@ cc_library( tf_cc_test( name = "error_util_test", srcs = ["utils/error_util_test.cc"], + tags = ["no_rocm"], deps = [ ":error_util", "//tensorflow/compiler/xla:test", diff --git a/tensorflow/compiler/tests/BUILD b/tensorflow/compiler/tests/BUILD index f2b22b9631d..70aa9cda531 100644 --- a/tensorflow/compiler/tests/BUILD +++ b/tensorflow/compiler/tests/BUILD @@ -395,6 +395,7 @@ tf_xla_py_test( srcs = ["depthwise_conv_op_test.py"], shard_count = 5, tags = [ + "no_rocm", "noasan", "nomsan", "notsan", diff --git a/tensorflow/compiler/xla/service/gpu/BUILD b/tensorflow/compiler/xla/service/gpu/BUILD index 6f2b30e600a..8e255bb07e0 100755 --- a/tensorflow/compiler/xla/service/gpu/BUILD +++ b/tensorflow/compiler/xla/service/gpu/BUILD @@ -1455,7 +1455,7 @@ cc_library( tf_cc_test( name = "buffer_comparator_test", srcs = ["buffer_comparator_test.cc"], - tags = tf_cuda_tests_tags(), + tags = ["no_rocm"] + tf_cuda_tests_tags(), deps = [ ":buffer_comparator", "//tensorflow/compiler/xla:shape_util",