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

📈 Offline Throughput Benchmark

🛠️ Structured Output Benchmark

📚 Long Document QA Benchmark

🗂️ Prefix Caching Benchmark

Replay Timed Traces

🧪 Hashing Benchmarks

⚡ Request Prioritization Benchmark

👁️ Multi-Modal Benchmark

🔬 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.

Embedding Benchmark

Benchmark the performance of embedding requests in vLLM.

Reranker Benchmark

Benchmark the performance of rerank requests in vLLM.