网易首页 > 网易号 > 正文 申请入驻

英伟达自毁CUDA门槛!15行Python写GPU内核,性能匹敌200行C++

0
分享至

梦晨 发自 凹非寺
量子位 | 公众号 QbitAI

GPU编程变天了。

英伟达发布最新版CUDA 13.1,官方直接定性:这是自2006年诞生以来最大的进步

核心变化是推出全新的CUDA Tile编程模型,让开发者可以用Python写GPU内核,15行代码就能达到200行CUDA C++代码的性能。



消息一出,芯片界传奇人物Jim Keller立即发问:

  • 英伟达是不是亲手终结了CUDA的“护城河”?如果英伟达也转向Tile模型,AI内核将更容易移植到其他硬件上。



Jim Keller参与设计过AMD Zen架构、苹果A系列芯片、特斯拉自动驾驶芯片的”硅仙人”,他的判断在行业里相当有分量。

那么问题来了:CUDA这次到底改了什么?为什么会被认为是”自毁长城”?

GPU编程范式从“线程”到“瓦片”

要理解这次更新的意义,得先回顾一下传统CUDA编程有多折磨人。

过去20年,CUDA一直采用SIMT(单指令多线程)模型,开发者写代码时,需要手动管理线程索引、线程块、共享内存布局、线程同步,每一个细节都要自己操心。

想要充分利用GPU性能,特别是用上Tensor Core这类专用模块,更是需要深厚的经验积累。

CUDA Tile彻底改变了这套玩法:

开发者不再需要逐线程地编写执行路径,而是把数据组织成Tile(瓦片),然后定义在这些Tile上执行什么运算。至于怎么把这些运算映射到GPU的线程、Warp和Tensor Core上,编译器和运行时会自动搞定。

就好像NumPy之于Python。



英伟达为此构建了两个核心组件:

CUDA Tile IR是一套全新的虚拟指令集,它在高级语言和硬件之间加了一层抽象,确保基于Tile编写的代码能在不同代际的GPU上运行,从当前的Blackwell到未来的架构都能兼容。

cuTile Python则是面向开发者的接口,直接用Python写GPU内核,门槛一下子从“HPC专家“降到了”会写Python的数据科学家都可以干。



另外,这次更新还带来了一系列面向Blackwell的性能优化:

  • cuBLAS引入了FP64和FP32精度在Tensor Core上的仿真功能
  • 新增的Grouped GEMM API在MoE(混合专家模型)场景下能实现高达4倍加速
  • cuSOLVER的批处理特征分解在Blackwell RTX PRO 6000上相比L40S实现了约2倍的性能提升
  • 开发者工具Nsight Compute新增了对CUDA Tile内核的性能分析支持,可以把性能指标直接映射回cuTile Python源代码。

目前CUDA Tile仅支持Blackwell架构(计算能力10.x和12.x),开发重点集中在AI算法上。英伟达表示未来会扩展到更多架构,并推出C++实现。

硅仙人的质疑:降低门槛是一把双刃剑

那么Jim Keller为什么说英伟达可能”终结了自己的护城河”?

关键就在于Tile编程模型不是英伟达独有的。AMD、Intel以及其他AI芯片厂商的硬件,在底层架构上同样可以支持基于Tile的编程抽象。

过去CUDA难以移植,很大程度上是因为SIMT模型与英伟达硬件深度绑定,开发者要针对具体的GPU架构手写优化代码。这些代码换到别家硬件上,要么跑不了,要么性能大打折扣。

但Tile模型天然具有更高的抽象层次。当开发者习惯了“只管定义Tile运算,硬件细节交给编译器”这种思维方式后,理论上同一套算法逻辑更容易适配到其他支持Tile编程的硬件上。

正如Jim Keller所说:”AI内核将更容易移植。”

不过英伟达也考虑了后手,CUDA Tile IR提供了跨代兼容性,但这种兼容性是建立在CUDA平台之上的。

开发者写的代码确实更容易移植了,但移植的目标是英伟达自家的不同代GPU,而非竞争对手的硬件。

从这个角度看,CUDA代码可以从Blackwell无缝迁移到下一代英伟达GPU,但要迁移到AMD或Intel的平台上,依然需要重写。

不管护城河是加深还是削弱,有一点是确定的:GPU编程的门槛确实在大幅降低。

过去能熟练驾驭CUDA的开发者是稀缺资源,会写Python的人一抓一大把,但能把代码优化到跑满Tensor Core的专家寥寥无几。

CUDA Tile和cuTile Python打通了这个瓶颈。英伟达在开发者博客中提到,一个15行的Python内核性能可以媲美200行手动优化的CUDA C++代码。

大量数据科学家和AI研究者从此可以直接上手写高性能GPU代码,不用再等HPC专家来帮忙优化。

参考链接:
[1]https://developer.nvidia.com/blog/focus-on-your-algorithm-nvidia-cuda-tile-handles-the-hardware
[2]https://x.com/jimkxa/status/1997732089480024498

特别声明:以上内容(如有图片或视频亦包括在内)为自媒体平台“网易号”用户上传并发布,本平台仅提供信息存储服务。

Notice: The content above (including the pictures and videos if any) is uploaded and posted by a user of NetEase Hao, which is a social media platform and only provides information storage services.

相关推荐
热点推荐
晚间公告|5月14日这些公告有看头

晚间公告|5月14日这些公告有看头

第一财经资讯
2026-05-14 18:27:12
NBA球队老板手撕华裔女猎手:100亿的仙人跳 搞砸了

NBA球队老板手撕华裔女猎手:100亿的仙人跳 搞砸了

慕容律师
2026-05-13 14:15:09
创历史首人耻辱纪录!坎宁安空砍39+7+9 季后赛12战狂送69失误

创历史首人耻辱纪录!坎宁安空砍39+7+9 季后赛12战狂送69失误

醉卧浮生
2026-05-14 11:21:08
11年前优衣库男女主现状曝光,他们还在一起生了两个孩子

11年前优衣库男女主现状曝光,他们还在一起生了两个孩子

半糖甜而不腻
2026-04-06 12:09:15
程维高并非被一撸到底,下台后享受副省级待遇,活了78岁

程维高并非被一撸到底,下台后享受副省级待遇,活了78岁

史不语
2026-05-13 08:10:10
特朗普抵达北京,放弃钓鱼台国宾馆,为何执意下榻四季酒店?

特朗普抵达北京,放弃钓鱼台国宾馆,为何执意下榻四季酒店?

青松解局
2026-05-14 18:34:45
中国未来最大的敌人不是特朗普,而是手握杀招、重构科技规则的他

中国未来最大的敌人不是特朗普,而是手握杀招、重构科技规则的他

妙知
2026-04-06 12:12:54
拿600万!全场5中1仅3分,拿到顶薪就躺平,球迷:在场上像没睡醒

拿600万!全场5中1仅3分,拿到顶薪就躺平,球迷:在场上像没睡醒

弄月公子
2026-05-14 07:42:59
最后时刻,黄仁勋挤上飞向北京的“空军一号”

最后时刻,黄仁勋挤上飞向北京的“空军一号”

中国新闻周刊
2026-05-13 20:56:03
马斯克出席北京欢迎仪式!穿西服配绿领带很绅士,还举着手机拍照

马斯克出席北京欢迎仪式!穿西服配绿领带很绅士,还举着手机拍照

八八尚语
2026-05-14 16:21:30
CBA半决赛4名高水平外籍裁判出炉:欧洲两人 韩国泰国各一人

CBA半决赛4名高水平外籍裁判出炉:欧洲两人 韩国泰国各一人

醉卧浮生
2026-05-14 16:06:39
兄妹勇救4名落水者后遭拉黑,被告知“被救者身份不便公开”;哥哥系盐城公交车司机,大气发声:救人是自愿的;官方:不影响见义勇为认定

兄妹勇救4名落水者后遭拉黑,被告知“被救者身份不便公开”;哥哥系盐城公交车司机,大气发声:救人是自愿的;官方:不影响见义勇为认定

扬子晚报
2026-05-14 12:13:49
黄仁勋:这会是一次非常成功的会晤

黄仁勋:这会是一次非常成功的会晤

财闻
2026-05-14 11:52:57
罕见啊!湖南一地电线杆上惊现红纸辞帖,网友:多年不见又现江湖

罕见啊!湖南一地电线杆上惊现红纸辞帖,网友:多年不见又现江湖

火山詩话
2026-05-14 13:46:29
事态升级了!四川凌晨追打案后续:6人全带走,女子袭警细节被扒

事态升级了!四川凌晨追打案后续:6人全带走,女子袭警细节被扒

小鋭有话说
2026-05-13 15:41:41
随特朗普抵京:马斯克第四个下机 黄仁勋换上西装

随特朗普抵京:马斯克第四个下机 黄仁勋换上西装

看看新闻Knews
2026-05-14 01:34:05
车辆斜停,浙江一司机瘫坐在驾驶室,言语混乱、不时傻笑,民警打开车门:比酒驾还严重!

车辆斜停,浙江一司机瘫坐在驾驶室,言语混乱、不时傻笑,民警打开车门:比酒驾还严重!

环球网资讯
2026-05-14 18:25:11
王洪文38岁正国级,狂草藏三处神技,你绝对没见过!

王洪文38岁正国级,狂草藏三处神技,你绝对没见过!

书画相约
2026-05-14 09:15:45
受害者给许家印写了副对联,上联:人民币不是许家印,下联太绝了

受害者给许家印写了副对联,上联:人民币不是许家印,下联太绝了

微野谈写作
2026-05-14 06:45:05
赖亚文为什么不招朱婷进国家队:因为赖亚文对朱婷太了解

赖亚文为什么不招朱婷进国家队:因为赖亚文对朱婷太了解

烟浔渺渺
2026-05-14 17:44:59
2026-05-14 20:28:49
量子位 incentive-icons
量子位
追踪人工智能动态
12624文章数 176462关注度
往期回顾 全部

科技要闻

马斯克说会谈很顺利 黄仁勋点赞 库克比耶

头条要闻

外媒询问中方对鲁比奥的制裁是否已经解除 外交部回应

头条要闻

外媒询问中方对鲁比奥的制裁是否已经解除 外交部回应

体育要闻

争议抽象天王山,和季后赛最稳定中锋

娱乐要闻

何九华官宣当爸!全程不提孩子妈

财经要闻

李强会见美国工商界代表

汽车要闻

双零重力座椅/AI智能体/调光天幕 启境GT7内饰发布

态度原创

游戏
艺术
家居
亲子
公开课

LPL第二赛段:不拖的涅槃,就是好涅槃!LNG零封LGD

艺术要闻

美国务卿鲁比奥点赞中式美学,实景令人惊叹!

家居要闻

精神奢享 对话塔尖需求

亲子要闻

摆烂在家的孩子,为什么会疯狂要钱?

公开课

李玫瑾:为什么性格比能力更重要?

无障碍浏览 进入关怀版