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

[CUDA] Set current device before cudaGraphLaunch (#2351)

Browse Source
This commit is contained in:
Cheng
2025-07-10 23:24:02 +09:00
committed by GitHub
parent 8c7bc30ce4
commit 8fb3e7a26c
2 changed files with 11 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -93,6 +93,7 @@ class CommandEncoder {
void insert_graph_dependencies(GraphNode node);
void insert_graph_dependencies(std::vector<GraphNode> nodes);
Device& device_;
CudaStream stream_;
cudaGraph_t graph_;
Worker worker_;