pytorch / kineto

A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How can I capture cudaEventRecord?

xmty777 opened this issue · comments

I want to capture cudaEventRecord in the trace, while it was filtered out in #543
I wonder if there is an easy way for me to capture it in pytorch profiler, or only rebuild pytorch? Thanks a lot!

@aaronenyeshi any ideas why these were blacklisted. Curious if we can maybe configure this in an env variable or something?

@aaronenyeshi any ideas why these were blacklisted. Curious if we can maybe configure this in an env variable or something?

@chaekit may have more context on this blacklist. @xmty777 feel free to open a PR that enables them via env var.

You can also build your own local pytorch, and comment out these lines: https://github.com/pytorch/kineto/blob/main/libkineto/src/CuptiActivityProfiler.cpp#L363 , in the path: pytorch/pytorch/third_party/kineto/libkineto/src/CuptiActivityProfiler.cpp.