cuda.core.experimental.Graph

class cuda.core.experimental.Graph

Represents an executable graph.

A graph groups a set of CUDA kernels and other CUDA operations together and executes them with a specified dependency tree. It speeds up the workflow by combining the driver activities associated with CUDA kernel launches and CUDA API calls.

Graphs must be built using a GraphBuilder object.

Methods

__init__()

close()

Destroy the graph.

launch(stream: Stream)

Launches the graph in a stream.

Parameters:

stream (Stream) – The stream in which to launch the graph

update(builder: GraphBuilder)

Update the graph using new build configuration from the builder.

The topology of the provided builder must be identical to this graph.

Parameters:

builder (GraphBuilder) – The builder to update the graph with.

upload(stream: Stream)

Uploads the graph in a stream.

Parameters:

stream (Stream) – The stream in which to upload the graph