#30838 [JIT] Refactor dtype traits into DTypeTrait and unify warp reductions
原始 PR · 作者 DarkSharpness · 合并时间 2026-07-18 10:07
重构 JIT kernel 的 dtype trait 和 warp reduction 系统
该 PR 值得精读,尤其关注:1) hipcc 兼容性技巧(SFINAE vs requires-expression);2) `redux.sync` 指令的使用条件和性能收益;3) `utils/` 包拆分的设计模式(如何保持兼容性);4) arch-specific JIT target 的实现(`_cuda_arch_suffix` 的细节)。建议所有 JIT kernel 相关开发者阅读并同步调整代码。
参与讨论