[UnifiedTree]: Fix the leaf determination logic in _cascade_evict. (#25068)

Co-authored-by: JINZ <1023553676@qq.com>
2026-05-13 10:57:32 +08:00
parent 409d350fb6
commit 6140e45ef3
2 changed files with 63 additions and 1 deletions
@@ -896,7 +896,13 @@ class UnifiedRadixCache(BasePrefixCache):
        target: EvictLayer = EvictLayer.DEVICE,
    ):
        """Cascade eviction from trigger to lower-or-equal priority components."""
-        is_leaf = len(node.children) == 0
+
+        is_leaf = False
+        if target == EvictLayer.DEVICE:
+            is_leaf = node in self.evictable_device_leaves
+        elif target == EvictLayer.HOST:
+            is_leaf = node in self.evictable_host_leaves
+
        trigger_priority = trigger.eviction_priority(is_leaf)

        for comp in self._components_tuple: