Skip to content

[AMD] Add HiCache KV offloading for Qwen3.5 MXFP4 MI355X cookbook - #35879

Open
ChangLiu0709 wants to merge 2 commits into
sgl-project:mainfrom
ChangLiu0709:amd/qwen35-mi355x-hicache
Open

[AMD] Add HiCache KV offloading for Qwen3.5 MXFP4 MI355X cookbook#35879
ChangLiu0709 wants to merge 2 commits into
sgl-project:mainfrom
ChangLiu0709:amd/qwen35-mi355x-hicache

Conversation

@ChangLiu0709

@ChangLiu0709 ChangLiu0709 commented Aug 21, 2026

Copy link
Copy Markdown
Contributor

Summary

Add HiCache host-DRAM KV offloading support for the Qwen3.5-397B-A17B MXFP4 MI355X deployment, and bump the MI355X Docker image.

Changes

Deployment Generator (qwen35-deployment.jsx)

  • Add a KV Offloading UI toggle with two options: Disabled (default) and HiCache.
    • Only visible when MI355X + FP4 + 397B is selected.
    • When HiCache is enabled, emits:
      • --enable-hierarchical-cache
      • --hicache-ratio 1.5
      • --hicache-write-policy write_through
      • --hicache-io-backend direct
      • --hicache-mem-layout page_first_direct
    • These flags enable host-DRAM KV cache offloading, allowing higher concurrency beyond GPU-resident KV limits for agentic workloads.

Cookbook Page (Qwen3.5.mdx)

  • Bump MI355X Docker image: v0.5.15.post1-rocm720-mi35x-20260715v0.5.17-rocm720-mi35x-20260818.
  • Add HiCache KV offloading note in the deployment notes section.

Motivation

Aligned with InferenceX #2693, which adds HiCache host-DRAM KV tier arms to the Qwen3.5 MXFP4 MI355X AgentX sweep (TP4 concurrency 40–64, TP2 concurrency 20–32 with HiCache).

cc @yichiche


CI States

Latest PR Test (Base): ✅ Run #32719069137
Latest PR Test (Extra): ❌ Run #32719068970
Latest PR Test (AMD ROCm 7.2): ➖ No AMD PR run found for this commit.

…ump MI355X docker image

- Add 'KV Offloading' UI toggle (Disabled/HiCache) in the deployment
  generator, visible only for MI355X + FP4 + 397B. When enabled, emits
  --enable-hierarchical-cache, --hicache-ratio 1.5,
  --hicache-write-policy write_through, --hicache-io-backend direct,
  and --hicache-mem-layout page_first_direct.
- Bump MI355X docker image from v0.5.15.post1-20260715 to v0.5.17-20260818.
- Add HiCache prose note in the deployment notes section.

Aligned with InferenceX sgl-project#2693.
@github-actions github-actions Bot added the documentation Improvements or additions to documentation label Aug 21, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

documentation Improvements or additions to documentation

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant