add a new activity type in pytorch (pytorch#758)

Summary: Pull Request resolved: pytorch#758 Differential Revision: D45126544 fbshipit-source-id: 7a5b54d98bf51640489b6e5b328820920985b73b
fenypatel99 · May 10, 2023 · cd0c4e6 · cd0c4e6
1 parent c51f346
commit cd0c4e6
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},