#46182 [Feat][1/N] CuTeDSL warmup infrastructure, FA4 MLA
原始 PR · 作者 LopezCastroRoberto · 合并时间 2026-07-01 03:17
引入 CuTeDSL 预热框架,避免 FA4 MLA 推理时 JIT 延迟峰值
建议开发者精读 `cutedsl_warmup.py` 的注册-收集-编译流程和 `fa4_cutedsl_config.py` 的编译 spec 生成逻辑,provider 模式可复用于其他 JIT 重内核。注意配置默认关闭,后续 PR 可能默认启用并补充测试。
参与讨论