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.
uv pip install triton
import triton
@triton.jit
...
Performance Optimization
Adjust Thread Block Size
vecadd: BLOCK_SIZE = 16
vecadd: BLOCK_SIZE = 32
