Skip to content

Commit bb1c0f9

Browse files
Fix an apparent mistake in GraphBuilder.add_child
While working on C++ stand-alone code executing what `test_graph.py` does in gh-843, I noticed that `add_child` passes dependendencies extracted from capturing stream inconsistently with num_dependencies parameter obtained in the same cuStreamGetCaptureInfo call. Incidentally, after correcting this error, I can no longer reproduce errors reported in gh-843
1 parent 85da64b commit bb1c0f9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

cuda_core/cuda/core/experimental/_graph.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -687,7 +687,7 @@ def add_child(self, child_graph: GraphBuilder):
687687
[
688688
handle_return(
689689
driver.cuGraphAddChildGraphNode(
690-
graph_out, deps_info_out[0], num_dependencies_out, child_graph._mnff.graph
690+
graph_out, *deps_info_out, num_dependencies_out, child_graph._mnff.graph
691691
)
692692
)
693693
]

0 commit comments

Comments
 (0)