cupy.prof.time_range#

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

一个上下文管理器，用于将封闭的代码块描述为嵌套的时间范围

>>> from cupy import prof
>>> with cupy.prof.time_range('some range in green', color_id=0):
...    # do something you want to measure
...    pass
参数:

  • message – 时间范围的名称。

  • 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。