Skip to content

Commit

Permalink
Add following eligible unified runtime ops for drawing flow in traced…
Browse files Browse the repository at this point in the history
… JSON file (#1024)

Summary:
Intel GPU toolchain is using the unified runtime for now. As the profiling tools XPTI, it should be compatible with the unified runtime and previous SYCL runtime, so this PR adds the following eligible unified runtime ops for drawing flow in traced JSON file when using unified runtime.
```
urEnqueueUSMFill
urEnqueueUSMFill2D
urEnqueueUSMMemcpy
urEnqueueUSMMemcpy2D
urEnqueueKernelLaunch
urEnqueueKernelLaunchCustomExp
urEnqueueCooperativeKernelLaunchExp
```

Pull Request resolved: #1024

Reviewed By: aaronenyeshi

Differential Revision: D68457492

Pulled By: sraikund16

fbshipit-source-id: fae9357c9c8c8b3793dbeb0104c72787127db3b0
  • Loading branch information
zejun-chen authored and facebook-github-bot committed Jan 22, 2025
1 parent 3234660 commit 09e10ad
Showing 1 changed file with 8 additions and 1 deletion.
9 changes: 8 additions & 1 deletion libkineto/src/plugin/xpupti/XpuptiActivityProfiler.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand Down

0 comments on commit 09e10ad

Please sign in to comment.