Skip to content

GitLab

Unverified Commit b1709305 authored by wxsm's avatar wxsm Committed by GitHub
Browse files

feat: radix tree code optimize (#1697)

parent 5ab20cce
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
python/sglang/srt/mem_cache/radix_cache.py
View file @ b1709305
...@@ -245,7 +245,7 @@ class RadixCache(BasePrefixCache): ...@@ -245,7 +245,7 @@ class RadixCache(BasePrefixCache):
def _split_node(self, key, child: TreeNode, split_len: int): def _split_node(self, key, child: TreeNode, split_len: int):
# new_node -> child # new_node -> child
new_node = TreeNode() new_node = TreeNode()
new_node.children = {key[split_len:][0]: child} new_node.children = {key[split_len]: child}
new_node.parent = child.parent new_node.parent = child.parent
new_node.lock_ref = child.lock_ref new_node.lock_ref = child.lock_ref
new_node.key = child.key[:split_len] new_node.key = child.key[:split_len]
...@@ -253,7 +253,7 @@ class RadixCache(BasePrefixCache): ...@@ -253,7 +253,7 @@ class RadixCache(BasePrefixCache):
child.parent = new_node child.parent = new_node
child.key = child.key[split_len:] child.key = child.key[split_len:]
child.value = child.value[split_len:] child.value = child.value[split_len:]
new_node.parent.children[key[:split_len][0]] = new_node new_node.parent.children[key[0]] = new_node
return new_node return new_node
def _insert_helper(self, node: TreeNode, key: List, value): def _insert_helper(self, node: TreeNode, key: List, value):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment