From 40808178a58671ed0d25d62a0adae20a4cabf844 Mon Sep 17 00:00:00 2001 From: Rachel Han Date: Thu, 25 Sep 2025 14:22:41 -0700 Subject: [PATCH] Add logger to libtpu sdk. PiperOrigin-RevId: 811500210 --- tsl/platform/BUILD | 6 ++++++ tsl/profiler/lib/BUILD | 8 +------- 2 files changed, 7 insertions(+), 7 deletions(-) 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"], }),