Pre-commit and revert fix

Author: vzhurba01

cuda_core/cuda/core/experimental/__init__.py

@@ -5,7 +5,13 @@
 from cuda.core.experimental import utils
 from cuda.core.experimental._device import Device
 from cuda.core.experimental._event import Event, EventOptions
-from cuda.core.experimental._graph import GraphCompleteOptions, GraphDebugPrintOptions, Graph, GraphBuilder, launch_graph
+from cuda.core.experimental._graph import (
+    Graph,
+    GraphBuilder,
+    GraphCompleteOptions,
+    GraphDebugPrintOptions,
+    launch_graph,
+)
 from cuda.core.experimental._launch_config import LaunchConfig
 from cuda.core.experimental._launcher import launch
 from cuda.core.experimental._linker import Linker, LinkerOptions

cuda_core/cuda/core/experimental/_graph.py

@@ -768,6 +768,10 @@ def launch_graph(parent_graph: GraphBuilder, child_graph: GraphBuilder):
     if not parent_graph.is_building:
         raise ValueError("Parent graph is being built.")
 
+    _, _, graph_out, dependencies_out, num_dependencies_out = handle_return(
+        driver.cuStreamGetCaptureInfo(parent_graph.stream.handle)
+    )
+
     child_node = handle_return(
         driver.cuGraphAddChildGraphNode(graph_out, dependencies_out, num_dependencies_out, child_graph._mnff.graph)
     )

cuda_core/tests/test_graph.py

@@ -10,10 +10,10 @@
 except ImportError:
     from cuda import nvrtc
 from cuda.core.experimental import (
-    GraphCompleteOptions,
-    GraphDebugPrintOptions,
     Device,
     GraphBuilder,
+    GraphCompleteOptions,
+    GraphDebugPrintOptions,
     LaunchConfig,
     Program,
     ProgramOptions,