Tag missing rocm-only targets as manual

PiperOrigin-RevId: 679016215
openxla · Sep 26, 2024 · a769695 · a769695
1 parent 3ca96df
commit a769695
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/xla/backends/profiler/gpu/BUILD b/xla/backends/profiler/gpu/BUILD
@@ -11,6 +11,7 @@ load(
 load("//xla/tests:build_defs.bzl", "xla_test")
 load(
     "//xla/tsl:tsl.bzl",
+    "if_google",
     "internal_visibility",
     "tsl_copts",
     "tsl_gpu_library",
@@ -223,7 +224,10 @@ tsl_gpu_library(
     tags = [
         "gpu",
         "rocm-only",
-    ],
+    ] + if_google([
+        # TODO(b/360374983): Remove this tag once the target can be built without --config=rocm.
+        "manual",
+    ]),
     visibility = ["//visibility:public"],
     deps = [
         "//xla/stream_executor/rocm:roctracer_wrapper",
@@ -262,7 +266,10 @@ tsl_gpu_library(
     tags = [
         "gpu",
         "rocm-only",
-    ],
+    ] + if_google([
+        # TODO(b/360374983): Remove this tag once the target can be built without --config=rocm.
+        "manual",
+    ]),
     visibility = ["//visibility:public"],
     deps = [
         ":rocm_collector",

diff --git a/xla/stream_executor/BUILD b/xla/stream_executor/BUILD
@@ -937,7 +937,10 @@ alias(
     tags = [
         "gpu",
         "rocm-only",
-    ],
+    ] + if_google([
+        # TODO(b/360374983): Remove this tag once the target can be built without --config=rocm.
+        "manual",
+    ]),
 )
 
 alias(