Skip to content

Add sglang-kimi-w4a16-moe-optimize eval instance - #12

Closed
Arist12 wants to merge 1 commit into
devfrom
add-sglang-kimi-w4a16-moe-optimize-v2
Closed

Arist12 wants to merge 1 commit into
devfrom
add-sglang-kimi-w4a16-moe-optimize-v2

Conversation

@Arist12

@Arist12 Arist12 commented Apr 7, 2026

Copy link
Copy Markdown
Collaborator

Summary

  • Adds sglang-kimi-w4a16-moe-optimize eval instance for AMD_OSS issue Add issue-level eval bundles for vLLM #35925 and #33123 #4 (Kimi-K2.5 model support)
  • Type: optimize (continuous decode latency metric, lower is better)
  • Pins sglang at pre-fix commit (parent of PR #17863) where W4A16 MoE dispatch crashes on ROCm
  • Agent must fix MoE dispatch AND optimize decode latency (TP=8, batch=1, input=8192, output=2048)
  • Model weights: Kimi-K2.5 (1T MoE, W4A16, 557GB) at /root/.cache/huggingface/Kimi-K2.5

Cross-validation results

  • Pre-fix: SCORE 0.0 (crash: gptq_marlin_repack not available on ROCm)
  • Post-fix: SCORE 100.0 (decode median: 24.0ms)
  • Reward-hacking: 2 vectors mitigated via immutable_files
  • Data leak: prevented (git history scrubbed, base image sglang replaced with pre-fix copy)

Files

  • Dockerfile: Pre-fix sglang checkout, data leak prevention, fork remotes
  • bench_kimi_w4a16.sh: Fixed benchmark params, agent-configurable backends via bench_config.env
  • test_harness.py: Binary scoring (crash=0, runs=100)
  • task_description.md: Symptom-only description, no solution hints
  • task.yaml: Optimization task with decode_median_ms metric
  • metadata.json: Hard difficulty, 8 GPUs required

…ization eval

Optimization-type eval for AMD_OSS issue #4. Agent must fix W4A16 MoE
dispatch (model crashes on ROCm) then optimize decode latency for
Kimi-K2.5 (1T MoE, TP=8, batch=1, input=8192, output=2048).

- Docker image pins sglang at pre-fix commit (before PR #17863)
- Base image sglang replaced with pre-fix code to prevent data leak
- Binary test harness (crash=0, runs=100), continuous metric via orchestrator
- Benchmark uses local model checkout (no HF download needed)
- Editable install at /sgl-workspace/sglang/ (base image packaging)
- immutable_files protects bench script and harness from replacement

Cross-validated by @john: pre-fix 0.0, post-fix 100.0 (24.0ms decode).
Reviewed by @alex (no static blockers).
@Arist12

Arist12 commented Apr 7, 2026

Copy link
Copy Markdown
Collaborator Author

Duplicate of #11

@Arist12 Arist12 marked this as a duplicate of #11 Apr 7, 2026
@Arist12 Arist12 closed this Apr 7, 2026
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