From c5aae20d8016d868a80ddcfabf296e95772830a6 Mon Sep 17 00:00:00 2001
From: Mudit Gokhale <muditgokhale@google.com>
Date: Sun, 26 Apr 2026 23:10:33 -0700
Subject: [PATCH] Enable custom configuration for TPU profiler options

PiperOrigin-RevId: 906146528
---
 tsl/profiler/protobuf/BUILD | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tsl/profiler/protobuf/BUILD b/tsl/profiler/protobuf/BUILD
index f127cb1f3..47698cab9 100644
--- a/tsl/profiler/protobuf/BUILD
+++ b/tsl/profiler/protobuf/BUILD
@@ -11,6 +11,9 @@ package_group(
     includes = [
         "@xla//xla/tsl/profiler:friends",
     ],
+    packages = [
+        "//third_party/py/torch_tpu/...",
+    ],
 )
 
 tf_proto_library(