bshillingford / python-cuda-profile

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Wrapper for CUDA profiler start/stop API functions. Zero dependencies.

Example:

import cudaprofile

cudaprofile.start()
# ... do expensive cuda stuff ...
cudaprofile.stop()

and run the script from nvprof or nvvp.

You may want to use nvprof with --profile-from-start-off and only call start() when desired.

About


Languages

Language:Python 100.0%