NVlabs/kda
Kernel Design Agents (KDA) is a agent-centric workflow to write high-performance CUDA Kernels.
解決的問題
Kernel Design Agents (KDA) 提供一個結構化的工作流程,讓開發者能使用編碼代理自動化研究、實作與優化性能敏感的 CUDA 內核。透過整合研究、驗證與迭代優化,簡化高效率 GPU 內核的開發流程。
如何運作
KDA 是一種以代理為中心的工作流程,引導編碼代理經歷特定的生命周期:定義任務合約(目標與限制)、草擬計畫,並以小規模、可驗證的迭代方式實作內核。它利用提示模板與專用的「技能」(例如 ncu-report-skill 和 KernelWiki),提供代理執行 CUDA 開發所需的特定知識與分析工具。
適用對象
專注於 GPU 性能優化與 CUDA 內核開發的開發者與研究人員,特別是針對 NVIDIA B200 與 B300 GPU 的場景。
主要亮點
- 迭代驗證:聚焦於小規模、逐步的變更,並立即驗證,以確保正確性與效能。
- 專用技能整合:整合
ncu-report-skill用於效能分析,以及KernelWiki用於領域知識。 - 以代理為中心的工作流程:提供從任務合約到最終發布的一致性結構化方法,獨立於任何單一基準測試框架或硬體目標。
- 社群願望清單:維護一個由社群主導的、需要優化的內核清單。
相關
- 專案
- 專案
- 專案
- 專案