Kernel Forge: An Agent Harness for LLM-based Generation and Optimization of CUDA Kernels
Kernel Forge:用于基于大语言模型的CUDA内核生成与优化的代理框架
机构 * University of Michigan(密歇根大学)
专题命中 软件智能体 :agent(title);agentic(abstract);分类 cs.AI
AI总结 研究针对机器学习模型中计算内核优化需专家手写代码的问题,提出开源端到端代理框架Kernel Forge,它支持多种工作负载,用蒙特卡洛树搜索优化路径,经实验评估,优化后的内核性能优于PyTorch即时模式。
Comments The code is available at: https://github.com/TheJoshBrod/KernelForge