arXiv論文メモ
新着一覧
cs.PF / cs.PL · 査読状況未確認

TritonとcuTileの性能差を診断するベンチマーク

TileBench: A Controlled Benchmark for Performance Evaluation and Bottleneck Diagnosis of Tile-Based Programming Models

Bowen Cui, Zhongchun Zhou, Hao Wu, Tejas Ramesh, Junyu Yin, Jialiang Gu, Keren Zhou

この論文をやさしく読む

ひとことで言うと

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
査読・掲載
査読状況未確認
arXivで読むPDF

更新履歴

取得できた版を表示。版の更新は査読済みを意味しません。過去版の本文差分は未解析です。

原文の要旨

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 / 要約の誤りについて