Skip to content

Triton Usage Experience and Performance Analysis

作者:Lee 等
最后更新:

Overview ​

Triton is a GPGPU programming framework optimized for deep learning. Simple and high-performance, efficiently developed using the Python environment~

Installation and Usage ​

Use pip.

bash
uv pip install triton
python
import triton

@triton.jit
...

Performance Optimization ​

Adjust Thread Block Size ​

vecadd: BLOCK_SIZE = 16 ​

vecadd: BLOCK_SIZE = 32 ​

页面历史