Skip to content

[Presets] Benchmark dataset support - #4134

Open
peterschmidt85 wants to merge 1 commit into
masterfrom
preset-dataset
Open

[Presets] Benchmark dataset support#4134
peterschmidt85 wants to merge 1 commit 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

`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>
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