Skip to content

Commit ca1e582

Browse files
kartikxsimon-mo
authored andcommitted
Raise error for data-parallel with benchmark_throughput (vllm-project#16737)
Signed-off-by: Kartik Ramesh <kartikx2000@gmail.com> Co-authored-by: Simon Mo <simon.mo@hey.com> Signed-off-by: Frieda (Jingying) Huang <jingyingfhuang@gmail.com>
1 parent faf22ea commit ca1e582

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

benchmarks/benchmark_throughput.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -523,6 +523,13 @@ def validate_args(args):
523523
raise ValueError(
524524
"Tokenizer must be the same as the model for MII backend.")
525525

526+
# --data-parallel is not supported currently.
527+
# https://github.com/vllm-project/vllm/issues/16222
528+
if args.data_parallel_size > 1:
529+
raise ValueError(
530+
"Data parallel is not supported in offline benchmark, \
531+
please use benchmark serving instead")
532+
526533

527534
if __name__ == "__main__":
528535
parser = FlexibleArgumentParser(description="Benchmark the throughput.")

0 commit comments

Comments
 (0)