Added DCN topology level to Megascale stats.

PiperOrigin-RevId: 708459127
openxla · Dec 21, 2024 · dc700af · dc700af
1 parent 6b8b5a1
commit dc700af
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/xla/tsl/profiler/utils/xplane_schema.cc b/xla/tsl/profiler/utils/xplane_schema.cc
@@ -380,6 +380,7 @@ const MegaScaleStatTypeMap& GetMegaScaleStatTypeMap() {
        {"action_inputs", kMegaScaleActionInputs},
        {"transfer_source", kMegaScaleTransferSource},
        {"transfer_destinations", kMegaScaleTransferDestinations},
+       {"dcn_topology_level", kMegaScaleTransferDcnTopologyLevel},
        {"buffer_sizes", kMegaScaleBufferSizes},
        {"compute_operation", kMegaScaleComputeOperation},
        {"chunk", kMegaScaleChunk},

diff --git a/xla/tsl/profiler/utils/xplane_schema.h b/xla/tsl/profiler/utils/xplane_schema.h
@@ -364,6 +364,7 @@ enum MegaScaleStatType : uint8_t {
   kMegaScaleActionInputs,
   kMegaScaleTransferSource,
   kMegaScaleTransferDestinations,
+  kMegaScaleTransferDcnTopologyLevel,
   kMegaScaleBufferSizes,
   kMegaScaleComputeOperation,
   kMegaScaleChunk,