cupy.cuda.Graph#

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

The CUDA graph object.

Currently this class cannot be initiated by the user and must be created via stream capture. See begin_capture() for detail.

Methods

launch(self, stream=None)#

Launch the CUDA graph on the given stream.

Parameters:: stream (Stream) – A CuPy stream object. If not specified (using the default value None), the graph is launched on the current stream.

See also

cudaGraphLaunch()

upload(self, stream=None)#

Upload the CUDA graph to the given stream.

Parameters:: stream (Stream) – A CuPy stream object. If not specified (using the default value None), the graph is uploaded the current stream.

See also

cudaGraphUpload()

__eq__(value, /)#: Return self==value.

__ne__(value, /)#: Return self!=value.

__lt__(value, /)#: Return self<value.

__le__(value, /)#: Return self<=value.

__gt__(value, /)#: Return self>value.

__ge__(value, /)#: Return self>=value.

Attributes

graph#

graphExec#