diff --git a/libkineto/include/ActivityType.h b/libkineto/include/ActivityType.h index 588e47abe..312461045 100644 --- a/libkineto/include/ActivityType.h +++ b/libkineto/include/ActivityType.h @@ -33,6 +33,7 @@ enum class ActivityType { GLOW_RUNTIME, // host side glow runtime events CUDA_PROFILER_RANGE, // CUPTI Profiler range for performance metrics HPU_OP, // HPU host side runtime event + XPU_RUNTIME, // host side xpu runtime events ENUM_COUNT, // This is to add buffer and not used for any profiling logic. Add your new type before it. OPTIONAL_ACTIVITY_TYPE_START = GLOW_RUNTIME, diff --git a/libkineto/src/ActivityType.cpp b/libkineto/src/ActivityType.cpp index 2685ba2f9..bf4bf1e27 100644 --- a/libkineto/src/ActivityType.cpp +++ b/libkineto/src/ActivityType.cpp @@ -32,6 +32,7 @@ static constexpr std::array map{{ {"glow_runtime", ActivityType::GLOW_RUNTIME}, {"cuda_profiler_range", ActivityType::CUDA_PROFILER_RANGE}, {"hpu_op", ActivityType::HPU_OP}, + {"xpu_runtime", ActivityType::XPU_RUNTIME}, {"ENUM_COUNT", ActivityType::ENUM_COUNT} }}; diff --git a/libkineto/test/ConfigTest.cpp b/libkineto/test/ConfigTest.cpp index fe3f78ffd..70a021f0d 100644 --- a/libkineto/test/ConfigTest.cpp +++ b/libkineto/test/ConfigTest.cpp @@ -114,6 +114,10 @@ TEST(ParseTest, ActivityTypes) { EXPECT_TRUE(cfg2.parse("ACTIVITY_TYPES = cpu_op")); EXPECT_EQ(cfg2.selectedActivityTypes(), std::set({ActivityType::CPU_OP})); + + EXPECT_TRUE(cfg2.parse("ACTIVITY_TYPES = xpu_Runtime")); + EXPECT_EQ(cfg2.selectedActivityTypes(), + std::set({ActivityType::XPU_RUNTIME})); } TEST(ParseTest, SamplePeriod) {