#53721 [ROCm][Connector] SWA+HMA-support in MoRI-IO connector (Gemma4)
原始 PR · 作者 simondanielsson · 合并时间 2026-09-14 20:52
MoRI-IO 连接器支持 SWA+HMA,PD 吞吐数倍提升
值得精读,尤其适合关注 KV 连接器、PD 分离与 HMA 的工程师。重点看三处设计决策:一是通过 `SupportsHMA` + `request_finished_all_groups` 把『每组一个块列表』的契约从类型层(`BlockIds`)到协议层打通,并与 NIXL 保持一致的命名;二是『能裁剪就裁剪、不能裁剪就 fail closed』的策略——SW 组只传窗口尾部、其他 hybrid 直接抛异常,避免静默精度损失;三是顺手修掉的 local/remote block id 混用 bug,这个话题值得作为案例研究:一个看似只关乎容量的特性改造,如何暴露并修复既有的数据正确性问题。阅读时建议对照 `test_moriio_connector.py` 中 `test_get_exchange_clipped_blocks_clips_only_sw_group` 与 `test_metadata_hma_block_ids_preserved_per_group`,并留意 PR 未覆盖的滚动升级兼容性与多 full-attention 组场景。
参与讨论