Use stream as name

Author: vzhurba01

cuda_core/cuda/core/experimental/_graph.py

@@ -86,7 +86,7 @@ def _check_capture_stream_provided(self, *args, **kwargs):
             raise RuntimeError("Tried to use a stream capture operation on a graph builder without a stream")
 
     @property
-    def legacy_stream_capture(self) -> Stream:
+    def stream(self) -> Stream:
         return self._mnff.stream
 
     @property
@@ -178,7 +178,7 @@ def join(self, *graph_builders):
                 raise ValueError("The primary graph builder should not be joined. Others builders should instead be joined onto it.")
 
         for graph in graph_builders:
-            self._mnff.stream.wait(graph.legacy_stream_capture)
+            self._mnff.stream.wait(graph.stream)
             # TODO: Do we close them now or let weakref handle it during garbage collection?
             #       This is a perf question, is there a good default?
             graph.close()

cuda_core/tests/test_graph.py

@@ -36,13 +36,13 @@ def test_graph_straight(init_cuda):
 
     # Test start
     graph_builder = Device().build_graph()
-    config = LaunchConfig(grid=1, block=1, stream=graph_builder.legacy_stream_capture)
+    config = LaunchConfig(grid=1, block=1, stream=graph_builder.stream)
 
     assert graph_builder.is_capture_active() is False
     graph_builder.begin_capture()
-    launch(empty_kernel, {"grid": 1, "block": 1, "stream": graph_builder.legacy_stream_capture})
-    launch(empty_kernel, {"grid": 1, "block": 1, "stream": graph_builder.legacy_stream_capture})
-    launch(empty_kernel, {"grid": 1, "block": 1, "stream": graph_builder.legacy_stream_capture})
+    launch(empty_kernel, {"grid": 1, "block": 1, "stream": graph_builder.stream})
+    launch(empty_kernel, {"grid": 1, "block": 1, "stream": graph_builder.stream})
+    launch(empty_kernel, {"grid": 1, "block": 1, "stream": graph_builder.stream})
     graph_builder.end_capture()
 
 
@@ -61,16 +61,16 @@ def test_graph_fork_join(init_cuda):
     graph_builder = Device().build_graph()
     assert graph_builder.is_capture_active() is False
     graph_builder.begin_capture()
-    launch(empty_kernel, {"grid": 1, "block": 1, "stream": graph_builder.legacy_stream_capture})
+    launch(empty_kernel, {"grid": 1, "block": 1, "stream": graph_builder.stream})
 
     left, right = graph_builder.fork(2)
-    launch(empty_kernel, {"grid": 1, "block": 1, "stream": left.legacy_stream_capture})
-    launch(empty_kernel, {"grid": 1, "block": 1, "stream": left.legacy_stream_capture})
-    launch(empty_kernel, {"grid": 1, "block": 1, "stream": right.legacy_stream_capture})
-    launch(empty_kernel, {"grid": 1, "block": 1, "stream": right.legacy_stream_capture})
+    launch(empty_kernel, {"grid": 1, "block": 1, "stream": left.stream})
+    launch(empty_kernel, {"grid": 1, "block": 1, "stream": left.stream})
+    launch(empty_kernel, {"grid": 1, "block": 1, "stream": right.stream})
+    launch(empty_kernel, {"grid": 1, "block": 1, "stream": right.stream})
     graph_builder.join(left, right)
 
-    launch(empty_kernel, {"grid": 1, "block": 1, "stream": graph_builder.legacy_stream_capture})
+    launch(empty_kernel, {"grid": 1, "block": 1, "stream": graph_builder.stream})
     graph_builder.end_capture()
 
     graph_builder.debug_dot_print(b"vlad.dot")