[CUDA] Enable more graphs to be updatable (#2883)

2025-12-16 01:49:05 +08:00 · 2025-12-08 06:18:01 -08:00
parent a4b3bc969b
commit 27232db1ba
3 changed files with 28 additions and 24 deletions
--- a/mlx/backend/cuda/device.h
+++ b/mlx/backend/cuda/device.h
@@ -106,7 +106,7 @@ class CommandEncoder {
    cudaGraphNode_t node;
    // K = kernel
    // E = empty
-    // G* = subgraph (with metadata)
+    // () = subgraph (with metadata)
    // Symbols ':', '-' are reserved as separators
    std::string node_type;
    std::string id;