Co-authored-by: Chen-Chen Yeh <ge96noj@mytum.de>
This commit is contained in:
Daniel Yeh
2025-09-23 05:12:04 +02:00
committed by GitHub
parent ae438d05fa
commit bf01ad9367

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));