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

fix 3fs indices (#10855)

Browse Source
This commit is contained in:
pansicheng
2025-10-04 00:06:38 +08:00
committed by GitHub
parent 2c7f4ca2f2
commit 08af8ffb5c
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
python/sglang/srt/mem_cache/storage/hf3fs/storage_hf3fs.py
View File

@@ -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)
]