Kernel-Bench

KernelFoundry框架为LLM优化GPU核

研究人员开发了KernelFoundry，一个旨在优化大型语言模型GPU核的演化式框架。该系统利用MAP-Elites进行质量多样性搜索，通过元提示演化发现任务特定的优化策略，并进行基于模板的参数调优以适应硬件和输入。KernelFoundry在SYCL核的KernelBench基准测试中持续优于基线方法，平均提速2.3倍，并能生成CUDA核。