Skip to content

Commit 5e7713a

Browse files
committed
Finish graphs APIs and add tests
1 parent 3cfd11e commit 5e7713a

File tree

6 files changed

+1130
-197
lines changed

6 files changed

+1130
-197
lines changed

cuda_core/cuda/core/experimental/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
from cuda.core.experimental import utils
66
from cuda.core.experimental._device import Device
77
from cuda.core.experimental._event import Event, EventOptions
8-
from cuda.core.experimental._graph import Graph, GraphBuilder
8+
from cuda.core.experimental._graph import CompleteOptions, DebugPrintOptions, Graph, GraphBuilder, launch_graph
99
from cuda.core.experimental._launcher import LaunchConfig, launch
1010
from cuda.core.experimental._linker import Linker, LinkerOptions
1111
from cuda.core.experimental._module import ObjectCode

cuda_core/cuda/core/experimental/_device.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1274,7 +1274,7 @@ def sync(self):
12741274
handle_return(runtime.cudaDeviceSynchronize())
12751275

12761276
@precondition(_check_context_initialized)
1277-
def create_graph_bulder(self) -> GraphBuilder:
1277+
def create_graph_builder(self) -> GraphBuilder:
12781278
"""Create a new :obj:`~_graph.GraphBuilder` object.
12791279
12801280
Returns
@@ -1283,5 +1283,4 @@ def create_graph_bulder(self) -> GraphBuilder:
12831283
Newly created graph builder object.
12841284
12851285
"""
1286-
private_stream = self.create_stream()
1287-
return GraphBuilder._init(stream=private_stream, can_destroy_stream=True)
1286+
return GraphBuilder._init(stream=self.create_stream(), is_stream_owner=True)

0 commit comments

Comments
 (0)