Agentic CUDA Kernel Optimizer - 自動でCUDAカーネルを最適化
Agentic CUDA Kernel Optimizerは、ワークロードの説明からGPU実装を自動生成し、コード生成、正しさの検証、ベンチマーク、改良のサイクルを回すツールです。LangGraphを基盤に、エージェントがカーネル実装と起動設定を探索し、NVIDIAのドキュメントを参照したり、Nsight Computeのカウンタを分析して次の実験に活かします。各実験は記録され、最速の検証済み実装が保持されます。C++ハーネスがNVRTCでコンパイルし、CUDA Driver APIで起動。Pythonが比較と候補選択を担当します。
HNでの議論
11- fooblaster
なぜこれをClaude codeやcodexへの単一のプロンプトでより上手く達成できないのか、誰か説明してくれませんか？理解できていない気がします。
- generalizations
とてもクールですね。karpathy autoresearchも試してみましたか？これと比べてどう思いますか？
- asamadx
これをソロで出荷したことに敬意を表します。これほど技術的なプロジェクトは、通常、正しさを検証するだけでもチームが必要です。カーネルバリアント間の回帰テストはどのように処理していますか？エージェント型オプティマイザの最も難しい部分は、より速いカーネルを見つけることではなく、より速いものが静かに何かを壊していないことを証明することのように感じます。