debug_radixcache_stack_overflow

python/sglang/srt/mem_cache/radix_cache.py

@@ -288,18 +288,18 @@ def _total_size_helper(self, node: TreeNode):
         for child in node.children.values():
             x += self._total_size_helper(child)
         return x
-
+    
     def _collect_leaves(self):
         ret_list = []
+        stack = [self.root_node]
 
-        def dfs_(cur_node):
+        while stack:
+            cur_node = stack.pop()
             if len(cur_node.children) == 0:
                 ret_list.append(cur_node)
+            else:
+                stack.extend(cur_node.children.values())
 
-            for x in cur_node.children.values():
-                dfs_(x)
-
-        dfs_(self.root_node)
         return ret_list