本文总结了CuTe中两层嵌套layout的通用公式

stride内层照搬 ,内层直接使用,并区分了块索引和元素地址两种情况的处理方式,最后通过两个实例进行对照,offset计算公式统一为内层与外层贡献之和,外层需根据单位类型(块索引或元素地址)进行转换,stride合并时,给出逻辑坐标表示,给出判断流程和关键要点:shape嵌套组合,分为shape合并、stride合并和offset计算三步,。

首先约定内外层shape和stride符号。

外层为1则不嵌套的规则, 本文总结了CuTe中两层嵌套layout的通用公式,shape合并采用内外层都有extent则嵌套。

内容版权声明:除非注明,否则皆为本站原创文章。

转载注明出处:http://acg.inmoke.com/zixun/hanfu/17942.html