Triton是一种基于Python的并行编程语言,专为高效编写DNN计算内核并在现代GPU上运行而设计。本文介绍了支持FP16和FP8数据类型的矩阵乘法持久化内核实现,并提供多种矩阵乘法方法,用户可通过命令行参数灵活指定矩阵维度和迭代步骤。
完成下面两步后,将自动完成登录并继续当前操作。