[Refactor] Benchmark: Add typed DatasetArgs/Loader registry and CPU dataset unit tests (#19147)

Co-authored-by: Liangsheng Yin <lsyincs@gmail.com>
This commit is contained in:
Ratish P
2026-02-24 12:22:01 -08:00
committed by GitHub
co-authored by Liangsheng Yin
parent 65de904130
commit ae6f6e1495
19 changed files with 807 additions and 160 deletions
+2 -5
View File
@@ -36,11 +36,8 @@ import requests
from tqdm.asyncio import tqdm
from transformers import AutoTokenizer, PreTrainedTokenizerBase
from sglang.benchmark.datasets import (
DatasetRow,
get_dataset,
get_mooncake_request_over_time,
)
from sglang.benchmark.datasets import DatasetRow, get_dataset
from sglang.benchmark.datasets.mooncake import get_mooncake_request_over_time
from sglang.benchmark.utils import (
get_tokenizer,
parse_custom_headers,