diff --git a/tsl/platform/BUILD b/tsl/platform/BUILD
index 382b2938b..895a62b0e 100644
--- a/tsl/platform/BUILD
+++ b/tsl/platform/BUILD
@@ -27,6 +27,12 @@ load(
     "cc_library",
 )
 
+config_setting(
+    name = "libtpu_on_gce",
+    values = {"copt": "-DLIBTPU_ON_GCE"},
+    visibility = ["//visibility:public"],
+)
+
 package(
     # copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
     default_visibility = [
diff --git a/tsl/profiler/lib/BUILD b/tsl/profiler/lib/BUILD
index 214b3bbf4..d77ead545 100644
--- a/tsl/profiler/lib/BUILD
+++ b/tsl/profiler/lib/BUILD
@@ -298,19 +298,13 @@ cc_library(
     visibility = ["//visibility:public"],
 )
 
-config_setting(
-    name = "libtpu_on_gce",
-    values = {"copt": "-DLIBTPU_ON_GCE"},
-    visibility = ["//visibility:public"],
-)
-
 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_external"],
+        "//tsl/platform:libtpu_on_gce": [":nvtx_utils_external"],
         # Place holder for internal dependencies
         "//conditions:default": [":nvtx_utils_impl"],
     }),