Co-authored-by: Chen-Chen Yeh <[email protected]>
This commit is contained in:
Daniel Yeh
2025-09-22 20:12:04 -07:00
committed by GitHub
co-authored by Chen-Chen Yeh
parent ae438d05fa
commit bf01ad9367
+1
View File
@@ -299,6 +299,7 @@ void CommandEncoder::add_graph_node(cudaGraph_t child) {
graph_exec.instantiate(child);
device_.make_current();
CHECK_CUDA_ERROR(cudaGraphLaunch(graph_exec, stream()));
return;
}
cudaGraphNode_t node;
CHECK_CUDA_ERROR(cudaGraphAddChildGraphNode(&node, graph_, NULL, 0, child));