diff --git a/plugin/tensorboard_plugin_profile/protobuf/dcn_slack_analysis.proto b/plugin/tensorboard_plugin_profile/protobuf/dcn_slack_analysis.proto index e1c14a70f..2c90ee2d0 100644 --- a/plugin/tensorboard_plugin_profile/protobuf/dcn_slack_analysis.proto +++ b/plugin/tensorboard_plugin_profile/protobuf/dcn_slack_analysis.proto @@ -2,6 +2,11 @@ syntax = "proto3"; package tensorboard_plugin_profile; +message OpInstance { + uint64 start_time_ns = 1; + uint64 duration_ns = 2; +} + message DcnSlack { string rendezvous = 1; // Xprof observed send start time. @@ -39,6 +44,14 @@ message DcnSlack { // Send op name string send_op_name = 8; + + // Timestamp for the send/send-done/recv/recv-done ops + OpInstance send = 9; + OpInstance send_done = 10; + OpInstance recv = 11; + OpInstance recv_done = 12; + + string transfer_type = 13; } message DcnSlackSummary { @@ -59,6 +72,14 @@ message DcnSlackSummary { // Send op name. string send_op_name = 8; + + // Stall duration based on the op. + uint64 send_duration_us = 9; + uint64 recv_duration_us = 10; + uint64 send_done_duration_us = 11; + uint64 recv_done_duration_us = 12; + + string transfer_type = 13; } message DcnSlackAnalysis {