跳到正文
GitHub · 灵魂体仓库·· 19 天前精选AI 评分68

kadirnar 开源 kernel-agent:让 Claude 自动编写 CUDA/Triton GPU kernel 加速模型推理

kadirnar/kernel-agent

来源与原文

GitHub · 灵魂体仓库

发布时间:(北京时间)

阅读原始来源 · github.com ↗
完整链接与内容说明

https://github.com/kadirnar/kernel-agent

本站目前仅有 AI 导读与摘要,未展示原文正文;技术细节、代码和完整结论请查看原始来源。

AI 导读

kadirnar 在 GitHub 发布 kernel-agent,输入 Hugging Face 模型 URL 后自动分析 GPU 瓶颈,由 Claude 用 CUDA C++、CuTe DSL、Triton 或 TileLang 编写自定义 kernel,并尝试 static caches、CUDA graphs 等模型级优化,候选方案需通过正确性校验与基准测试,仅保留输出一致且更快者。支持 LLM、STT、TTS 和扩散模型,内置 VoxCPM2 工作负载(案例测得 11.3× vs eager、7.7× vs torch.compile),图中 Qwen3-0.6B 实测 2.03×(1,532.4→754.9 ms)。

推荐理由

这个项目把正确性验证讲得很细,读者可以借此了解自动生成 GPU kernel 时如何防作弊和保证输出一致。

来源:GitHub · 灵魂体仓库 · github.com