In FetchEventArgs request, set 'hosts' to the host that the selected …

…event belongs to. PiperOrigin-RevId: 678437554
tensorflow · Sep 24, 2024 · 88f2da6 · 88f2da6
1 parent 94740b1
commit 88f2da6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/plugin/trace_viewer/tf_trace_viewer/tf-trace-viewer.html b/plugin/trace_viewer/tf_trace_viewer/tf-trace-viewer.html
@@ -974,6 +974,8 @@
       requestURL.searchParams.set('start_time_ms', event.start);
       requestURL.searchParams.set('duration_ms', event.duration);
       requestURL.searchParams.set('unique_id',event.args['uid']);
+      // Extract hostname from process container. e.g. "mach01 /device:TPU:0 (pid3)" ==> "mach01"
+      requestURL.searchParams.set('hosts', event.parentContainer.parent.name.split(' ')[0]);
       this._isLoading = true;
       this._throbber.className = 'active';
       const fetchData = await new Promise(function(resolve, reject) {