NVlabs/kda

何を解決するか

Kernel Design Agents (KDA) は、コーディングエージェントを用いてパフォーマンスに敏感なCUDAカーネルの研究、実装、最適化を自動化するための構造化されたワークフローを提供します。研究、検証、反復的改善を統合することで、高性能GPUカーネルの作成プロセスを簡素化します。

どう動くか

KDAはエージェント中心のワークフローであり、コーディングエージェントが特定のライフサイクルを経るよう導きます：タスク契約（目的と制約）の定義、計画の作成、そして小さな検証済みイテレーションでのカーネル実装。プロンプトテンプレートと専門的な「スキル」（例：ncu-report-skill と KernelWiki）を活用し、CUDA開発に必要なエージェント固有の知識とプロファイリングツールを提供します。

対象ユーザー

NVIDIA B200およびB300 GPUをターゲットとする、GPUパフォーマンス最適化およびCUDAカーネル開発に注力する開発者や研究者。

主な特徴

• 反復的検証：小さな段階的変更に注力し、即座に検証することで正しさとパフォーマンスを確保。
• 専門スキルの統合：ncu-report-skill によるプロファイリングや KernelWiki によるドメイン知識の統合。
• エージェント中心のワークフロー：ベンチマークハーネスやハードウェアターゲットに依存しない、タスク契約から最終的なプロモーションまでの一貫したアプローチを提供。
• コミュニティの希望リスト：最適化が必要なカーネルのコミュニティ主導のリストを維持。