cupy.prof.time_range

cupy.prof.time_range(message, color_id=None, argb_color=None, sync=False)

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

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