TritonとcuTileの性能差を診断するベンチマーク
TileBench: A Controlled Benchmark for Performance Evaluation and Bottleneck Diagnosis of Tile-Based Programming Models
この論文をやさしく読む
ひとことで言うと
B200 GPU上で45演算をそろえて測り、TritonとcuTileの性能差が処理の種類で変わると示した。
何に役立つ?
GPUカーネルの実装方式や調整方法を選び、遅い原因を診断する際の比較基盤になる。
この研究の面白いところ
参照実装、検証済み実装、自動調整、ルーフライン指標をそろえ、LLM生成カーネルの比較も行った。
どこまで分かった?
結果はNVIDIA B200上での評価であり、他のGPUでも同じ優劣になるとは要旨からは言えない。
v1のアブストラクトに基づくAI解説。日本語訳とは別に、用途の解釈を含みます。
アブストラクトの日本語訳
TritonやcuTileのようなタイル単位のプログラミングモデルは、高性能カーネルの開発を簡単にすることを目指すが、実際の性能、調整時の挙動、使いやすさを体系的に比較することは難しい。本研究は、NVIDIA B200 GPU上で、演算の意味と実装構造をそろえてTritonとcuTileを評価するベンチマークTileBenchを提示する。 TileBenchには、AI用カーネルの多様なパターンとメモリ・計算特性をカバーする45の演算が含まれる。各課題にはPyTorchによる参照実装、検証済みのTritonとcuTileの実装、標準化されたデータ型と入力サイズの系列、初期設定と自動調整設定、ルーフラインに基づく指標、プロファイリングによる原因診断が用意される。評価では、性能差は処理内容に依存した。cuTileはTensor CoreとTMAに適した少数のカーネル群で優れ、Tritonは不規則、ストリーミング型、メモリ帯域に制約される多くの演算で強かった。LLMが生成した両言語のカーネルも評価し、同じ反復改良手順ではTritonの方が一貫して使用トークン当たりの効率が高いと分かった。TileBenchは公開されている。
v1の要旨から自動生成。本文の精読・人による確認は未実施。
- 初稿
- 2026-09-24(UTC)
- 最新改訂
- 2026-09-24 · v1
- 査読・掲載
- 査読状況未確認
更新履歴
- v1 2026-09-24 この版を読む
取得できた版を表示。版の更新は査読済みを意味しません。過去版の本文差分は未解析です。
原文の要旨
Tile-based programming models, such as Triton and cuTile, aim to simplify high-performance kernel development, but their practical performance, tuning behavior, and usability remain difficult to compare systematically. We present TileBench, a controlled benchmark for evaluating Triton and cuTile on NVIDIA B200 GPUs under matched operator semantics and comparable implementation structures. TileBench contains 45 operators covering diverse AI-kernel patterns and memory/computation behaviors. Each task provides a PyTorch reference, verified Triton and cuTile implementations, standardized data-types (dtype) and input-size sweeps, default and autotuned configurations, roofline-based metrics, and profiling-guided diagnosis. Our evaluation shows that performance gaps are workload-dependent: cuTile excels on a small cluster of Tensor-Core/TMA-friendly kernels, while Triton is stronger on many irregular, streaming, and bandwidth-bound operators. We further evaluate LLM-generated cuTile and Triton kernels and find that Triton is consistently more token-efficient than cuTile under the same iterative refinement protocol. TileBench is publicly available at https://github.com/Deep-Learning-Profiling-Tools/Tilebench.
arXiv ID: 2609.29067 / 要約の誤りについて