gpu-mode/reference-kernels

解決する問題

このリポジトリは、GPUカーネルコンペティションの参照実装を一元的に集めたものです。KernelBotがGPU Modeコミュニティ（//discord.gg/gpumode）で主催するコンペティションに参加する人々が、自身のカスタムGPUカーネルの正しさとパフォーマンスを評価するための基準を確保します。

動作方法

プロジェクトは、特定のディレクトリに問題を整理しています（例：PMPPの練習問題、AMDおよびNVIDIAのコンペティション、BioMLカーネル、大学の課題など）。各問題には、PyTorchの参照実装（reference.py）、テストケースの生成用の問題仕様（task.yml）、および入出力のスキーマ（task.py）が含まれています。

対象者

ハイパフォーマンスコンピューティング（HPC）およびGPUプログラミングに特化した開発者や研究者、スタンフォードやプリンストン大学などの大学で並列プログラミングの授業を受ける学生。

特徴

• AMDおよびNVIDIA GPUを含む、さまざまなハードウェアプラットフォーム用の独自の参照実装。
• NVIDIA Nsight Computeとの統合をサポートし、プロファイリングと検証が可能。
• 大学の並列プログラミング授業用のインフラを提供。
• 線形代数からBioMLカーネルまで、多様な問題セットを含む。