diff --git a/libkineto/src/output_json.cpp b/libkineto/src/output_json.cpp index 16faab84a..e1e6ac9d4 100644 --- a/libkineto/src/output_json.cpp +++ b/libkineto/src/output_json.cpp @@ -796,6 +796,8 @@ void ChromeTraceLogger::handleActivity(const libkineto::ITraceActivity& op) { libkineto::ActivityType::CONCURRENT_KERNEL, libkineto::ActivityType::CUDA_RUNTIME, libkineto::ActivityType::CUDA_DRIVER, + libkineto::ActivityType::XPU_RUNTIME, + libkineto::ActivityType::XPU_DRIVER, libkineto::ActivityType::PRIVATEUSE1_RUNTIME, libkineto::ActivityType::PRIVATEUSE1_DRIVER}; if (excludedTypes.find(op.type()) == excludedTypes.end()) {