研究人员推出了 KernelMosaic,这是一个开放数据集和代理框架,旨在改进大型语言模型生成 CUDA 核的能力。该数据集将多样化的 CUDA 任务与经过验证的解决方案和优化轨迹配对,通过人工审查和面向多样性的选择进行策划。代理框架利用可重用的 CUDA 指导和执行反馈,引导 LLM 完成实现、修复和性能优化。在 KernelMosaic 上微调一个小型开源模型,显著提高了在 KernelBench 基准测试上的正确性,并增加了比 PyTorch eager 更快的核任务的比例,证明了该数据集在增强 CUDA 核生成能力方面的有效性。 AI
影响 增强了 LLM 在生成专用代码方面的能力,可能加速 GPU 计算密集型领域的发展。
排序理由 该集群描述了一篇介绍用于 AI 模型训练的数据集和框架的新学术论文。[lever_c_demoted from research: ic=1 ai=1.0]
- alphaXiv
- arXiv
- CatalyzeX
- CUDA
- DagsHub
- Gotit.pub
- Hugging Face
- IArxiv
- KernelBench
- KernelMosaic
- PyTorch
- ScienceCast
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →