[xla:gpu] Use RecordParams execution scope id to record all XLA comma…

…nds into command buffers Compute command execution scope from assigned stream id and default execution scope passed via record parameters. Add a test for running memsets in different execution scopes. PiperOrigin-RevId: 609630212
tensorflow · Feb 23, 2024 · 6043adb · 6043adb
1 parent bda8d22
commit 6043adb
Show file tree

Hide file tree

Showing 4 changed files with 165 additions and 47 deletions.
diff --git a/third_party/xla/xla/service/gpu/runtime/BUILD b/third_party/xla/xla/service/gpu/runtime/BUILD
@@ -114,6 +114,7 @@ cc_library(
         "@local_tsl//tsl/platform:errors",
         "@local_tsl//tsl/platform:logging",
         "@local_tsl//tsl/platform:statusor",
+        "@local_tsl//tsl/profiler/lib:scoped_annotation",
     ],
 )