Implemented fix to work with both CTK 12.9 and CTK 13.0

oleksandr-pavlyk · oleksandr-pavlyk · commit 7f9c16b29c27 · 2025-08-21T10:16:58.000-05:00
diff --git a/cuda_core/cuda/core/experimental/_graph.py b/cuda_core/cuda/core/experimental/_graph.py
@@ -683,11 +683,14 @@ def add_child(self, child_graph: GraphBuilder):
             driver.cuStreamGetCaptureInfo(stream_handle)
         )
 
+        # See https://github.com/NVIDIA/cuda-python/pull/879#issuecomment-3211054159
+        # for rationale
+        deps_info_trimmed = deps_info_out[:num_dependencies_out]
         deps_info_update = [
             [
                 handle_return(
                     driver.cuGraphAddChildGraphNode(
-                        graph_out, *deps_info_out, num_dependencies_out, child_graph._mnff.graph
+                        graph_out, *deps_info_trimmed, num_dependencies_out, child_graph._mnff.graph
                     )
                 )
             ]

Original file line number	Diff line number	Diff line change
`@@ -683,11 +683,14 @@ def add_child(self, child_graph: GraphBuilder):`
`683`	`683`	`driver.cuStreamGetCaptureInfo(stream_handle)`
`684`	`684`	`)`
`685`	`685`
	`686`	`+ # See https://github.com/NVIDIA/cuda-python/pull/879#issuecomment-3211054159`
	`687`	`+ # for rationale`
	`688`	`+ deps_info_trimmed = deps_info_out[:num_dependencies_out]`
`686`	`689`	`deps_info_update = [`
`687`	`690`	`[`
`688`	`691`	`handle_return(`
`689`	`692`	`driver.cuGraphAddChildGraphNode(`
`690`		`- graph_out, *deps_info_out, num_dependencies_out, child_graph._mnff.graph`
	`693`	`+ graph_out, *deps_info_trimmed, num_dependencies_out, child_graph._mnff.graph`
`691`	`694`	`)`
`692`	`695`	`)`
`693`	`696`	`]`