add MTIA device properties

Differential Revision: D68873647
pytorch · Jan 31, 2025 · 44bb2f3 · 44bb2f3
1 parent daa5648
commit 44bb2f3
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/libkineto/src/CuptiActivityProfiler.cpp b/libkineto/src/CuptiActivityProfiler.cpp
@@ -314,7 +314,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(