张大妈

挑战英伟达CUDA护城河:华人团队实现AI自动寻优,矩阵乘法核心首超CUDA闭源库

源自公众号:DeepTech深科技

01-27 14:07

一项名为CUDA-L2的开源AI系统,通过结合大模型与强化学习,在英伟达GPU上自动寻优矩阵乘法核心,其性能首次超越了英伟达官方的闭源库cuBLAS。这项技术突破有望直接提升大模型运行效率,对打破行业技术壁垒具有重要意义。

挑战英伟达CUDA护城河:华人团队实现AI自动寻优,矩阵乘法核心首超CUDA闭源库智能速览

  • CUDA-L2在A100 GPU上比英伟达cuBLAS快19%至26%。

  • 系统利用AI模型和强化学习,自动搜索最优程序实现方案。

  • 矩阵乘法性能提升将成比例加快大模型的运行效率。

  • 成果已开源,旨在推动工业界实际应用并打破技术壁垒。

挑战英伟达CUDA护城河:华人团队实现AI自动寻优,矩阵乘法核心首超CUDA闭源库精华内容

矩阵乘法是AI计算的基石,其优化难度极高。那么,一个AI系统是如何实现超越顶尖人工优化的呢?

AI计算的基石

矩阵乘法是AI模型中最基础且最耗时的计算操作,几乎构成了所有智能计算的基石,从图像生成到语言理解都离不开它。然而,在GPU上高效执行这一操作极具挑战,需要工程师针对不同硬件和问题规模进行复杂且耗时的手动调整,以合理分配计算任务并优化数据流。

这项工作的核心难点在于,如何充分利用GPU上成千上万的计算核心,并让数据在高速但有限的内存间高效流转,不同规模的矩阵需要完全不同的优化策略。这长期是顶级工程师的专属领域。

AI担当设计师

CUDA-L2系统引入了一个全新的设计师:一个经过海量代码训练的大模型。该模型本身就具备丰富的编程知识,但要成为优化大师,还需强化学习的训练。

系统不会直接教授模型如何编写,而是设定一个以实际运行速度为目标的奖励机制。模型尝试生成不同的计算程序,系统通过实际运行来打分,运行越快分数越高,反之则扣分。经过数百万次的尝试,模型逐渐摸索出编写超高速矩阵乘法程序的秘诀,甚至能根据硬件性能报告进行微调。

超越与开源的意义

在A100高性能GPU上的系统性评测中,CUDA-L2生成的程序比英伟达官方的cuBLAS库快19%,在模拟真实服务器环境中优势扩大至26%。

这一性能提升的直接影响是巨大的。大模型超过一半的GPU计算时间都消耗在矩阵乘法上,因此其性能提升将成比例地加速大模型运行。据估算,矩阵乘法速度提升10%,可能为全球GDP带来约4400亿美元的增长。更重要的是,这项成果已开源,为打破技术壁垒、促进整个行业进步提供了可能。

CUDA-L2的出现展示了AI优化底层代码的巨大潜力,为算力瓶颈提供了新的解决方案。随着这类技术的成熟,未来的AI应用是否将迎来一轮全面的效率革命?

内容由AI生成
0
扫一下,分享更方便,购买更轻松
0评论

当前文章无评论,是时候发表评论了
提示信息

取消
确认
评论举报

最新文章 热门文章