#37835 [ROCm] Add UE8M0 scale packing for Triton silu_mul_quant
原始 PR · 作者 AndreasKaratzas · 合并时间 2026-08-19 23:20
ROCm Triton 量化路径新增 UE8M0 scale 打包支持
值得精读:一是“float32 中间计算 + 位运算打包 UE8M0”的两阶段思路,对理解 UE8M0 编码(float32 指数位提取、4 字节对齐、strided copy)很有价值;二是 Triton/C++ 双路径分叉的设备守卫模式,可作为平台差异化 kernel 实现的参考。对 ROCm/XPU 平台维护者尤其建议细看。评审中关于 `view` 保持维度与 `copy_` 非连续语义的交锋也值得留意,作者对 tensor 内存布局的理解是正确的。