diff --git a/benchmark/hf3fs/bench_client.py b/benchmark/hf3fs/bench_client.py index 0af3c80c7..ef4a967c4 100644 --- a/benchmark/hf3fs/bench_client.py +++ b/benchmark/hf3fs/bench_client.py @@ -29,7 +29,7 @@ def test(): file_size = 1 << 40 bytes_per_page = 16 << 20 entries = 32 - file_ops = Hf3fsUsrBioClient(file_path, file_size, bytes_per_page, entries) + file_ops = Hf3fsUsrBioClient(file_path, file_size, bytes_per_page, entries, 5) print("test batch_read / batch_write") num_pages = 128 @@ -74,7 +74,7 @@ def bench(): numel = bytes_per_page // dtype.itemsize file_ops = [ - Hf3fsUsrBioClient(file_path, file_size, bytes_per_page, entries) + Hf3fsUsrBioClient(file_path, file_size, bytes_per_page, entries, 5) for _ in range(numjobs) ]