From 16f1f97eedeaa96d631fab38e889fca085aeff47 Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Fri, 6 Feb 2026 13:32:55 -0800
Subject: [PATCH] Replace xla.python._profiler to custom xprof client
 implementation.

PiperOrigin-RevId: 866593860
---
 tsl/profiler/lib/BUILD | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tsl/profiler/lib/BUILD b/tsl/profiler/lib/BUILD
index cad3c9faf..d75e7fef3 100644
--- a/tsl/profiler/lib/BUILD
+++ b/tsl/profiler/lib/BUILD
@@ -178,6 +178,7 @@ cc_library(
     visibility = internal_visibility([
         "@xla//xla/tsl:internal",
         "//third_party/py/jax:__subpackages__",
+        "@xla//xla/tsl/profiler:friends",
     ]),
     deps = [
         "//tsl/platform",