Triton の使用体験とパフォーマンス分析

概要

Triton はディープラーニングのために最適化された GPGPU プログラミングフレームワークで、シンプルかつ高性能、Python 環境で効率的に記述できます~

インストールおよび使用方法

pip を使用します。

uv pip install triton

import triton

@triton.jit
...

パフォーマンス最適化

スレッドブロックサイズの調整

vecadd: BLOCK_SIZE = 16

vecadd: BLOCK_SIZE = 32