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-skillKernelWiki),提供代理執行 CUDA 開發所需的特定知識與分析工具。

適用對象

專注於 GPU 性能優化與 CUDA 內核開發的開發者與研究人員,特別是針對 NVIDIA B200 與 B300 GPU 的場景。

主要亮點

  • 迭代驗證:聚焦於小規模、逐步的變更,並立即驗證,以確保正確性與效能。
  • 專用技能整合:整合 ncu-report-skill 用於效能分析,以及 KernelWiki 用於領域知識。
  • 以代理為中心的工作流程:提供從任務合約到最終發布的一致性結構化方法,獨立於任何單一基準測試框架或硬體目標。
  • 社群願望清單:維護一個由社群主導的、需要優化的內核清單。

相關

  • 專案
  • 專案
  • 專案
  • 專案