From daafee676fc31626edb42e74ef7c227a0822305e Mon Sep 17 00:00:00 2001
From: Cheng <zcbenz@gmail.com>
Date: Thu, 31 Jul 2025 22:01:05 +0900
Subject: [PATCH] Fix wrong graph key when using concurrent context (#2447)

---
 mlx/backend/cuda/device.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/mlx/backend/cuda/device.cpp b/mlx/backend/cuda/device.cpp
index ddd235d582..5871ce3e22 100644
--- a/mlx/backend/cuda/device.cpp
+++ b/mlx/backend/cuda/device.cpp
@@ -318,6 +318,7 @@ void CommandEncoder::commit() {
     // Reset state
     node_count_ = 0;
     graph_node_count_ = 0;
+    empty_node_count_ = 0;
     from_nodes_.clear();
     to_nodes_.clear();
     graph_key_.clear();