From 7a79d9add33b6da4c3aeb6ac53f1e6c49e037da9 Mon Sep 17 00:00:00 2001
From: HelloClyde <helloclyde@qq.com>
Date: Wed, 15 Oct 2025 17:09:40 +0800
Subject: [PATCH] Fix the issue where subgraph does not work under --cache-none
 mode (#10329)

---
 comfy_execution/caching.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/comfy_execution/caching.py b/comfy_execution/caching.py
index 41224ce3b..d58f931f2 100644
--- a/comfy_execution/caching.py
+++ b/comfy_execution/caching.py
@@ -418,8 +418,11 @@ class DependencyAwareCache(BasicCache):
             The subcache object for the node.
         """
         subcache = await super()._ensure_subcache(node_id, children_ids)
+        await self.cache_key_set.add_keys(children_ids)
         for child_id in children_ids:
             self.descendants[node_id].add(child_id)
+            if child_id not in self.ancestors:
+                self.ancestors[child_id] = set()
             self.ancestors[child_id].add(node_id)
         return subcache