[XPTI] Add following eligible unified runtime

ops for drawing flow in traced JSON file urEnqueueUSMFill urEnqueueUSMFill2D urEnqueueUSMMemcpy urEnqueueUSMMemcpy2D urEnqueueKernelLaunch urEnqueueKernelLaunchCustomExp urEnqueueCooperativeKernelLaunchExp Signed-off-by: Chen, Zejun <[email protected]>
pytorch · Dec 27, 2024 · 6ff5f1b · 6ff5f1b
1 parent c2bbbdd
commit 6ff5f1b
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/libkineto/src/plugin/xpupti/XpuptiActivityProfiler.cpp b/libkineto/src/plugin/xpupti/XpuptiActivityProfiler.cpp
@@ -10,14 +10,21 @@ std::vector<std::array<unsigned char, 16>>
     XpuptiActivityProfilerSession::deviceUUIDs_ = {};
 std::vector<std::string> XpuptiActivityProfilerSession::correlateRuntimeOps_ = {
     "piextUSMEnqueueFill",
+    "urEnqueueUSMFill",
     "piextUSMEnqueueFill2D",
+    "urEnqueueUSMFill2D",
     "piextUSMEnqueueMemcpy",
+    "urEnqueueUSMMemcpy",
     "piextUSMEnqueueMemset",
     "piextUSMEnqueueMemcpy2D",
+    "urEnqueueUSMMemcpy2D",
     "piextUSMEnqueueMemset2D",
     "piEnqueueKernelLaunch",
+    "urEnqueueKernelLaunch",
     "piextEnqueueKernelLaunchCustom",
-    "piextEnqueueCooperativeKernelLaunch"};
+    "urEnqueueKernelLaunchCustomExp",
+    "piextEnqueueCooperativeKernelLaunch",
+    "urEnqueueCooperativeKernelLaunchExp"};
 
 // =========== Session Constructor ============= //
 XpuptiActivityProfilerSession::XpuptiActivityProfilerSession(