From 1265955b34dd63622caecb71fdae550fe5cb44fb Mon Sep 17 00:00:00 2001
From: rattus <46076784+rattus128@users.noreply.github.com>
Date: Tue, 5 May 2026 09:40:57 +1000
Subject: [PATCH] ops: handle multi-compute of the same weight (#13705)

If the same weight is used multiple times within the same prefetch
window, it should only apply compute state mutations once. Mark the
weight as fully resident on the first pass accordingly.
---
 comfy/ops.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/comfy/ops.py b/comfy/ops.py
index 4f0338346..585c185a3 100644
--- a/comfy/ops.py
+++ b/comfy/ops.py
@@ -253,6 +253,9 @@ def resolve_cast_module_with_vbar(s, dtype, device, bias_dtype, compute_dtype, w
     if bias is not None:
         bias = post_cast(s, "bias", bias, bias_dtype, prefetch["resident"], update_weight)
 
+    if prefetch["signature"] is not None:
+        prefetch["resident"] = True
+
     return weight, bias