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

[HotFix]: Hot fix import path in 3fs_bench_client.py (#10463)

Browse Source
This commit is contained in:
hzh0425
2025-09-15 14:45:46 +08:00
committed by GitHub
parent f73aae0bfc
commit 2a37b24d23
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
benchmark/hf3fs/bench_client.py
View File

@@ -7,7 +7,7 @@ from typing import List
import torch import torch
from tqdm import tqdm from tqdm import tqdm
from sglang.srt.mem_cache.storage.hf3fs.client_hf3fs import Hf3fsClient from sglang.srt.mem_cache.storage.hf3fs.hf3fs_usrbio_client import Hf3fsUsrBioClient
def print_stats(x: List[int]): def print_stats(x: List[int]):
@@ -29,7 +29,7 @@ def test():
file_size = 1 << 40 file_size = 1 << 40
bytes_per_page = 16 << 20 bytes_per_page = 16 << 20
entries = 32 entries = 32
file_ops = Hf3fsClient(file_path, file_size, bytes_per_page, entries) file_ops = Hf3fsUsrBioClient(file_path, file_size, bytes_per_page, entries)
print("test batch_read / batch_write") print("test batch_read / batch_write")
num_pages = 128 num_pages = 128
@@ -74,7 +74,7 @@ def bench():
numel = bytes_per_page // dtype.itemsize numel = bytes_per_page // dtype.itemsize
file_ops = [ file_ops = [
Hf3fsClient(file_path, file_size, bytes_per_page, entries) Hf3fsUsrBioClient(file_path, file_size, bytes_per_page, entries)
for _ in range(numjobs) for _ in range(numjobs)
] ]