Skip to content

[Presets] Benchmark dataset support - #4134

Merged
peterschmidt85 merged 3 commits into
masterfrom
preset-dataset
Aug 13, 2026
Merged

[Presets] Benchmark dataset support#4134
peterschmidt85 merged 3 commits into
masterfrom
preset-dataset

Conversation

@peterschmidt85

@peterschmidt85 peterschmidt85 commented Aug 12, 2026

Copy link
Copy Markdown
Contributor

Dataset

By default, benchmarks use synthetic prompts shaped by input_tokens and output_tokens. Set dataset to benchmark on real text instead: a dataset the benchmark tool supports, or a Hugging Face dataset ID.

dataset: sharegpt

The dataset provides the requests, so input_tokens, output_tokens, and shared_prefix_tokens can't be set with it, and the preset records the measured means. A gated dataset requires HF_TOKEN in env.

dstack preset shows the dataset in place of the request shape:

$ dstack preset
 ID        BASE                       GPU           CONSTRAINTS             BENCHMARK                            STATUS        SUBMITTED
 47e2b788  nvidia/NVIDIA-Nemotron-…   RTXPRO4500:1  data=speed_bench conc=4 tok/s/user=186 ttft=71.6ms ctx=256K   verified (3)  3 min ago

🤖 Generated with Claude Code

Andrey Cheptsov and others added 3 commits August 12, 2026 23:40
`dataset` selects what every benchmark in a preset session measures: the
synthetic `random` prompts shaped by `input_tokens` and `output_tokens`, a
dataset the benchmark tool supports, or a Hugging Face dataset ID. A custom
dataset provides the requests, so the request-shape properties can't be set
with it, and the preset records the measured means instead.

The dataset is part of the contract: it is written to the session
constraints, the agent reports it with the benchmark, the preset records it,
and `dstack preset` shows it. A session that doesn't set it renders exactly
as before.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
@peterschmidt85
peterschmidt85 merged commit 706cac9 into master Aug 13, 2026
27 checks passed
@peterschmidt85
peterschmidt85 deleted the preset-dataset branch August 13, 2026 15:55
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant