Skip to content

GitLab

Unverified Commit 2a37b24d authored by hzh0425's avatar hzh0425 Committed by GitHub
Browse files

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

parent f73aae0b
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
benchmark/hf3fs/bench_client.py
View file @ 2a37b24d
...@@ -7,7 +7,7 @@ from typing import List ...@@ -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(): ...@@ -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(): ...@@ -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)
] ]
......
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