add MTIA device properties (#1033)

Summary: Pull Request resolved: #1033 Reviewed By: sanrise Differential Revision: D68873647 fbshipit-source-id: f3ce2ca1078c52c726230b00d44d42ff878d9647
pytorch · Feb 4, 2025 · b2115df · b2115df
1 parent d92080f
commit b2115df
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/libkineto/src/CuptiActivityProfiler.cpp b/libkineto/src/CuptiActivityProfiler.cpp
@@ -312,7 +312,11 @@ void CuptiActivityProfiler::processTraceInternal(ActivityLogger& logger) {
   }
   for (const auto& session : sessions_) {
     if (auto props = session->getDeviceProperties(); !props.empty()) {
-      device_properties.push_back(props);
+      if (std::find(
+              device_properties.begin(), device_properties.end(), props) ==
+          device_properties.end()) {
+        device_properties.push_back(props);
+      }
     }
   }
   logger.handleTraceStart(