Prhub

vllm-project/vllm

A high-throughput and memory-efficient inference and serving engine for LLMs

监控状态:已开启 最近同步:2026-08-20 17:15 同步状态:空闲 下次计划:2026-08-20 18:15

PR 列表

更多筛选
2026-05-16

#42409 [ROCm] Widen AITER fused AR RMSNorm 1-stage gate

原始 PR · 作者 akii96 · 合并时间 2026-05-16 01:44

重构 重要性 5.38 洞察度 5.00

放宽 AITER 1-stage AR+RMS kernel 准入条件

推荐合并。PR 逻辑清晰、影响局部、收益明确,且经过维护者批准。无需深入精读,但可作为 ROCm 上 AITER 集成中与内核约束对齐的简洁示例。

功能 重要性 6.08 洞察度 4.00

MR v2 支持权重重载(sleep mode)

建议在完成 v2 完全迁移后,移除此委托方法并直接内联实现。同时应补充单元测试覆盖 `reload_weights` 调用后的缓存重置行为。

#41775 [Model Runner V2] FP32 gumbel sampling.

原始 PR · 作者 PatchouliTIS · 合并时间 2026-05-16 00:20

性能优化 重要性 7.10 洞察度 6.00

Gumbel 采样默认使用 FP32 以提升性能

值得精读学习如何在 Triton 内核中安全切换 FP32/FP64 并处理边界值;以及从环境变量演化到引擎标志的设计决策过程,体现了代码的健壮性和可维护性。

2026-05-15
功能 重要性 8.60 洞察度 6.00

为 gfx950 添加 FP8 ASM prefill 加速 MLA 后端

推荐精读该 PR,尤其关注:自动检测函数 `_fp8_mla_prefill_supported` 的容错设计、持久调度元数据预分配策略、以及 Workspace Manager 的集成方式。Review 讨论中 host-device sync 的消除技巧值得学习。

性能优化 重要性 8.40 洞察度 5.00

将 IR op 优先级和 torch wrap 设置移至 Worker 初始化时一次性完成

值得精读。展示了如何识别和消除运行时重复配置,是性能优化的典型模式。设计上新增 `set_default` 方法分离初始化与运行时逻辑,对类似问题有参考价值。

缺陷修复 重要性 6.34 洞察度 6.00

修复 MLA indexer 中 DeepGEMM context_lens 非连续问题

该 PR 值得精读,展示了在处理 CUDA graph 兼容性时如何避免动态内存分配的设计模式:通过预分配 1D 平坦缓冲区并在运行时用 view 重塑,而非调用 `.contiguous()`。同样的思路可应用于其他需要为外部 kernel 提供连续 tensor 的场合。

性能优化 重要性 5.75 洞察度 4.00

优化 MLA 注意力 _v_up_proj 的 bmm 效率

建议合并。该 PR 是一个清晰的性能微优化与代码清理,逻辑正确且风险极低。值得关注的是如何利用 `torch.bmm` 的 `out` 视图来避免额外复制,类似技巧可用于其他类似场景。

参与讨论