From e6baccb0934f7a293b2d1f8b388f60af4fc505f7 Mon Sep 17 00:00:00 2001
From: Feny Patel <fenypatel@meta.com>
Date: Mon, 6 Jan 2025 10:02:53 -0800
Subject: [PATCH] add new activity type for profiling hccl events in kineto
 (#1026)

Summary:

Add "MTIA_COLLECTIVE_COMM" activity type in kineto for tracing hccl profiler events

Differential Revision: D66713888
---
 libkineto/include/ActivityType.h | 1 +
 libkineto/src/ActivityType.cpp   | 1 +
 2 files changed, 2 insertions(+)

diff --git a/libkineto/include/ActivityType.h b/libkineto/include/ActivityType.h
index 84887c0b5..d1c210271 100644
--- a/libkineto/include/ActivityType.h
+++ b/libkineto/include/ActivityType.h
@@ -41,6 +41,7 @@ enum class ActivityType {
   XPU_RUNTIME, // host side xpu runtime events
   COLLECTIVE_COMM, // collective communication
   MTIA_WORKLOADD, // MTIA workloadd events
+  MTIA_COLLECTIVE_COMM, // MTIA collective communication events
 
   // PRIVATEUSE1 Activity types are used for custom backends.
   // The corresponding device type is `DeviceType::PrivateUse1` in PyTorch.
diff --git a/libkineto/src/ActivityType.cpp b/libkineto/src/ActivityType.cpp
index 1b1cda8a5..93848f7ab 100644
--- a/libkineto/src/ActivityType.cpp
+++ b/libkineto/src/ActivityType.cpp
@@ -39,6 +39,7 @@ static constexpr std::array<ActivityTypeName, activityTypeCount + 1> map{
      {"xpu_runtime", ActivityType::XPU_RUNTIME},
      {"collective_comm", ActivityType::COLLECTIVE_COMM},
      {"mtia_workloadd", ActivityType::MTIA_WORKLOADD},
+     {"mtia_collective_comm", ActivityType::MTIA_COLLECTIVE_COMM},
      {"privateuse1_runtime", ActivityType::PRIVATEUSE1_RUNTIME},
      {"privateuse1_driver", ActivityType::PRIVATEUSE1_DRIVER},
      {"ENUM_COUNT", ActivityType::ENUM_COUNT}}};