本文介绍Triton张量描述符,用于描述张量内存布局、简化GEMM核实现,并在SM90+ GPU上支持TMA操作。描述符分为设备端和主机端两类,包含基地址、形状、步长和块大小等元数据。作者分别用原始指针、设备描述符和主机描述符三种方式实现GEMM并测试,结果均正确。在RTX 5080上,大尺寸矩阵中描述符版本略快于原始指针,8192³时主机描述符约快7%。
完成下面两步后,将自动完成登录并继续当前操作。