Benchmark Model

modular/skills/benchmark-model

作者 modularb9b3a8e86700无许可证200 个星标收录于 2026年10月8日更新于 2026年10月8日仓库今天更新

Benchmark a model served on MAX with the `max benchmark` command: measure throughput (tokens/sec), latency (TTFT, TPOT, inter-token latency), and GPU utilization by driving load against a running `max serve` endpoint. Use this whenever the user wants to benchmark, load-test, or measure the performance of a MAX model, get tokens-per-second / TTFT / TPOT numbers, run a concurrency or request-rate sweep, compare latency vs throughput, size a deployment, or produce benchmark JSON, even if they don't say "benchmark" by name. Also use when a `max benchmark` run fails to connect or reports zero/garbage numbers.

AI 生成的概览

使用 max benchmark 命令对运行中的 MAX 模型服务进行基准测试,测量吞吐量与延迟。

功能
该技能指导智能体使用 max benchmark 命令对运行中的 MAX 模型服务进行负载测试。内容包括检查服务健康状态与已服务模型名称、选择工作负载(random、sharegpt 或 arxiv-summarization 数据集)、执行单请求或并发/请求速率扫描、将结果保存为 JSON,以及采集 GPU 统计信息。它还说明如何解读吞吐量(tokens/秒)、TTFT、TPOT 和 token 间延迟等指标,并排查运行失败或数值异常的问题。
适用场景
当你需要已部署在 MAX 上的模型性能数据时使用,例如每秒 token 数、TTFT 或 TPOT、并发或请求速率扫描、延迟与吞吐量的权衡,或部署规模估算。它也适用于 max benchmark 运行无法连接或报告零值、异常数值的情况。不适用于启动服务、内核级性能剖析或输出正确性检查。
运行要求
需要 pip 或 pixi 安装的 MAX,以及正在运行的 max serve 端点;采集 GPU 统计信息时,基准测试需与服务器运行在同一台 NVIDIA 机器上。该技能不包含脚本,仅有参考文档。

Benchmark a model on MAX

max benchmark measures a running model server. It's a load generator: it sends inference requests to a live max serve endpoint, times them, and reports throughput (tokens/sec) and latency (TTFT, TPOT, inter-token latency). A benchmark combines two things: a server under test, and a workload that matches the question you're asking.

Every run reports both throughput and latency, so there's no mode to select. Decide what you want to learn first, then pick the workload that measures it. Single-stream latency and peak throughput come from different workloads, so the workload you choose is the measurement. Get that right and a clean number falls out.

Use this skill when you want a performance number for a model on MAX: tokens/sec, TTFT / TPOT, a concurrency or request-rate sweep, a latency-vs-throughput tradeoff, or deployment sizing.

Don't use this skill when no server is running yet. max benchmark is a client, so start a server with the serve-model skill first. To find out where inference time goes at the kernel level, use profile-model. To check whether the output is correct, treat it as a parity task (import-model, then debug-model) rather than a benchmark.

This skill works anywhere MAX is installed (pip or pixi). Add pixi run in a pixi project.

References

The following table lists the reference files and when to read each one:

FileRead when
references/flags.md [blocked]Choosing any flag or dataset beyond the ones below
references/metrics.md [blocked]Turning the throughput and latency numbers into a conclusion
references/troubleshooting.md [blocked]A run won't connect, requests fail, or the numbers look wrong

Read the reference for what you're doing, not all of them upfront.

1. Check the server and read its model name

bash
curl -s http://localhost:8000/v1/health     # 200 = readycurl -s http://localhost:8000/v1/models      # note the served model name

Both checks matter before you spend a run:

  • The benchmark's --model must equal the server's --served-model-name exactly, or every request fails. Take the value from /v1/models rather than guessing it.
  • If that served name is an alias rather than a Hugging Face ID (no / in it), --tokenizer defaults to it and can't resolve, and the run dies with "not a valid model identifier." Pass the model's real Hugging Face ID as --tokenizer.

If nothing is serving, start a server with max serve (for a custom architecture, use the serve-model skill). One more server setting matters: --max-batch-size caps real concurrency. A sweep to --max-concurrency 32 against a server started with --max-batch-size 1 queues requests instead of batching them, so raise the server's batch size to match the sweep or the high-concurrency points mean nothing.

Wait for Server ready then benchmark. Benchmarking during compile or warmup produces garbage first-token times.

2. Pick the workload for your question

The workload is the measurement. Match it to what you want to learn:

What you want to knowWorkload
Best-case single-request latency (TTFT, TPOT)--max-concurrency 1, fixed random shape, small --num-prompts
Peak throughput and where latency degrades--max-concurrency 1,2,4,8,16,32 sweep, more prompts
Performance under a realistic mix--dataset-name sharegpt, moderate concurrency
Behavior at a target load--request-rate 1,2,4,8 sweep (requests/sec)

A sweep answers the first two rows at once: the concurrency-1 point is the best-case latency number, and the peak across the sweep is the throughput number. Reach for a dedicated concurrency-1 run when you only want the latency figure and don't want to pay for the rest of the curve.

The key knobs are the following (references/flags.md has the full catalog):

  • --dataset-name: pick random (synthetic, shape it with --random-input-len and --random-output-len), sharegpt (real chat), or arxiv-summarization (long context). random works best for clean, reproducible micro-measurements.
  • --max-concurrency and --request-rate: take a single value or a comma-separated sweep (1,2,4,8). A sweep is how you find the throughput knee.
  • --endpoint: use /v1/completions for base LMs, which need no chat template, or /v1/chat/completions for instruct and chat models, which must have a chat template or the requests return 400.
  • --max-output-len: sets the decode length, which dominates how long the run takes.
  • --num-prompts: required for single-turn runs.

3. Run it, save results, add GPU stats

For best-case single-request latency, pin concurrency to 1 and keep the shape fixed:

bash
pixi run max benchmark --backend modular --base-url http://localhost:8000 \  --model <served-model-name> --endpoint /v1/completions \  --dataset-name random --random-input-len 128 --random-output-len 128 \  --max-output-len 128 --num-prompts 32 --max-concurrency 1 \  --result-filename results/latency.json --collect-gpu-stats

For peak throughput and the latency knee, sweep concurrency and send more prompts:

bash
pixi run max benchmark --backend modular --base-url http://localhost:8000 \  --model <served-model-name> --endpoint /v1/completions \  --dataset-name random --random-input-len 512 --random-output-len 128 \  --max-output-len 128 --num-prompts 200 \  --max-concurrency 1,2,4,8,16,32 \  --result-filename results/throughput.json --collect-gpu-stats

Note these three things about saving and instrumenting a run:

  • --result-filename: writes metrics to JSON and creates the directories it needs. Set it whenever you want to track or compare runs; without it, MAX saves nothing. --metadata key=value stamps the JSON, for example --metadata tp=1 gpu=b200. A sweep also drops a results-<N>-median.json per step under --log-dir.
  • --collect-gpu-stats: adds GPU utilization and peak memory. This works only when the benchmark runs on the same machine as the server (NVIDIA).
  • For version-controlled configs, put options under a benchmark_config: key in a YAML file and pass --config-file file.yaml. Keys use snake_case, and CLI flags override the file.

4. Read the metrics

The run prints throughput and latency, and a sweep prints one row per point. The headline numbers are the following:

  • Output token throughput (tok/s): the main throughput number.
  • TTFT (time to first token): prefill responsiveness. Watch p50 and p99.
  • TPOT and ITL (time per output token and inter-token latency): decode speed.
  • GPU utilization and peak memory: reported with --collect-gpu-stats.

For how to turn these numbers into a conclusion, and the latency-vs-throughput tradeoff a sweep reveals, see references/metrics.md.

Troubleshooting

Match the symptom against references/troubleshooting.md, which covers connection failures, model-name mismatches, tokenizer-alias errors, chat-template 400s, flat throughput from a batch-size cap, and warmup-skewed first-token times. Confirm that curl /v1/health returns 200 before you check anything else.

来源与署名

来源:modular/skills位于benchmark-model提交b9b3a8e

许可证: 无许可证

内容归原作者所有。SourceWeft 从公开仓库中收录这些内容。

举报或申请下架