cupy.cuda.profile

cupy.cuda.profile()[source]

Enable CUDA profiling during with statement.

This function enables profiling on entering a with statement, and disables profiling on leaving the statement.

>>> with cupy.cuda.profile():
...    # do something you want to measure
...    pass

Note

When starting nvprof from the command line, manually setting --profile-from-start off may be required for the desired behavior.