add a new activity type in pytorch (#758)

Summary: Pull Request resolved: #758 Reviewed By: aaronenyeshi Differential Revision: D45126544 fbshipit-source-id: bd5e4eb8d844eadf91efb8bd712e0a351cd77594
pytorch · May 16, 2023 · 60b08b0 · 60b08b0
1 parent dbc742b
commit 60b08b0
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},