add a new activity type in pytorch

Differential Revision: D45126544 fbshipit-source-id: 92214624f0a6771fd3be92952fead8f644da412a
pytorch · May 9, 2023 · 4e95eb1 · 4e95eb1
1 parent c51f346
commit 4e95eb1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/libkineto/include/ActivityType.h b/libkineto/include/ActivityType.h
@@ -32,6 +32,7 @@ enum class ActivityType {
 
     // Optional Activity types
     GLOW_RUNTIME, // host side glow runtime events
+    MTIA_RUNTIME, // host side MTIA runtime events
     CUDA_PROFILER_RANGE, // CUPTI Profiler range for performance metrics
     MTIA_CCP_EVENTS, // MTIA ondevice CCP events
     HPU_OP, // HPU host side runtime event

diff --git a/libkineto/src/ActivityType.cpp b/libkineto/src/ActivityType.cpp
@@ -31,6 +31,7 @@ static constexpr std::array<ActivityTypeName, activityTypeCount + 1> map{{
     {"python_function", ActivityType::PYTHON_FUNCTION},
     {"overhead", ActivityType::OVERHEAD},
     {"glow_runtime", ActivityType::GLOW_RUNTIME},
+    {"mtia_runtime", ActivityType::MTIA_RUNTIME},
     {"cuda_profiler_range", ActivityType::CUDA_PROFILER_RANGE},
     {"mtia_ccp_events", ActivityType::MTIA_CCP_EVENTS},
     {"hpu_op", ActivityType::HPU_OP},