added support for DMA-only tiles in trace

Signed-off-by: Paul Schumacher <[email protected]>
Xilinx · Dec 20, 2023 · a3ce366 · a3ce366
1 parent 05e25ce
commit a3ce366
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/runtime_src/xdp/profile/plugin/aie_trace/aie_trace_metadata.cpp b/src/runtime_src/xdp/profile/plugin/aie_trace/aie_trace_metadata.cpp
@@ -117,7 +117,8 @@ namespace xdp {
         && shimTileMetricsSettings.empty() && shimGraphMetricsSettings.empty()) {
       isValidMetrics = false;
     } else {
-      getConfigMetricsForTiles(aieTileMetricsSettings, aieGraphMetricsSettings, module_type::core);
+      // Use DMA type here to include both core-active tiles and DMA-only tiles
+      getConfigMetricsForTiles(aieTileMetricsSettings, aieGraphMetricsSettings, module_type::dma);
       getConfigMetricsForTiles(memTileMetricsSettings, memGraphMetricsSettings, module_type::mem_tile);
       getConfigMetricsForInterfaceTiles(shimTileMetricsSettings, shimGraphMetricsSettings);
       setTraceStartControl(compilerOptions.graph_iterator_event);
@@ -314,8 +315,8 @@ namespace xdp {
 
     std::set<tile_type> allValidTiles;
     auto validTilesVec = metadataReader->getTiles("all", type, "all");
-    std::unique_copy(validTilesVec.begin(), validTilesVec.end(), std::inserter(allValidTiles, allValidTiles.end()), 
-                     tileCompare);
+    std::unique_copy(validTilesVec.begin(), validTilesVec.end(), 
+                     std::inserter(allValidTiles, allValidTiles.end()), tileCompare);
 
     // STEP 1 : Parse per-graph and/or per-kernel settings
 

diff --git a/src/runtime_src/xdp/profile/plugin/aie_trace/aie_trace_metadata.h b/src/runtime_src/xdp/profile/plugin/aie_trace/aie_trace_metadata.h
@@ -131,6 +131,7 @@ class AieTraceMetadata {
 
     std::map<module_type, std::string> defaultSets {
       { module_type::core,     "functions"},
+      { module_type::dma,      "functions"},
       { module_type::mem_tile, "input_channels"},
       { module_type::shim,     "input_ports"}
     };