Skip to main content

Benchmark CLI

还没翻译译自英文(docs.vllm.ai)本页 241 段都还没译

Benchmark CLI

This section guides you through running benchmark tests with the extensive datasets supported on vLLM.

It's a living document, updated as new features and datasets become available.

Tip

The benchmarks described on this page are mainly for evaluating specific vLLM features as well as regression testing.

For benchmarking production vLLM servers, we recommend GuideLLM, an established performance benchmarking framework with live progress updates and automatic report generation. It is also more flexible than vllm bench serve in terms of dataset loading, request formatting, and workload patterns.

Model Forward Benchmark

For exact-batch model timing, including steady-state pipeline-parallel forwards, see vllm bench forward.

Attention Kernel Benchmark

Run the attention kernel smoke benchmark from the repository root using the exact backend names:

python3 benchmarks/attention_benchmarks/benchmark.py \
  --backends FLASH_ATTN FLASHINFER --batch-specs "8q1s1k"

If any configuration fails, the benchmark exits with status 1 after reporting results and writing any requested CSV or JSON output. Successful runs exit with status 0. Profiling runs also propagate failures, but do not save timing metrics.

Dataset Overview

DatasetOnlineOfflineData Path
ShareGPT✅✅wget https://huggingface.co/datasets/anon8231489123/ShareGPT_Vicuna_unfiltered/resolve/main/ShareGPT_V3_unfiltered_cleaned_split.json
ShareGPT4V (Image)✅✅wget https://huggingface.co/datasets/Lin-Chen/ShareGPT4V/resolve/main/sharegpt4v_instruct_gpt4-vision_cap100k.json
Note that the images need to be downloaded separately. For example, to download COCO's 2017 Train images:
wget http://images.cocodataset.org/zips/train2017.zip
ShareGPT4Video (Video)✅✅git clone https://huggingface.co/datasets/ShareGPT4Video/ShareGPT4Video
BurstGPT✅✅wget https://github.com/HPMLL/BurstGPT/releases/download/v1.1/BurstGPT_without_fails_2.csv
Random✅✅synthetic
RandomMultiModal (Image/Video)✅✅synthetic
RandomForReranking✅✅synthetic
Prefix Repetition✅✅synthetic
HuggingFace-VisionArena✅✅lmarena-ai/VisionArena-Chat
HuggingFace-MMVU✅✅yale-nlp/MMVU
HuggingFace-InstructCoder✅✅likaixin/InstructCoder
HuggingFace-AIMO✅✅AI-MO/aimo-validation-aime, AI-MO/NuminaMath-1.5, AI-MO/NuminaMath-CoT
HuggingFace-Other✅✅lmms-lab/LLaVA-OneVision-Data, Aeala/ShareGPT_Vicuna_unfiltered
HuggingFace-MTBench✅✅philschmid/mt-bench
HuggingFace-HumanEval✅✅openai/openai_humaneval
HuggingFace-GSM8K✅✅openai/gsm8k
HuggingFace-Blazedit✅✅vdaita/edit_5k_char, vdaita/edit_10k_char
HuggingFace-ASR✅✅openslr/librispeech_asr, facebook/voxpopuli, LIUM/tedlium, edinburghcstr/ami, speechcolab/gigaspeech, kensho/spgispeech, ArtificialAnalysis/Earnings22-Cleaned-AA, D4nt3/esb-datasets-earnings22-validation-tiny-filtered
Spec Bench✅✅wget https://raw.githubusercontent.com/hemingkx/Spec-Bench/refs/heads/main/data/spec_bench/question.jsonl
SPEED-Bench✅✅curl -LsSf https://raw.githubusercontent.com/NVIDIA-NeMo/Skills/refs/heads/main/nemo_skills/dataset/speed-bench/prepare.py | python3 -
Custom✅✅Local file: data.jsonl
Custom Audio✅✅Local file: audio_data.jsonl
Custom Image✅✅Local file: image_data.jsonl

Legend:

  • ✅ - supported
  • 🟡 - Partial support
  • 🚧 - to be supported

Note

HuggingFace dataset's dataset-name should be set to hf. For local dataset-path, please set hf-name to its Hugging Face ID like

--dataset-path /datasets/VisionArena-Chat/ --hf-name lmarena-ai/VisionArena-Chat

Examples

🚀 Online Benchmark

Show more

📈 Offline Throughput Benchmark

Show more

🛠️ Structured Output Benchmark

Show more

📚 Long Document QA Benchmark

Show more

🗂️ Prefix Caching Benchmark

Show more

Replay Timed Traces

Show more

🧪 Hashing Benchmarks

Show more

⚡ Request Prioritization Benchmark

Show more

👁️ Multi-Modal Benchmark

Show more

🔬 Multimodal Processor Benchmark

Benchmark per-stage latency of the multimodal (MM) input processor pipeline, including the encoder forward pass. This is useful for profiling preprocessing bottlenecks in vision-language models.

Show more

Embedding Benchmark

Benchmark the performance of embedding requests in vLLM.

Show more

Reranker Benchmark

Benchmark the performance of rerank requests in vLLM.

Show more