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

debug_dot_str(self, flags=0)#

Make DOT formatted string of CUDA graph definition for debugging.

Parameters:

flags (unsigned int) – Flags to specify information to be included.

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