From c424c5012662c43b72cf4975fc4d5641a33cf215 Mon Sep 17 00:00:00 2001
From: Clive Verghese <cliveverghese@google.com>
Date: Mon, 2 Oct 2023 12:16:26 -0700
Subject: [PATCH] Add host analysis to TPU timeline.

PiperOrigin-RevId: 570139622
---
 .../protobuf/dcn_slack_analysis.proto                     | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/plugin/tensorboard_plugin_profile/protobuf/dcn_slack_analysis.proto b/plugin/tensorboard_plugin_profile/protobuf/dcn_slack_analysis.proto
index 2c90ee2d0..5b311afe6 100644
--- a/plugin/tensorboard_plugin_profile/protobuf/dcn_slack_analysis.proto
+++ b/plugin/tensorboard_plugin_profile/protobuf/dcn_slack_analysis.proto
@@ -3,8 +3,8 @@ syntax = "proto3";
 package tensorboard_plugin_profile;
 
 message OpInstance {
-  uint64 start_time_ns = 1;
-  uint64 duration_ns = 2;
+  uint64 start_time_ps = 1;
+  uint64 duration_ps = 2;
 }
 
 message DcnSlack {
@@ -52,6 +52,8 @@ message DcnSlack {
   OpInstance recv_done = 12;
 
   string transfer_type = 13;
+
+  OpInstance host_graph_execution = 14;
 }
 
 message DcnSlackSummary {
@@ -80,6 +82,8 @@ message DcnSlackSummary {
   uint64 recv_done_duration_us = 12;
 
   string transfer_type = 13;
+
+  int64 host_stall_us = 14;
 }
 
 message DcnSlackAnalysis {