Skip to content

Commit 1b97fa7

Browse files
authored
[BUGFIX] fix value oom in radix tree (sgl-project#17400)
1 parent 236772c commit 1b97fa7

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

python/sglang/srt/mem_cache/hiradix_cache.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -796,7 +796,7 @@ def _insert_helper_host(
796796
new_node.parent = node
797797
new_node.key = key
798798
new_node.value = None
799-
new_node.host_value = host_value
799+
new_node.host_value = host_value.clone()
800800
new_node.hash_value = hash_value
801801
node.children[child_key] = new_node
802802
return matched_length
@@ -897,7 +897,7 @@ def insert(
897897
# shared-prefix node should also reflect max priority
898898
new_node.priority = max(new_node.priority, priority)
899899
if new_node.evicted:
900-
new_node.value = value[:prefix_len]
900+
new_node.value = value[:prefix_len].clone()
901901
self.evictable_size_ += len(new_node.value)
902902
else:
903903
self._inc_hit_count(new_node, chunked)
@@ -914,7 +914,7 @@ def insert(
914914
new_node = TreeNode(priority=priority)
915915
new_node.parent = node
916916
new_node.key = key
917-
new_node.value = value
917+
new_node.value = value.clone()
918918
node.children[child_key] = new_node
919919
self.evictable_size_ += len(value)
920920

python/sglang/srt/mem_cache/mamba_radix_cache.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1052,7 +1052,7 @@ def _insert_helper(
10521052
new_node = TreeNode()
10531053
new_node.parent = node
10541054
new_node.key = key
1055-
new_node.value = value
1055+
new_node.value = value.clone()
10561056
new_node.mamba_value = mamba_value
10571057
self.full_lru_list.insert_mru(new_node)
10581058
self.mamba_lru_list.insert_mru(new_node)

0 commit comments

Comments
 (0)