cupy.cuda.Graph — CuPy 13.4.1 documentation (original) (raw)

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.

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.

__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#