Hacker News· qwertyforce·· 3 小时前AI 评分28
Triton 编程语言与编译器入门笔记:从 Tile、Program 到 GPU kernel 启动
Triton Notes
AI 导读
OpenAI 最初开发的 Triton 是一门让自定义 GPU kernel 更易编写的编程语言与编译器,用块(tile)级抽象替代 CUDA 的逐线程管理,编译器负责把块操作映射到 GPU 线程。笔记以向量加法和矩阵乘法两个 kernel 为例,讲解 host 端校验、grid 设置与 kernel 启动,以及 tile 和 program 两个核心抽象。
来源:Hacker News · junupark.xyz