[misc] Add bench_serving compatibility shim (#28997)
This commit is contained in:
@@ -7,9 +7,9 @@
|
||||
Benchmark online serving with dynamic requests.
|
||||
|
||||
Usage:
|
||||
python3 -m sglang.bench_serving --backend sglang --num-prompt 10
|
||||
python3 -m sglang.benchmark.serving --backend sglang --num-prompt 10
|
||||
|
||||
python3 -m sglang.bench_serving --backend sglang --dataset-name random --num-prompts 3000 --random-input 1024 --random-output 1024 --random-range-ratio 0.5
|
||||
python3 -m sglang.benchmark.serving --backend sglang --dataset-name random --num-prompts 3000 --random-input 1024 --random-output 1024 --random-range-ratio 0.5
|
||||
"""
|
||||
|
||||
import argparse
|
||||
@@ -2124,7 +2124,7 @@ class LoRAPathAction(argparse.Action):
|
||||
getattr(namespace, self.dest).append(lora_name)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
def cli_main():
|
||||
parser = ArgumentParser(description="Benchmark the online serving throughput.")
|
||||
parser.add_argument(
|
||||
"--backend",
|
||||
@@ -2651,3 +2651,7 @@ if __name__ == "__main__":
|
||||
args = parser.parse_args()
|
||||
_validate_parsed_gsp_args(parser, args)
|
||||
run_benchmark(args)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
cli_main()
|
||||
|
||||
Reference in New Issue
Block a user