概述

Triton 是为深度学习优化的 GPGPU 编程框架，简洁、高性能，使用 python 环境高效编写～

安装及使用

使用 pip。

uv pip install triton

import triton

@triton.jit
...

性能优化

调整线程块大小

vecadd: BLOCK_SIZE = 16

vecadd: BLOCK_SIZE = 32