Skip to content

Add the legacy DeepSeek-R1 and GPT-OSS-120B drafters - #116

Merged
arav-agarwal2 merged 1 commit into
mlcommons:mainfrom
arekay-nv:drafters/legacy-dsr1-gptoss-heads
Oct 9, 2026
Merged

arav-agarwal2 merged 1 commit into
mlcommons:mainfrom
arekay-nv:drafters/legacy-dsr1-gptoss-heads

Conversation

@arekay-nv

Copy link
Copy Markdown
Contributor

Summary

Adds the speculative-decoding drafters carried over from legacy submissions for the two non-agentic benchmarks to src/submission_checker/data/approved_drafters.yaml. Until now neither benchmark had an entry, so approved-drafter rejected every DeepSeek-R1 or GPT-OSS-120B point that uses speculative decoding.

Benchmark model_id weight_checksum Drafter
deepseek-r1 deepseek-ai/DeepSeek-R1 git-sha1:56d4cbbb4d29f4355bab4b9a39ccb717a14ad5ad Native MTP head (layer 61)
deepseek-r1 centml/DeepSeek-R1-NVFP4-v2-mlpinf git-sha1:93947a0d7bd04f73ff98636f6b18ff5839e7aaf9 Native MTP head (layer 61). The checkpoint is post-training quantized on the MLPerf Inference deepseek-r1 calibration set and keeps the MTP layer.
gpt-oss-120b nvidia/gpt-oss-120b-Eagle3-long-context git-sha1:633caf45f31288cbb70ee237f7c939db707ecc94 EAGLE3 head. nvidia/gpt-oss-120b-Eagle3 redirects to this repository. The head is unquantized (quant_algo: null).

All three repositories are public and ungated on Hugging Face at the pinned revisions.

Approval cohort

The new entries are recorded as task-force seed entries (§2.9.4: "the benchmark task force seeds the list with ≥ 1 approved head/drafter per benchmark model"). They use 2026-09-C1, the same cohort as the initial approvals, so the two-cohort lead time makes them eligible from target cohort 2026-10-C1.

Changes

  • approved_drafters.yaml: three entries, plus header comments on where they come from and why they use this cohort.
  • tests/submission_checker/test_checks_specdec_agentic.py: _REFERENCE_HEADS now includes the new heads, so the bundled-list, cohort and lead-time tests cover them.
  • README.md: the drafter section lists the new heads and their eligibility.

Test plan

  • pytest tests/submission_checker/test_checks_specdec_agentic.py: 36 passed
  • Full suite: 1527 passed
  • ruff check clean. ruff format --check already flags this test file on main, on lines this PR does not touch.

🤖 Generated with Claude Code

Carry three weight-identified drafters over from legacy submissions:

- deepseek-r1: the native MTP head (layer 61) in deepseek-ai/DeepSeek-R1
  @ 56d4cbbb and centml/DeepSeek-R1-NVFP4-v2-mlpinf @ 93947a0d. The NVFP4
  checkpoint is post-training quantized on the MLPerf Inference
  deepseek-r1 calibration set and keeps the MTP layer.
- gpt-oss-120b: nvidia/gpt-oss-120b-Eagle3-long-context @ 633caf45, the
  repository nvidia/gpt-oss-120b-Eagle3 redirects to. The head is
  unquantized.

They are recorded as task-force seed entries (§2.9.4) with the other
initial approvals in 2026-09-C1, so 2026-10-C1 submissions can use them.
The bundled-list tests now cover the new heads, and the README lists them.

Co-Authored-By: Claude Opus 5.5 (1M context) <noreply@anthropic.com>
@github-actions

github-actions Bot commented Oct 9, 2026

Copy link
Copy Markdown

MLCommons CLA bot All contributors have signed the MLCommons CLA ✍️ ✅

@arav-agarwal2
arav-agarwal2 merged commit c1d7077 into mlcommons:main Oct 9, 2026
7 checks passed
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.

2 participants