Do not include CUDA deps for tpu diags external build.

PiperOrigin-RevId: 809023167

Author: tensorflower-gardener

tsl/profiler/lib/BUILD

@@ -9,9 +9,11 @@ load(
 )
 load(
     "@xla//xla/tsl/profiler/builds:build_config.bzl",
+    "nvtx_pie_external_deps",
     "tf_profiler_copts",
     "tf_profiler_pybind_cc_library_wrapper",
 )
+load("//third_party/bazel_skylib/lib:dicts.bzl", "dicts")
 
 # copybara:uncomment package(default_applicable_licenses = ["//tensorflow:license"])
 
@@ -292,7 +294,7 @@ cc_library(
 )
 
 cc_library(
-    name = "nvtx_utils_libtpu",
+    name = "nvtx_utils_external",
     srcs = ["nvtx_utils_stub.cc"],
     hdrs = ["nvtx_utils.h"],
     visibility = ["//visibility:public"],
@@ -308,11 +310,13 @@ cc_library(
     name = "nvtx_utils",
     hdrs = ["nvtx_utils.h"],
     visibility = ["//visibility:public"],
-    deps = select({
-        # Remove this once we clean up all the CUDA specific deps.
-        "libtpu_on_gce": [":nvtx_utils_libtpu"],
-        "//conditions:default": [":nvtx_utils_impl"],
-    }),
+    deps = select(dicts.add(
+        nvtx_pie_external_deps(),
+        {
+            ":libtpu_on_gce": [":nvtx_utils_external"],
+            "//conditions:default": [":nvtx_utils_impl"],
+        },
+    )),
 )
 
 cc_library(