co-authored by
Chen-Chen Yeh
parent
ae438d05fa
commit
bf01ad9367
@@ -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));
|
||||
|
||||
Reference in New Issue
Block a user