首页 > 计算机科学

英伟达护城河被AI攻破,字节清华CUDAAgent,让人人能搓CUDA内核

机器学习算法与自然语言处理 2026-03-04 00:00
文章摘要
背景:GPU内核优化是深度学习的基础,但高度专业化,依赖人类专家。现有AI生成CUDA代码的方法多关注代码正确性,难以实现深度硬件级优化。研究目的:字节跳动与清华大学的研究团队旨在开发一个能像人类专家一样思考、通过强化学习直接基于硬件性能进行优化的AI智能体——CUDA Agent,以自动生成高性能CUDA内核。结论:CUDA Agent在KernelBench基准测试中表现优异,在简单到复杂任务上均显著超越torch.compile等基线,证明了基于硬件反馈的强化学习能使大模型内化“硬件直觉”,为全自动高性能计算基础设施开辟了新路径。
英伟达护城河被AI攻破,字节清华CUDAAgent,让人人能搓CUDA内核
本站注明稿件来源为其他媒体的文/图等稿件均为转载稿,本站转载出于非商业性的教育和科研之目的,并不意味着赞同其观点或证实其内容的真实性。如转载稿涉及版权等问题,请作者速来电或来函联系。
最新文章
沉寂两年后,Ilya创办的SSI获英伟达重注,投资额被曝达50亿美元
沉寂两年后,Ilya创办的SSI获英伟达重注,投资额被曝达50亿美元
两年不发产品,Ilya到底在秘密研究什么?
2026-07-28
老黄「开源协议」就剩一家没签,是谁啊好难猜啊
老黄「开源协议」就剩一家没签,是谁啊好难猜啊
A社,全美最不吃压力之人
2026-07-28
从DeepSeekmHC到xHC:如何高效scale残差流?
从DeepSeekmHC到xHC:如何高效scale残差流?
xHC首次把语言模型的残差流高效扩展到16流。在18B和28BMoE语言模型的预训练实验中,xHC在DeepSeekManifold-constrainedHyper-Connections(mHC)的基础上,把收益又提升了近一倍
2026-07-27
Book学术官方微信
Book学术官方微信
Book学术文献互助
Book学术文献互助群
群 号:604180095
Book学术
文献互助 智能选刊 最新文献 互助须知 联系我们:info@booksci.cn
Book学术提供免费学术资源搜索服务,方便国内外学者检索中英文文献。致力于提供最便捷和优质的服务体验。
Copyright © 2023 Book学术 All rights reserved.
ghs 京公网安备 11010802042870号 京ICP备2023020795号-1