From c3a5a9d1392a5bc39e5db0d87ab6c16ef39d45b4 Mon Sep 17 00:00:00 2001 From: Feny Patel Date: Tue, 4 Feb 2025 07:37:07 -0800 Subject: [PATCH] add MTIA device properties (#1033) Summary: Pull Request resolved: https://github.com/pytorch/kineto/pull/1033 Reviewed By: sanrise Differential Revision: D68873647 --- libkineto/src/CuptiActivityProfiler.cpp | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/libkineto/src/CuptiActivityProfiler.cpp b/libkineto/src/CuptiActivityProfiler.cpp index c7f1af8ae..e40b90964 100644 --- 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(