AI-CUDA-Engineer-Archive データセットは、AI によって生成された CUDA カーネルのコレクションであり、オープンソース モデルのその後のトレーニングと、より優れた CUDA 関数モジュールの開発を容易にすることを目的としています。このデータセットには 30,000 を超える CUDA カーネルが含まれており、それらはすべて AI 主導の CUDA エンジニアによって生成され、そのうち 17,000 を超えるカーネルが正しいことが検証されており、約 50% のカーネルが PyTorch のネイティブ ランタイムよりも優れています。 これには、PyTorch リファレンス実装、PyTorch/NCU/Clang-tidy パフォーマンス分析データ、各タスクの複数のカーネル バージョン、エラー情報、PyTorch ネイティブおよびコンパイルされたランタイムとのパフォーマンス比較データが含まれています。

引用

@article{lange2025aicudaengineer、 タイトル={AI CUDA エンジニア: エージェント CUDA カーネルの発見、最適化、および構成} 著者={Lange, Robert Tjarko、Prasad, Aaditya、Sun, Qi、Faldor, Maxence、Tang, Yujin、Ha, David} ジャーナル={arXivプレプリント}、 年={2025} }