Sort events after processing dcn messages

PiperOrigin-RevId: 580329190
tensorflow · Nov 8, 2023 · 35e1480 · 35e1480
1 parent 5df41c0
commit 35e1480
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tensorflow/core/profiler/convert/process_megascale_dcn.cc b/tensorflow/core/profiler/convert/process_megascale_dcn.cc
@@ -47,6 +47,8 @@ void ProcessMegascaleDcn(XSpace* space) {
   for (XPlane* device_xplane : device_xplanes) {
     dcn_events_processor.AddTpuCollectiveDcnTrafficToXPlane(device_xplane);
   }
+
+  SortXSpace(space);
 }
 }  // namespace profiler
 }  // namespace tensorflow