diff --git a/libkineto/src/plugin/xpupti/XpuptiActivityProfiler.cpp b/libkineto/src/plugin/xpupti/XpuptiActivityProfiler.cpp index e722e49a..b2a56c6f 100644 --- a/libkineto/src/plugin/xpupti/XpuptiActivityProfiler.cpp +++ b/libkineto/src/plugin/xpupti/XpuptiActivityProfiler.cpp @@ -10,14 +10,21 @@ std::vector> XpuptiActivityProfilerSession::deviceUUIDs_ = {}; std::vector XpuptiActivityProfilerSession::correlateRuntimeOps_ = { "piextUSMEnqueueFill", + "urEnqueueUSMFill", "piextUSMEnqueueFill2D", + "urEnqueueUSMFill2D", "piextUSMEnqueueMemcpy", + "urEnqueueUSMMemcpy", "piextUSMEnqueueMemset", "piextUSMEnqueueMemcpy2D", + "urEnqueueUSMMemcpy2D", "piextUSMEnqueueMemset2D", "piEnqueueKernelLaunch", + "urEnqueueKernelLaunch", "piextEnqueueKernelLaunchCustom", - "piextEnqueueCooperativeKernelLaunch"}; + "urEnqueueKernelLaunchCustomExp", + "piextEnqueueCooperativeKernelLaunch", + "urEnqueueCooperativeKernelLaunchExp"}; // =========== Session Constructor ============= // XpuptiActivityProfilerSession::XpuptiActivityProfilerSession(