From 586078dcad0e50e5856306e909ba0ef9ee04d74b Mon Sep 17 00:00:00 2001 From: Bhupendra Dubey Date: Sat, 21 Mar 2026 01:21:15 -0700 Subject: [PATCH] Enable native TPU profiling with XProf with libkineto integration. This implementation allows users to move away from the intermediate internal TPU profiling layer (`torch_tpu._internal.profiler`) towards a native profiling experience on PyTorch using `torch.profiler.` Also it allows user to capture pytorch side events and those added by record_function. PiperOrigin-RevId: 887198591 --- xla/tsl/profiler/rpc/client/BUILD | 1 + 1 file changed, 1 insertion(+) diff --git a/xla/tsl/profiler/rpc/client/BUILD b/xla/tsl/profiler/rpc/client/BUILD index 97c55eb9ac55c..40fba67114eed 100644 --- a/xla/tsl/profiler/rpc/client/BUILD +++ b/xla/tsl/profiler/rpc/client/BUILD @@ -66,6 +66,7 @@ cc_library( "//xla/tsl/profiler/rpc:__pkg__", "//learning/pathways/util/platform:__subpackages__", "//third_party/pathways/util/platform:__subpackages__", + "//third_party/py/torch_tpu:__subpackages__", ]), deps = [ "//xla/tsl/lib/io:zlib_compression_options",