EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

debug radixcache stack_overflow (#1499)

Browse Source
This commit is contained in:
luzengxiangcn
2024-09-24 19:58:01 +08:00
committed by GitHub
parent 28b4d8e144
commit e6692bf4a5
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
python/sglang/srt/mem_cache/radix_cache.py
View File

@@ -291,15 +291,15 @@ class RadixCache(BasePrefixCache):
def _collect_leaves(self): def _collect_leaves(self):
ret_list = [] ret_list = []
stack = [self.root_node]
def dfs_(cur_node): while stack:
cur_node = stack.pop()
if len(cur_node.children) == 0: if len(cur_node.children) == 0:
ret_list.append(cur_node) 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 return ret_list