From 60b08b0827c38f47e45ee29b3dc04b0720e53535 Mon Sep 17 00:00:00 2001 From: Feny Patel Date: Tue, 16 May 2023 16:38:58 -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: bd5e4eb8d844eadf91efb8bd712e0a351cd77594 --- 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},