diff --git a/tensorflow/cc/profiler/BUILD b/tensorflow/cc/profiler/BUILD index d18a0bcab0c..5b4a105eb28 100644 --- a/tensorflow/cc/profiler/BUILD +++ b/tensorflow/cc/profiler/BUILD @@ -9,6 +9,7 @@ tf_cuda_cc_test( name = "profiler_test", srcs = ["profiler_test.cc"], tags = [ + "no_rocm", # stream level tracing not supported on ROCm "nogpu", # b/77649654 ], deps = [ diff --git a/tensorflow/core/BUILD b/tensorflow/core/BUILD index bac454b5b84..766c6f04149 100644 --- a/tensorflow/core/BUILD +++ b/tensorflow/core/BUILD @@ -4613,7 +4613,7 @@ tf_cc_test( size = "small", srcs = ["common_runtime/constant_folding_test.cc"], linkstatic = tf_kernel_tests_linkstatic(), - tags = tf_cuda_tests_tags(), + tags = tf_cuda_tests_tags() + ["no_rocm"], deps = [ ":core", ":core_cpu", @@ -4679,6 +4679,7 @@ tf_cuda_cc_test( size = "small", srcs = ["common_runtime/process_function_library_runtime_test.cc"], linkstatic = tf_kernel_tests_linkstatic(), + tags = ["no_rocm"], deps = [ ":core_cpu", ":core_cpu_internal", diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD index a8050e7afb7..8ab447d03c9 100644 --- a/tensorflow/python/BUILD +++ b/tensorflow/python/BUILD @@ -2500,6 +2500,7 @@ cuda_py_test( ], tags = [ "no_cuda_on_cpu_tap", + "no_rocm", "no_windows", ], )