zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-08-13 20:56:45 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix wrong graph key when using concurrent context (#2447)

Browse Source
This commit is contained in:
Cheng 2025-07-31 22:01:05 +09:00 committed by GitHub
parent d32519c8ee
commit daafee676f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -318,6 +318,7 @@ void CommandEncoder::commit() {
// Reset state // Reset state
node_count_ = 0; node_count_ = 0;
graph_node_count_ = 0; graph_node_count_ = 0;
empty_node_count_ = 0;
from_nodes_.clear(); from_nodes_.clear();
to_nodes_.clear(); to_nodes_.clear();
graph_key_.clear(); graph_key_.clear();