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

页面历史