Skip to content

GitLab

Unverified Commit 08af8ffb authored by pansicheng's avatar pansicheng Committed by GitHub
Browse files

fix 3fs indices (#10855)

parent 2c7f4ca2
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 2 deletions
python/sglang/srt/mem_cache/storage/hf3fs/storage_hf3fs.py
View file @ 08af8ffb
......@@ -524,7 +524,9 @@ class HiCacheHF3FS(HiCacheStorage):
flat = not self.is_zero_copy
values = (
[
self.mem_pool_host.get_data_page(host_indices[i * page_num], flat=flat)
self.mem_pool_host.get_data_page(
host_indices[i * self.mem_pool_host.page_size], flat=flat
)
for i in range(page_num)
]
if self.is_zero_copy
......@@ -574,7 +576,9 @@ class HiCacheHF3FS(HiCacheStorage):
# host_indices to kv_buffer
flat = not self.is_zero_copy
values = [
self.mem_pool_host.get_data_page(host_indices[i * page_num], flat=flat)
self.mem_pool_host.get_data_page(
host_indices[i * self.mem_pool_host.page_size], flat=flat
)
for i in range(page_num)
]
......
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