diff --git a/cuda_core/cuda/core/experimental/_graph.py b/cuda_core/cuda/core/experimental/_graph.py index cc394a8e6a..b8ebe9ae52 100644 --- 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[0], num_dependencies_out, child_graph._mnff.graph + graph_out, *deps_info_trimmed, num_dependencies_out, child_graph._mnff.graph ) ) ]