cupy.prof.TimeRangeDecorator#

class cupy.prof.TimeRangeDecorator(message=None, color_id=None, argb_color=None, sync=False)[source]#

在 NVIDIA 性能分析器中用范围标记函数调用的装饰器

使用装饰器的函数调用会在 NVIDIA 性能分析器时间线中标记为范围。

>>> from cupy import prof
>>> @cupy.prof.TimeRangeDecorator()
... def function_to_profile():
...     pass
参数:

  • message (str) – 范围的名称，默认为 func.__name__。

  • color_id – 范围颜色 ID

  • argb_color – ARGB 格式的范围颜色 (例如，表示绿色的 0xFF00FF00)

  • sync (bool) – 如果为 True，则在调用 cupy.cuda.nvtx.RangePush() 或 cupy.cuda.nvtx.RangePop() 之前，等待 GPU 上所有未完成的处理完成。

另请参阅

cupy.cuda.nvtx.RangePush() cupy.cuda.nvtx.RangePop()

警告

此装饰器已弃用。请改用 cupyx.profiler.time_range。

方法

__call__(func)[source]#

将自身作为函数调用。

__enter__()[source]#
__exit__(exc_type, exc_value, traceback)[source]#
__eq__(value, /)#

返回 self==value。

__ne__(value, /)#

返回 self!=value。

__lt__(value, /)#

返回 self<value。

__le__(value, /)#

返回 self<=value。

__gt__(value, /)#

返回 self>value。

__ge__(value, /)#

返回 self>=value。