深度解析 CuTeDSL Swizzle 布局与 M 参数计算原理

snowclipsed · x · 2026-09-02

推文探讨了 CuTeDSL 中 <3,3,3> swizzle 与 (8,64) block 的组合原因。引用的博客解释了在 8 行 64 个 bf16 元素且 16 字节对齐的布局下,B 和 S 的计算逻辑。针对 Ampere 架构示例中 M=3 而非 4 的问题,解释了 CuTeDSL 在布局偏移单元而非直接字节地址上推理 M,因此指针偏移是基于 layout codomain 的元素。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →