ITBEAR科技资讯
网站首页 科技资讯 财经资讯 分享好友

字节跳动与清华AIR携手:CUDA Agent助力大模型编写超速GPU内核

时间:2026-08-18 13:45:00来源:互联网编辑:快讯

字节跳动旗下Seed团队与清华大学智能产业研究院(AIR)联合宣布,推出面向GPU内核优化的智能体强化学习系统CUDA Agent。该系统通过将大型语言模型深度融入CUDA开发环境,实现了底层代码生成效率的突破性提升,为AI基础设施优化提供了全新解决方案。

传统大模型在生成CUDA代码时面临效率瓶颈。以KernelBench基准测试数据为例,基础模型Seed1.6虽能完成74.0%的任务,但仅有27.2%的生成内核运行速度优于torch.compile编译器,整体几何平均速度甚至落后于传统编译方案。这种"语法正确但性能不足"的矛盾,制约了AI在底层代码优化领域的应用潜力。

研发团队构建了包含性能分析工具、正确性验证模块和安全沙盒的闭环训练环境。通过近端策略优化(PPO)算法实施150步深度强化训练,CUDA Agent在250项基准测试中取得显著突破:任务通过率跃升至98.8%,其中96.8%的生成内核运行速度超越torch.compile,平均加速比达到传统方案的1.8倍。这种性能跃升源于系统对内存访问模式、线程调度策略等底层细节的精准优化能力。

在技术开放方面,研究团队已发布包含6000个优化样本的CUDA-Agent-Ops-6K数据集,同步公开SKILL.md技术规范及训练配方文档。虽然完整模型权重暂未开源,但开放的组件已覆盖从数据构建到奖励函数设计的完整技术链条,为开发者提供了可复现的研究框架。该成果在自动驾驶实时感知、量化交易低延迟策略等对响应速度要求严苛的场景中具有直接应用价值,相关技术正在与多家行业头部企业开展联合验证。

更多热门内容
工控电控新蓝海:机器人进厂潮下,控制大脑与软件成核心赛道
控制大脑与运动控制,恰是国内工控电控厂商的主场,把伺服、PLC、运动控制卡封装成"机器人控制大脑+技能平台",比造机身更顺手。 回看小米机器人上岗,自动化工控的戏码变了:上半场拼设备国产化,下半场拼控制大脑与…

2026-08-18

2026年中国航天再提速:长征十二号助力星网组网,卫星互联网迈入新阶段
往前数12天,8月4日,同样是长征十二号,同样在海南,同样是一组卫星被送上了天。 千帆星座在酒泉和太原也是连续发射,38颗卫星被送上了天。 长征十二号火箭的运载能力在近地轨道达到12吨以上,是专为高密度组网…

2026-08-18

马斯克抢先占轨道?中国工业发力海南“下饺子” 太空竞赛下半场谁主沉浮
西方媒体现在只要一提到太空低轨卫星,就把马斯克捧成不可战胜的神,仿佛他的星链已经把近地轨道全占满了,中国只能在后头吃灰。 马斯克是在火箭运输端做减法,而中国航天走的是一条更绝的路——在卫星制造和供应链上做乘…

2026-08-18

朱雀三号遥二将发射 陆地垂直回收助力我国卫星产业新突破
若试验成功,这将是我国民营商业航天首款实现轨道级一子级陆地回收的液体运载火箭,华源证券指出,产业估值逻辑有望从"概念想象"切换至"可复用火箭产业化验证"。国联民生数据显示,全球低轨卫星轨道申报总量已超130万…

2026-08-18