修复 MLA indexer 中 DeepGEMM context_lens 非连续问题
该 PR 值得精读,展示了在处理 CUDA graph 兼容性时如何避免动态内存分配的设计模式:通过预分配 1D 平坦缓冲区并在运行时用 view 重塑,而非调用 `.contiguous()`。同样的思路可应用于其他需要为外部 kernel 提供连续 tensor 的场合。
A high-throughput and memory-efficient inference and serving engine for LLMs
修复 MLA indexer 中 DeepGEMM context_lens 非连续问题
该 PR 值得精读,展示了在处理 CUDA graph 兼容性时如何避免动态内存分配的设计模式:通过预分配 1D 平坦缓冲区并在运行时用 view 重塑,而非调用 `.contiguous()`。同样的思路可应用于其他需要为外部 kernel 提供连续 tensor 的场合。
原始 PR · 作者 yewentao256 · 合并时间 2026-05-15 23:14
优化 MLA 注意力 _v_up_proj 的 bmm 效率
建议合并。该 PR 是一个清晰的性能微优化与代码清理,逻辑正确且风险极低。值得关注的是如何利用 `torch.bmm` 的 `out` 视图来避免额外复制,类似技巧可用于其他类似场景。
原始 PR · 作者 yewentao256 · 合并时间 2026-05-15 23:14
修复 KV Connector pre_forward 调用顺序
值得快速合并。这是 Model Runner V2 迁移的明确 bugfix,代码量小且已通过测试验证。
扩展 OAI Triton MoE ROCm 支持至 RDNA4
值得精读:本 PR 展示了在跨平台代码中处理设备功能检测的简洁方法,避免了 capability 元组因供应商实现不同而产生的别名问题。关注点:如何通过字符串匹配(`on_gfx1x`)避免硬编码 capability 数字,以及如何通过集中化辅助函数消除重复。
IPC权重同步优化:多GPU支持与分块打包传输
### 建议 该 PR 涉及权重传输核心路径的重要改造,值得 RLHF 开发者精读。重点关注:`packed_tensor.py` 的分块设计、`ipc_engine.py` 的多 GPU 全收集实现、以及 API 变更对下游的影响。建议后续 PR 跟进修复超大张量边界问题,并考虑增加正式 HTTP 分块端点。
原始 PR · 作者 benchislett · 合并时间 2026-05-15 22:29
修复 DFlash 与 FP8 KV-Cache 的兼容性崩溃
值得精读,展示了推测解码与 KV-Cache 量化组合时常见的配置传递遗漏问题,可作为类似集成场景的参考。
原始 PR · 作者 bigPYJ1151 · 合并时间 2026-05-15 18:58
修复 CPU 上 embedding 模型未编译的问题
该 PR 是 CPU 后端 embedding 模型性能退化的重要修复,建议精读并确认随机种子重置顺序是否需要调整。虽修复核心问题,但种子顺序可能引入隐藏的随机状态一致性问题,建议修复。
修复 Gemma3 多 GPU 下形状不匹配 bug
值得合并,属于典型的一行修复 bug。可快速批准合并。建议未来在类似 `RowParallelLinear` 的使用处也显式指定 `input_is_parallel` 参数以增强可读性。
参与讨论