cupy.cuda.Graph#

class cupy.cuda.Graph(*args, **kwargs)[source]#

CUDA 图对象。

目前用户无法直接实例化此类，必须通过流捕获创建。详见 begin_capture()。

方法

launch(self, stream=None)#

在给定的流上启动 CUDA 图。

参数:: stream (Stream) – 一个 CuPy 流对象。如果未指定（使用默认值 None），则图在当前流上启动。

另请参阅

cudaGraphLaunch()

upload(self, stream=None)#

将 CUDA 图上传到给定的流。

参数:: stream (Stream) – 一个 CuPy 流对象。如果未指定（使用默认值 None），则图上传到当前流。

另请参阅

cudaGraphUpload()

__eq__(value, /)#: 返回 self==value。

__ne__(value, /)#: 返回 self!=value。

__lt__(value, /)#: 返回 self<value。

__le__(value, /)#: 返回 self<=value。

__gt__(value, /)#: 返回 self>value。

__ge__(value, /)#: 返回 self>=value。

属性

graph#

graphExec#