From f3e5ca541425231798b3f99777e6991f9c52c17b Mon Sep 17 00:00:00 2001
From: Awni Hannun <awni@apple.com>
Date: Thu, 11 Dec 2025 19:13:44 -0800
Subject: [PATCH] [CUDA] Add host nodes to subgraph types for graph update
 (#2901)

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

diff --git a/mlx/backend/cuda/device.cpp b/mlx/backend/cuda/device.cpp
index 2e43e2df2..e7224642a 100644
--- a/mlx/backend/cuda/device.cpp
+++ b/mlx/backend/cuda/device.cpp
@@ -348,6 +348,9 @@ std::pair<std::string, bool> subgraph_to_key(cudaGraph_t graph) {
         key += subkey;
         break;
       }
+      case cudaGraphNodeTypeHost:
+        key += "H";
+        break;
       case cudaGraphNodeTypeMemset:
         key += "M";
         break;