cupy.cuda.Graph#

class cupy.cuda.Graph(*args, **kwargs)[源代码]#

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#