AI Infra 动态 20 条
主流训推框架主分支 PR,按前沿价值与性能影响降序。
[https://nvbugs/6700677][fix] Fix Gemma4 startup on non-SM100 GPUs
本日合并至 main
· 10-21 22:45 UTC+8 · @lfr-0531
Gemma4 现在为异构与旧版 Transformers 配置逐层解析 head dim 和 KV-head 数,把解析结果直接传入 QKNormRoPEAttention,KV-cache 容量估算与 attention metadata 也改用逐层几何信息。价值在于修复非 SM100 系列 GPU 上的启动失败,同时保持非 SM100f 上 FlashInfer FA2 的默认路径与 SM100f 路由不变,提升多配置兼容性与显存估算正确性。
推测解码
算子与内核控制流优化硬件适配
[Doc] cookbook: Add GLM-5.3 AgentX section and refresh with eight open InferenceX PRs
本日合并至 main
· 10-21 13:22 UTC+8 · @nvpohanh
在 cookbook 的 GLM-5.3 页面新增 AgentX 长上下文部署小节,镜像 SemiAnalysis InferenceX 的 GLM-5.3 NVFP4 B300 SGLang MTP 启动命令,并叠加八个尚未合并的 InferenceX PR 中按并发档位细调的配置。价值在于把已验证的 AgentX 部署配置整理成可直接复制的启动参数,便于复现与调优;该条目为文档改动。
推测解码
控制流优化硬件适配性能优化
[Doc] Kimi-K3: DCP selector and TOKENSPEED_MLA draft backend on GB200/GB300
本日合并至 main
· 10-21 12:20 UTC+8 · @hshrivastava-droid
在 Kimi-K3 的 GB200/GB300 交互式 recipe 中,spec decoding 开关改为在 --speculative-config 内写入 TOKENSPEED_MLA draft 后端,顶层 --attention-backend 仍只作用于目标模型。价值在于让 draft 模型在 Blackwell 上使用 TOKENSPEED MLA 后端,而 B200、B300、AMD 与 Ascend 配置保持不变。
推测解码
控制流优化硬件适配服务与推理
[Fix] spec: Take post-verify KV move targets from the verify window
本日合并至 main
· 10-21 12:12 UTC+8 · @hnyls2002
accept token 的 KV 搬移目标改为从各请求 verify window 前端的 out cache loc 读取,不再在 verify forward 之后重新读取 req to token。价值在于消除与 overlap scheduler 的竞态:请求在 EOS 或停止条件后行被交给新请求时,KV 会被写进新拥有者(包括 radix cache 前缀)的槽位并污染后续命中;影响 CUDA 上的 NGRAM 与 UNO tree verify。
推测解码
控制流优化硬件适配服务与推理
[Experimental] Qwen 3.5 gated attention block (FROST, cuDNN Frontend) for Rubin
本日合并至 main
· 10-21 10:02 UTC+8 · @RomanAnders90
把 Qwen 3.5 的 full-attention 子层封装为单个实验性 FlashInfer API,后端是从 cuDNN Frontend 引入的 CuTe DSL FROST kernel,一次调用完成 QKV+gate 投影、带部分 RoPE 的 QK-RMSNorm、paged-KV 写穿、head dim 256 的 GQA、sigmoid 输出门与 out 投影,并针对三种 serving 形态分别编译 plan。价值在于为 Rubin 提供整块融合的 attention 算子,减少 kernel 启动与中间访存。
推测解码
算子与内核控制流优化硬件适配
[Doc] cookbook: add TPU7x recipe for GLM-5.3-Flash
本日合并至 main
· 10-21 08:10 UTC+8 · @Bob-Chen222
在 cookbook 中为 GLM-5.3-Flash 增加 TPU7x 硬件条目与一个已验证的 Low Latency FP8 配置(单 TPU7x 主机 4 芯片、TP8,DSA 走 Pallas sparse MLA kernel,MTP 5/1/6 链式 spec worker),并对仅 GPU 可用的选项加上 TPU 守卫。价值在于让 TPU 后端有可复现的部署配方,并避免误选 CUDA/ROCm 专属开关。
推测解码
控制流优化硬件适配性能优化
[Bugfix][DeepSeek V4] Load redundant EPLB expert slots
本日合并至 main
· 10-21 05:16 UTC+8 · @simon-veitner-redhat
get_expert_mapping 现在会传入 num_redundant_experts,使 load_weights 在 FusedMoE 路径上真正填充冗余 EPLB 专家槽位(deep_gemm_mega_moe 路径此前已加载副本)。价值在于修复 --enable-eplb 且冗余专家数非 0 时槽位保留未初始化内存的问题,保证 EPLB 副本权重正确,且仅影响 NVIDIA 的 V4 与 V4.1。
推测解码
硬件适配性能优化分布式训练
[Doc] Document the DeepSeek-V4.1-Flash B200 AgentX profile in the guide and command builder
本日合并至 main
· 10-21 03:45 UTC+8 · @Ankur-singh
在 DeepSeek-V4.1-Flash 的 recipe 中,把 B200 vLLM AgentX lane 的全部 17 个合格配置点写入指南与命令构建器,覆盖 TP4、TP2、DEP2、DEP4 下逐并发的 --max-num-seqs、batched tokens、capture 与 decode-replay graph 参数。价值在于把测量过的部署配置固化为可复制的启动参数,便于按并发档位复现与调优;该条目为文档改动。
推测解码
算子与内核硬件适配性能优化
[AMD] Add DeepSeek-V4.1-Flash MI35x nightly accuracy test
本日更新,目标 main · 10-21 02:17 UTC+8 · @michaelzhang-ai
新增 MI35x 上的 GSM8K 8-shot 精度测试,服务参数取自 cookbook 的 MI350X High-Throughput cell,包括 TP4/EP4、DSPARK 投机与 cuda-graph-max-bs-decode 256 等。价值在于为 AMD MI35x 的 FP4 推理路径提供 CI 覆盖,填补此前该模型没有 AMD 测试的空白。
推测解码
硬件适配性能优化分布式训练
Sync tied embedding copies under meta-device init
本日合并至 main
· 10-21 02:17 UTC+8 · @tdene
修复 meta device 初始化下 tied embedding 副本不同步的问题:此前本应让副本取值一致的全归约被跳过,该改动使副本在 FSDP 场景下也能同步。价值在于避免 --use-megatron-fsdp 配合 --init-model-with-meta-device 以及 PP1 或 MTP 做预训练时绑定权重失配。
推测解码
算子与内核硬件适配分布式训练
[AMD] Add Kimi-K2, DeepSeek-V3.2 tests to nightly CI
本日更新,目标 main · 10-21 02:16 UTC+8 · @michaelzhang-ai
为 MI325(MI30x)新增 DeepSeek-V3.2 基础与 MTP 精度+性能测试、Kimi-K2-Instruct-0905 精度测试,同时更新 MI35x 测试、合并测试任务并修复若干 CI 失败。价值在于扩充 AMD nightly 的覆盖范围并稳定测试流程,使 AMD 平台上的精度与性能回归能更早暴露。
推测解码
算子与内核性能优化服务与推理
fix(gdn): fence TMEM loads before cg1 shared_acc releases
本日合并至 main
· 10-21 01:06 UTC+8 · @dundysm
在 Blackwell GDN chunked kernel 的 cg1 shared acc 释放前插入 cute.arch.fence_view_async_tmem_load(),使随后的 SYNCS.ARRIVE 继承 LDTM scoreboard 等待掩码,避免 tcgen05.mma 覆盖仍在被读取的 TMEM 单元。价值在于消除 tmem_cg1_shared_acc_stages=1 时 producer 与 consumer 的竞态,提升 kernel 的正确性与数值稳定性。
推测解码
算子与内核硬件适配性能优化
[Docs] Point users at --disable-flashinfer-autotune
本日合并至 main
· 10-21 22:55 UTC+8 · @mgoin
把文档、recipe、测试与评测配置中的 --no-enable-flashinfer-autotune 和 kernel-config 写法统一替换为推荐 flag --disable-flashinfer-autotune,并在帮助文本与启动日志中补充说明与提示。价值在于减少用户复制过时参数的问题,并让关闭 autotune 以加快启动的路径更容易被采用。
推测解码
算子与内核性能优化
[XPU] Enable spec-decode adaptive verification without a confidence head
本日合并至 main
· 10-21 22:19 UTC+8 · @yma11
把无需 confidence head 的自适应验证(承接 52228)在 XPU 路径上启用,使 draft 验证长度可动态调整。价值在于让 Intel Arc Pro B60 多卡 TP8 的 DFlash2 场景获得加速,同时指出 FA kernel 仍存在性能差距,可在后续 PR 中解决。
推测解码
硬件适配服务与推理
[Bugfix] Fix FlashInfer non-causal draft attention (DFlash/DSpark) on Blackwell
本日更新,目标 main · 10-21 17:57 UTC+8 · @mgoin
修复 Blackwell 上 FlashInfer 非因果 draft attention 在 FULL-graph capture 下重放结果错误的问题:该 prefill wrapper 只有以 use_cuda_graph=True 与 persistent buffer 构造时才可安全重放,因此调整 SM100 上的 graph 支持级别,让该路径保持 eager。价值在于解决 DFlash/DSpark draft 在 Blackwell 上接受率接近零或出现非法内存访问的问题,而 SM90 的行为不变。
推测解码
算子与内核硬件适配
[CPU] Improve long-context MTP attention scheduling on AMX
本日合并至 main
· 10-21 12:18 UTC+8 · @tianmu-li
把 CPU attention 的 GQA 分组调度扩展到符合条件的因果多 token 请求(含 MTP 验证与短 prefill),按每个请求的 query 长度、因果性与 AMX 几何评估分组 query head 和 split-KV 工作量,再由成本模型选出方案。价值在于改善长上下文 MTP 在 AMX 上的调度效率,并适用于 auto、FP8 E4M3 与 FP8 E5M2 KV cache。
推测解码
控制流优化性能优化
[AMD] Register 8 CPU-bound unit tests for AMD 1-GPU PR CI
本日更新,目标 main · 10-21 02:17 UTC+8 · @michaelzhang-ai
为 AMD 的 stage-b-test-1-gpu-small-amd PR CI 注册 8 个只用纯 Python/CPU torch 算子的单元测试(这些测试此前已在 NVIDIA per-commit CI 运行),使该套件从 88 个增至 96 个测试。价值在于在不引入 FlashInfer、FA3、ModelOpt/Marlin 量化与 EAGLE 依赖的前提下扩大 AMD CI 覆盖。
推测解码
算子与内核分布式训练
[Core] Don't apply spec decode async scheduling fallbacks with MRV2
本日合并至 main
· 10-21 01:41 UTC+8 · @njhill
把与 spec decode 相关的 async scheduling 限制(CPU ngram、suffix decoding、disable_padded_drafter_batch=True)与已有的 V1+PP 检查合并为单一的 V1-only 不兼容判定,该判定同时用于默认回退与显式开启时的校验,MRV2 下不再强制关闭或报错。价值在于修正 ngram 等场景下 async scheduling 被默认错误禁用的问题。
推测解码
控制流优化服务与推理
[Bugfix][Structured Output] Feed the token that implicitly ends reasoning to the grammar
本日更新,目标 main · 10-21 17:29 UTC+8 · @simon-veitner-redhat
对 ParserEngine 系解析器,把同时属于内容的 reasoning-end token 也喂给 grammar(与 extract_content_ids 的行为一致),闭合标签仍被跳过。价值在于修复 GLM 等模型重复输出结束标记、required/named tool call 丢失的问题,提升结构化输出在投机解码下的正确性。
推测解码
服务与推理
[Fix] observability: Correct speculative FPM timing and decode lengths
本日合并至 main
· 10-21 15:31 UTC+8 · @tedzhouhk
让每条结果保留自身的计时区间与不可变的 count/sum/variance,不再合并重叠迭代或读取缺失、过期的 CPU 长度镜像;无本地请求的 attention-DP rank 保留其 dummy forward 记录,idle heartbeat 跟随 scheduler 状态。价值在于修正投机解码的 FPM 计时归属与 decode 长度统计,避免长 forward 因发布队列安静而被误判为空闲。
推测解码
服务与推理