CuTe线程值布局

Lei Mao's Log Book Lei Mao's Log Book ·

CuTe线程值布局(TV布局)用于将张量划分为小张量,使线程组中的每个线程按照特定模式访问数据。TV布局将线程与数据坐标映射,便于分块复制和MMA操作。反向TV布局则将数据元素坐标映射到线程索引,以验证数据访问模式。CuTe TV分区通过计算线程坐标简化了数据访问过程。

原文英文,约800词,阅读约需3分钟。
阅读原文