Skip to content

[Perf] [AMD] Kimi-K3: static FP8 dense layers and tuned decode GEMMs - #46653

Open
aryatsc wants to merge 8 commits into
sgl-project:mainfrom
modal-projects:devin/pr-fp8
Open

aryatsc wants to merge 8 commits into
sgl-project:mainfrom
modal-projects:devin/pr-fp8

Conversation

@aryatsc

@aryatsc aryatsc commented Oct 26, 2026 •

Copy link
Copy Markdown
Contributor

Motivation

K3's dense layers run bf16, or dynamic-FP8 with per-step activation scaling.

Modifications

Accuracy Tests

test_fp8_online_static_input.py passes.

Full K3 stack on MI355X TP8 (DFlash, prod config) vs B300, McNemar:

  • GPQA-D 89.4 vs 87.9 (p=0.55)
  • AIME25 96.7 vs 93.3 (p=1.0)
  • MATH500 92.2 vs 92.2 (p=1.0)
  • MMLU-Pro 89.6 vs 90.0 (p=0.77)
  • GSM8K 97.2%

Speed Tests and Profiling

Not measured on its own. It is included in the full-stack K3 MI355X numbers in the other K3 PRs.

Dependencies

Checklist


CI States

Latest PR Test (Base): ❌ Run #42125645010
Latest PR Test (Extra): ❌ Run #42125644899
Latest PR Test (AMD ROCm 10): ❌ Run #42125644998

Tip

Base CI only covers the core cases, and its workload may not exercise your change. Use /rerun-test to run the tests your change touches, or copy this prompt for an agent:

Find the registered tests this PR affects and run them with /rerun-test, following https://github.com/sgl-project/sglang/blob/main/.agents/skills/rerun-test/SKILL.md

aryatsc and others added 2 commits October 26, 2026 17:46
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>

# Conflicts:
#	python/sglang/srt/environ.py
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>

# Conflicts:
#	python/sglang/kernels/ops/moe/moe_sorting_small.py
@github-actions

This comment has been minimized.

Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>

# Conflicts:
#	python/sglang/kernels/ops/moe/__init__.py
@github-actions

This comment has been minimized.

…erCore.run

Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>

This branch has not been deployed

No deployments
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant