Skip to content

Commit 20f6528

Browse files
mzzchyfacebook-github-bot
authored andcommitted
Add new ActivityType for Insight event (pytorch#1090)
Summary: Pull Request resolved: pytorch#1090 Create Activity Type for insight. User need to specify the activity type in the config to enable it Reviewed By: fenypatel99 Differential Revision: D75237589 fbshipit-source-id: 0e7628ed51c374e68de08f01b7d5bd5c745b3508
1 parent 728825a commit 20f6528

File tree

3 files changed

+3
-0
lines changed

3 files changed

+3
-0
lines changed

libkineto/include/ActivityType.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@ enum class ActivityType {
3232
OVERHEAD, // CUPTI induced overhead events sampled from its overhead API.
3333
MTIA_RUNTIME, // host side MTIA runtime events
3434
MTIA_CCP_EVENTS, // MTIA ondevice CCP events
35+
MTIA_INSIGHT, // MTIA Insight Events
3536
CUDA_SYNC, // synchronization events between runtime and kernels
3637

3738
// Optional Activity types

libkineto/src/ActivityType.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@ static constexpr std::array<ActivityTypeName, activityTypeCount + 1> map{
3232
{"overhead", ActivityType::OVERHEAD},
3333
{"mtia_runtime", ActivityType::MTIA_RUNTIME},
3434
{"mtia_ccp_events", ActivityType::MTIA_CCP_EVENTS},
35+
{"mtia_insight", ActivityType::MTIA_INSIGHT},
3536
{"cuda_sync", ActivityType::CUDA_SYNC},
3637
{"glow_runtime", ActivityType::GLOW_RUNTIME},
3738
{"cuda_profiler_range", ActivityType::CUDA_PROFILER_RANGE},

libkineto/test/ConfigTest.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -103,6 +103,7 @@ TEST(ParseTest, ActivityTypes) {
103103
ActivityType::CUDA_DRIVER,
104104
ActivityType::CUDA_SYNC,
105105
ActivityType::MTIA_RUNTIME,
106+
ActivityType::MTIA_INSIGHT,
106107
ActivityType::MTIA_CCP_EVENTS}));
107108

108109
Config cfg2;

0 commit comments

Comments
 (0)