#26065 [XPU] fix correctness issue of GDN triton kernel for XPU
原始 PR · 作者 Xia-Weiwen · 合并时间 2026-05-25 13:18
修复 XPU 上 GDN kernel 长序列的正确性
值得精读,尤其是 `chunk_delta_h.py` 中的循环重构策略——将时间步设为外层循环有利于维护跨时间步的状态一致性,是 Triton 中复杂 kernel 的典型优化模式。review 中关于 A dtype 的讨论也值得关注,可作为后续精度增强的切入点。
参与讨论