[UnifiedTree]: Pin host buffers across async H→D in UnifiedRadixCache.load_back (#27444)

2026-06-13 15:16:56 +08:00
parent eb9483b5c2
commit 8ce05e8a20
4 changed files with 36 additions and 9 deletions
@@ -2994,8 +2994,11 @@ class UnifiedRadixCacheSuite:
        return chain

    def _release_ongoing_load_back_locks(self, tree):
-        for node, lock_params in list(tree.ongoing_load_back.values()):
+        for node, lock_params, host_lock_params in list(
+            tree.ongoing_load_back.values()
+        ):
            tree.dec_lock_ref(node, lock_params)
+            tree.dec_host_lock_ref(node, host_lock_params)
        tree.ongoing_load_back.clear()

    def _finish_pending_loads(self, tree):