diff --git a/libkineto/include/ActivityType.h b/libkineto/include/ActivityType.h index 1c0555417..e0e00999b 100644 --- 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 index ba6fb6742..fcbb80ba9 100644 --- a/libkineto/src/ActivityType.cpp +++ b/libkineto/src/ActivityType.cpp @@ -31,6 +31,7 @@ static constexpr std::array 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},