ROCm / ROCm/MAD

Onboard GLM-5.1-FP8 to MAD (vLLM WideEP disaggregation)

Open
#231 0 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
Python
Stars
43
Forks
62
Avg merge
7d 14h
Merged PRs (30d)
18

Description

Requesting team / customer: AMD FDE / Cohere Inference v-team (customer: Cohere — enterprise, on-prem/no-API; this benchmarking work supports Cohere's internal training pipeline evaluation and the longer-term goal of Cohere serving on AMD Instinct).
Jira Epic: pending — AIMODELS project key/access still being confirmed by requester; will link once created per Phase 1 of the intake process.

Model

HF repo: zai-org/GLM-5.1-FP8 (confirmed 2026-08-20 via HF Hub API as a distinct checkpoint from the bf16 zai-org/GLM-5.1)
Size / class: large MoE (MLA+DSA architecture)
Framework: vLLM disaggregated (prefill/decode split), MoRI EP backend
Scope: this is the WideEP disaggregation workstream specifically — separate from the vLLM co-located command-r7b/command-a-plus entries filed as companion issues. GLM-5.2 was considered earlier but is out of scope — GLM-5.1-FP8 is the only model in scope here. Targets disagg_models.yaml + run_xPyD_models.slurm / vllm_disagg_mori_io_ep.sh allowlists.
Config basis: MoRI EP flags and cuda-graph-bs sweeps copied from the existing DeepSeek-V3/DeepSeek-R1 entries as a starting point; UNTUNED for GLM-5.1-FP8 and flagged for revisit once real WideEP runs land.
kv-cache-dtype fp8 included in base_flags: originally justified by GLM-5.2's 1M-token context window OOMing KV cache in bf16 at meaningful concurrency. With GLM-5.2 now out of scope, this flag is being kept as a reasonable default for an FP8 checkpoint, but is unvalidated for GLM-5.1-FP8's own (smaller) context — revisit once real GLM-5.1-FP8 WideEP runs land.

Hardware target
AMD Instinct MI300X (gfx942). This scope is explicitly multi-node (xPyD disaggregated prefill/decode).
Workload / win-criteria
WideEP disaggregated serving benchmark — validates large-MoE expert-parallel + prefill/decode disaggregation on MI300X as a step toward Cohere production serving on Instinct. Success = clean run through the standard xPyD sweep at the tiers already agreed for Milestone 1.
Proposed changes

scripts/vllm_disagg/models.yaml — 1 new entry (GLM-5.1-FP8)
run_xPyD_models.slurm — VALID_MODELS addition (mandatory gate) + optional MORI_DP_MODE1_ALLOWED_MODELS / MORI_EP_VALID_MODELS additions
vllm_disagg_mori_io_ep.sh — MORI_DP_MODE1_ALLOWED_MODELS addition (only if DP_MODE=1 support pursued)

Draft config content available on request; no PR opened yet pending sign-off from requester and confirmation of Jira Epic link.
Open items before Phase 3 (Scoping) can close

Jira AIMODELS https://amd-hub.atlassian.net/browse/AIMODELS-1340
Confirm whether DP_MODE=1 (dp-attention MoRI EP) support is actually needed for GLM-5.1-FP8, or if DP_MODE=0 (tp-only) is sufficient for initial onboarding — affects whether the two DP_MODE=1 allowlist edits are needed

Filed per AMD's AI Models Intake Process (Confluence: MLSE space, "AI Models Intake Process"), Phase 2 "Technical Intake." This is 1 of 3 companion issues (command-r7b / command-a-plus / GLM-5.1-FP8) covering the same customer engagement.

Contributor guide

No contributing guide indexed for this repository

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start by comparing the existing DeepSeek-V3/DeepSeek-R1 entries in scripts/vllm_disagg/models.yaml and the related allowlists in run_xPyD_models.slurm and vllm_disagg_mori_io_ep.sh. Confirm whether DP_MODE=1 is required before editing the named entries, then run the standard multi-node xPyD sweep on MI300X. Done means GLM-5.1-FP8 passes the agreed Milestone 1 tiers cleanly.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, shell
Domain
devops, distributed-systems, machine-learning
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Active
Clarity
Mostly clear
Newbie friendliness
48/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.