From b4ecc3ef7e5858a081c0942ec769c425f3d910a0 Mon Sep 17 00:00:00 2001 From: Ezekiel Calubaquib Date: Mon, 31 Aug 2026 12:02:08 -0700 Subject: [PATCH] Allow visibility of profile_interface and profile_factory prior torch_tpu migration layout PiperOrigin-RevId: 974005318 --- tsl/profiler/lib/BUILD | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tsl/profiler/lib/BUILD b/tsl/profiler/lib/BUILD index 83d63330c..e0ffcd001 100644 --- a/tsl/profiler/lib/BUILD +++ b/tsl/profiler/lib/BUILD @@ -87,7 +87,7 @@ cc_library( "@xla//xla/python:__pkg__", "//learning/brain/tfrc/executor/api:__pkg__", "//net/grpc/internal/src/core/ext/xprof_profiler:__pkg__", - "//third_party/py/torch_tpu/pjrt:__pkg__", + "//third_party/py/torch_tpu:__subpackages__", ]), deps = [ ":profiler_interface", @@ -144,7 +144,7 @@ cc_library( "@xla//xla/tsl/profiler:internal", "@xla//xla/tsl/profiler:xla_profiler_backends", "//net/grpc/internal/src/core/ext/xprof_profiler:__pkg__", - "//third_party/py/torch_tpu/pjrt:__pkg__", + "//third_party/py/torch_tpu:__subpackages__", ]), deps = [ "//tsl/profiler/protobuf:xplane_proto_cc",