[Profiler] Do not aggregate frames with different devices (#9290)

apache · Oct 15, 2021 · c00ce37 · c00ce37
1 parent acff61c
commit c00ce37
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/runtime/profiling.cc b/src/runtime/profiling.cc
@@ -311,6 +311,9 @@ String ReportNode::AsTable(bool sort, bool aggregate) const {
       if (frame.find("Argument Shapes") != frame.end()) {
         name += Downcast<String>(frame["Argument Shapes"]);
       }
+      if (frame.find("Device") != frame.end()) {
+        name += Downcast<String>(frame["Device"]);
+      }
 
       if (aggregates.find(name) == aggregates.end()) {
         aggregates[name] = {i};