从这里开始
这个专题计划从 Triton 的编程模型出发,逐步编写基础 GPU 算子,并练习验证与性能比较。下面各节目前是内容提要。
编程模型
认识 Triton Program、张量块与基础计算的表达方式。
数据布局与访存
练习加载、掩码、计算与存储,并观察块形状的影响。
第一个算子
从简单运算开始,记录算子实现和正确性验证的流程。
性能比较
逐步补充基准测试与不同实现方式的性能对照。
这个专题计划从 Triton 的编程模型出发,逐步编写基础 GPU 算子,并练习验证与性能比较。下面各节目前是内容提要。
认识 Triton Program、张量块与基础计算的表达方式。
练习加载、掩码、计算与存储,并观察块形状的影响。
从简单运算开始,记录算子实现和正确性验证的流程。
逐步补充基准测试与不同实现方式的性能对照。