Skip to content

[AMD] Enable mamba-extra-buffer for Qwen3.5 on ROCm - #32552

Closed
nemanjaudovic wants to merge 27 commits into
sgl-project:amd_marchfrom
nemanjaudovic:enable-mamba-extra-buffer-rocm
Closed

nemanjaudovic wants to merge 27 commits into
sgl-project:amd_marchfrom
nemanjaudovic:enable-mamba-extra-buffer-rocm

Conversation

@nemanjaudovic

@nemanjaudovic nemanjaudovic commented Jul 27, 2026 •

Copy link
Copy Markdown

Ports upstream #30359 to amd_march: adds is_hip() to the
_validate_mamba_extra_buffer platform check so hybrid-mamba models (Qwen3.5 / Qwen3-Next)
can use the extra_buffer radix-cache strategy on ROCm instead of asserting at startup. The
required kernels (FLA chunk-scan, selective_state_update) are Triton and AMD-aware; the assert
was a validated-platform whitelist, not a hard limit.


CI States

Latest PR Test (Base): ❌ Run #34828715619
Latest PR Test (Extra): ❌ Run #34828715349
Latest PR Test (AMD ROCm 10): ➖ No AMD PR run found for this commit.

Kangyan-Zhou and others added 26 commits July 7, 2026 13:22
… NVFP4 online quantization (sgl-project#30397) (sgl-project#30424)

Co-authored-by: Brayden Zhong <b8zhong@uwaterloo.ca>
Co-authored-by: Brayden Zhong <brayden@radixark.ai>
… NVILA weight loading (sgl-project#30400) (sgl-project#30425)

Co-authored-by: Brayden Zhong <b8zhong@uwaterloo.ca>
Co-authored-by: Brayden Zhong <brayden@radixark.ai>
…TP: clamp padded-row seq_lens to >= 0 (sgl-project#30378) (sgl-project#30427)

Co-authored-by: DarkSharpness <76582120+DarkSharpness@users.noreply.github.com>
Co-authored-by: ziyi.xu <ziyi.xu@radixark.ai>
Co-authored-by: Claude Fable 5 <noreply@anthropic.com>
…er by default for large prefill chunks (sgl-project#30140) (sgl-project#30436)

Co-authored-by: YAMY <74099316+YAMY1234@users.noreply.github.com>
…ensor contiguous (sgl-project#27926) (sgl-project#30449)

Co-authored-by: Kaixi <matteochen3@gmail.com>
Co-authored-by: liqichao <liqichao@baidu.com>
Co-authored-by: chenbong <bhchen@stu.xmu.edu.cn>
…te all output slots on tie overflow (sgl-project#30512) (sgl-project#30559)

Co-authored-by: Jimmy Shong <69131491+Jiminator@users.noreply.github.com>
Co-authored-by: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
…te dispatch input for precision job (sgl-project#30495) (sgl-project#30566)

Co-authored-by: Xinyuan Tong <115166877+JustinTong0323@users.noreply.github.com>
…indices under tie overflow / inf scores (IMA in FA3 sparse decode) (sgl-project#30645) (sgl-project#30698)

Co-authored-by: DarkSharpness <76582120+DarkSharpness@users.noreply.github.com>
Co-authored-by: Claude Fable 5 <noreply@anthropic.com>
…E to main-first to avoid CUDA graph stream explosion (sgl-project#30460) (sgl-project#30714)

Co-authored-by: Cursor <cursoragent@cursor.com>
Co-authored-by: Jiminator <jimmysh341@gmail.com>
Co-authored-by: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
… scale layout (sgl-project#29275) (sgl-project#30724)

Co-authored-by: sonle5 <51179712+hdt98@users.noreply.github.com>
Co-authored-by: sunxxuns <126995791+sunxxuns@users.noreply.github.com>
…h on gfx950 (sgl-project#29479) (sgl-project#30725)

Co-authored-by: billishyahao <bill.he@amd.com>
Co-authored-by: HAI <hixiao@gmail.com>
…t kernel in decode graph when not recording (sgl-project#30302) (sgl-project#30726)

Co-authored-by: Rita Brugarolas <Rita.BrugarolasBrufau@amd.com>
…okens (sgl-project#30313) (sgl-project#30727)

Co-authored-by: Bingxu Chen <bingxche@amd.com>
Co-authored-by: YC Yen-Ching Tseng <yctseng@amd.com>
…DA-graph capture crash under torch_memory_saver (sgl-project#30557) (sgl-project#30730)

Co-authored-by: Zhiyao Jiang <jessicajiang324@gmail.com>
Co-authored-by: Xinyu Jiang <xinyuj2@andrew.cmu.edu>
…error (sgl-project#30374) (sgl-project#30729)

Co-authored-by: Wang, FangYuan <39615225+At1a8@users.noreply.github.com>
…trtllm long-context "!!!!" collapse (NaN routing) (sgl-project#31001) (sgl-project#31065)

Co-authored-by: Khoa Pham <khoa.pham@radixark.ai>
…oss PD and CUDA graph replay (sgl-project#30839) (sgl-project#31083)

Co-authored-by: Claude Fable 5 <noreply@anthropic.com>
…th prefill CP (sgl-project#30992) (sgl-project#31106)

Co-authored-by: zRzRzRzRzRzRzR <Yuxuan.Zhang2@liverpool.ac.uk>
@gemini-code-assist

Copy link
Copy Markdown
Contributor

Caution

The consumer version of Gemini Code Assist on GitHub has been sunset. All code review activity has officially ceased.

Signed-off-by: nemanjaudovic <nudovic@amd.com>
@Jiminator
Jiminator deleted the branch sgl-project:amd_march September 14, 2026 04:45
@Jiminator Jiminator closed this Sep 14, 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.

10 participants