Create Plugin Tracer for plugins.

PiperOrigin-RevId: 573248197
google · Oct 13, 2023 · c126286 · c126286
1 parent 6946885
commit c126286
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/tsl/profiler/BUILD b/tsl/profiler/BUILD
@@ -13,7 +13,7 @@ package_group(
         "//tensorflow/python/profiler/...",
         "//tensorflow/python/tpu/profiler/...",
         "//tsl/profiler/...",
-        "//xla/backends/profiler/gpu/...",
+        "//xla/backends/profiler/...",
     ],
 )
 

diff --git a/tsl/profiler/lib/BUILD b/tsl/profiler/lib/BUILD
@@ -334,6 +334,7 @@ cc_library(
     name = "profiler_collection",
     srcs = ["profiler_collection.cc"],
     hdrs = ["profiler_collection.h"],
+    visibility = set_external_visibility(["//tensorflow/compiler/xla/backends/profiler/plugin:__pkg__"]),
     deps = [
         ":profiler_interface",
         "//tsl/platform:status",