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

This commit is contained in:
hzh0425
2025-09-15 14:45:46 +08:00
committed by GitHub
parent f73aae0bfc
commit 2a37b24d23

View File

@@ -7,7 +7,7 @@ from typing import List
import torch
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]):
@@ -29,7 +29,7 @@ def test():
file_size = 1 << 40
bytes_per_page = 16 << 20
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")
num_pages = 128
@@ -74,7 +74,7 @@ def bench():
numel = bytes_per_page // dtype.itemsize
file_ops = [
Hf3fsClient(file_path, file_size, bytes_per_page, entries)
Hf3fsUsrBioClient(file_path, file_size, bytes_per_page, entries)
for _ in range(numjobs)
]