From c4989dec9042acf77064174cb92453aee963b82a Mon Sep 17 00:00:00 2001 From: Feny Patel Date: Tue, 16 May 2023 07:30:38 -0700 Subject: [PATCH] add a new activity type in pytorch (#758) Summary: Pull Request resolved: https://github.com/pytorch/kineto/pull/758 Reviewed By: aaronenyeshi Differential Revision: D45126544 fbshipit-source-id: 9d7dbe425d6f0e58b3bb4c073a88a699fe56d173 --- 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 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},