Repository navigation
[AMD] cookbook: add HiCache host-DRAM KV tier for Qwen3.5 MXFP4 on MI355X - #36245
Merged
yichiche merged 1 commit intoAug 25, 2026
Merged
Conversation
yichiche
requested review from
JustinTong0323,
sogalin,
wisclmy0611 and
zijiexia
as code owners
August 25, 2026 02:07
1 task
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Motivation
InferenceX #2693 adds HiCache host-DRAM KV arms to the Qwen3.5 MXFP4 MI355X AgentX sweep (TP4 at concurrency 40/48/56/64 and TP2 at 20/24/28/32). The cookbook currently has no way to generate that configuration: the MXFP4-on-MI355X recipe always emits
--disable-radix-cache, and HiCache extends the radix cache, so the two cannot be combined by hand-editing the generated command —_handle_cache_compatibility()raisesValueErrorwhen--enable-hierarchical-cacheand--disable-radix-cacheare both present.This adds a KV Cache Offloading selector to the Qwen3.5 deployment generator so the HiCache variant is a first-class option rather than something a reader has to reconstruct from the benchmark scripts.
While here, this also corrects a note that #35445 left stale. That PR started emitting
--kv-cache-dtype fp8_e4m3for the MXFP4-on-MI355X recipe but did not update the page, which still told readers "It is not included in the generated commands above."Modifications
docs/src/snippets/autoregressive/qwen35-deployment.jsxkvOffloadoption group, KV Cache Offloading, withDisabled(default) andHost DRAM (HiCache). It is shown only formi355x+fp4, which is the arm HiCache is tuned on; every other combination is untouched.--enable-hierarchical-cache --hicache-ratio 1.5 --hicache-write-policy write_through --hicache-io-backend direct --hicache-mem-layout page_first_directand drops--disable-radix-cache. Values match the InferenceX recipe.docs/cookbook/autoregressive/Qwen/Qwen3.5.mdx--disable-radix-cache, and the host-DRAM budget implied by--hicache-ratio 1.5.TP2/EP1, the other half of InferenceX #2693, needs no change here: the generator's
mi355x.fp4entry carries noepkey, so it never emitted--expert-parallel-sizeand was already EP1. EP2 only ever applied to the B200 arm.Accuracy Tests
Not applicable — documentation only, no runtime code paths are touched.
Speed Tests and Profiling
Not applicable — documentation only.
The generator was exercised directly to confirm the emitted commands are valid and that nothing else regressed:
--enable-hierarchical-cache--disable-radix-cacheThe HiCache flag values are taken from the InferenceX recipe; that sweep has not published results yet, so no throughput numbers are quoted here.
Checklist
CI States
Latest PR Test (Base): ✅ Run #32800180789
Latest PR Test (Extra): ❌ Run #32800186335
Latest PR Test (AMD ROCm 7.2): ➖ No AMD PR run found for this commit.