cupy.prof.TimeRangeDecorator¶
-
class
cupy.prof.
TimeRangeDecorator
(message=None, color_id=None, argb_color=None, sync=False)[source]¶ Decorator to mark function calls with range in NVIDIA profiler
Decorated function calls are marked as ranges in NVIDIA profiler timeline.
>>> from cupy import prof >>> @cupy.prof.TimeRangeDecorator() ... def function_to_profile(): ... pass
Parameters: - message (str) – Name of a range, default use
func.__name__
. - color_id – range color ID
- argb_color – range color in ARGB (e.g. 0xFF00FF00 for green)
- sync (bool) – If
True
, waits for completion of all outstanding processing on GPU before callingcupy.cuda.nvtx.RangePush()
orcupy.cuda.nvtx.RangePop()
See also
cupy.nvtx.range()
cupy.cuda.nvtx.RangePush()
cupy.cuda.nvtx.RangePop()
Methods
- message (str) – Name of a range, default use