zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fix (#2613)

Browse Source 
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
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
mlx/backend/cuda/device.cpp
View File

@@ -299,6 +299,7 @@ void CommandEncoder::add_graph_node(cudaGraph_t child) {
graph_exec.instantiate(child); graph_exec.instantiate(child);
device_.make_current(); device_.make_current();
CHECK_CUDA_ERROR(cudaGraphLaunch(graph_exec, stream())); CHECK_CUDA_ERROR(cudaGraphLaunch(graph_exec, stream()));
return;
} }
cudaGraphNode_t node; cudaGraphNode_t node;
CHECK_CUDA_ERROR(cudaGraphAddChildGraphNode(&node, graph_, NULL, 0, child)); CHECK_CUDA_ERROR(cudaGraphAddChildGraphNode(&node, graph_, NULL, 0, child));