DLPack 助手#

cupy.fromDlpack(dltensor)

从 DLPack 张量到 ndarray 的零拷贝转换。

时间范围#

cupy.prof.TimeRangeDecorator([message, ...])

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

cupy.prof.time_range(message[, color_id, ...])

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

计时助手#

cupyx.time.repeat(func[, args, kwargs, ...])

用于测量 CPU 和 GPU 消耗时间的计时工具。

分析器#

cupy.cuda.profile()

在 with 语句期间启用 CUDA 分析。

cupy.cuda.profiler.start

profilerStart() 启用分析。

cupy.cuda.profiler.stop

profilerStop() 禁用分析。

设备同步检测#

警告

这些 API 在 CuPy v10 中已弃用,并将在未来的版本中移除。

cupyx.allow_synchronize(allow)

允许或暂时禁止当前线程中的设备同步。

cupyx.DeviceSynchronized([message])

在禁止设备同步时检测到同步时引发。