初创公司硬刚英伟达:在AMD卡上模拟CUDA,原版程序直接编译运行

克雷西 发自 凹非寺
量子位 | 公众号 QbitAI

无需任何修改转换,AMD显卡也跑起原版CUDA程序了!

英国一家初创公司,推出了针对AMD的CUDA程序编译工具,而且免费商用。

工具一经发布就引发了网友们的广泛热议,并登顶了HackerNews热榜

d2c29c211e1df4bf21d253f501302f2c.png

该工具名为SCALE,开发者给它的定位是一个GPGPU(通用GPU)编程工具包。

b6dcc05e56efa9589904e45783817497.png

目前已有大模型框架llama-cpp等9个程序通过测试,实现了正常运行。

与其他实现方式不同,SCALE直接模拟CUDA工具包的安装,不需要转换成其他语言就能源到源地完成编译。

因此,像内联PTX这种英伟达专用的中间语言,SCALE同样能够提供支持。

无需转换,让AMD跑起CUDA

官网介绍显示,SCALE主要有三个组成部分——兼容的nvcc编译器、CUDA运行时和驱动API的AMD实现,以及ROCm库。

其中编译器可以把用包括nvcc、内联PTX等在内的CUDA专属语言编写的程序,直接编译成可在AMD GPU上运行的二进制代码。

ROCm库则用来提供“CUDA-X”API,SCALE在处理cuBLAS和cuSOLVER等库时均采用这种方式。

c6da0219ee8da8a7ab629d4f8616215c.png

SCALE的关键创新点在于按原样接受CUDA程序,而不必将它们移植到另一种语言,兼容nvcc和clang等多种编译方式,同时现有的构建工具和脚本(如cmake)可以正常工作。

按照官方的说法,SCALE实现了对CUDA的完全兼容,让开发者无需针对不同GPU平台分别编写代码

这一点与AMD推出的HIP存在很大不同,因为HIP是通过一定方式对CUDA代码进行重写,遇到复杂的宏可能无法正确理解,也不支持内联PTX等专属语言。

甚至SCALE作者认为,HIP并不能解决CUDA兼容性问题。

另外,SCALE的语言是CUDA的超集,提供一些可选的语言扩展,可以使希望摆脱nvcc的开发者更轻松、更高效地编写GPU代码。

作者表示,希望未来开发者可以只编写一次代码,就能不同硬件平台上运行,并正在着手通过弥合流行的CUDA编程语言和其他硬件供应商之间的兼容性差距。

目前,SCALE对AMD各系列GPU的支持情况如下:

  • 已支持:gfx1030(RX6000系列)和gfx1100(RX7000系列)

  • “似乎有效”(seem to work):gfx1010(RX5000系列)和gfx1101

  • 正在适配:gfx900(RX Vega系列)

另外,作者测试了一些CUDA开源项目,并利用SCALE成功运行了9款CUDA应用。

2648a9cc83c0932323d7539e5f0126b8.png

不过,SCALE毕竟是个全新的项目,所以作者也准备了从安装到编译的一系列教程,给出了不同类型的示例程序。

教程关键步骤都附有相关代码,甚至包括了如何判断自己GPU的型号,可以说是非常细致了。

72814c54c21baf5940d0e158083cc848.png

如果使用过程中遇到问题,作者也介绍了常见故障的排除方式,同时还开通了Discord论坛,可与开发团队直接沟通交流。

26cbf1f1a256f87fbc9f9a6613cf6cda.png

打造出SCALE的这家初创公司名为Spectral Compute,2018年成立于英国,自称对CPU和GPU的架构有深入的了解,目标是帮助开发者高效利用算力资源。

1756280ca486f6367707289dc363a861.png

网友:挑战英伟达护城河?

有网友认为,如果SCALE真能有(宣传中的)效果,那么将对英伟达护城河发起挑战,让AMD与之展开直接竞争。

9e5917f42774e78ec259a7a2a359508a.png

不过,现在下定论未免为时过早,毕竟SCALE官方也承认目前和原版CUDA相比还存在一些缺陷。

而且开发者也明确表示有一些CUDA API和功能并不被支持,但也没有给出具体的列表。

048c37168f323798eee27262d4676e5e.png

关于“AMD方案”的更多不足,一名自称和SCALE团队交流过的网友表示,目前的SCALE还无法操作TensorCore,也就意味着AMD上跑不了FlashAttention加速框架。

另外由于N卡上有强大的矩阵乘法单元,所以即使能编译运行,AMD卡上的性能可能也不及N卡。

f567ea792fbfba1cfcc904037390792f.png

更有网友认为,英伟达一家独大的原因是AMD不愿意投资让其GPU拥有更高的机器学习性能(而不只是拥有CUDA这一个优势)。

1d55f355360e518f7a522feccaa49492.png

即便能够高效运行,AMD卡是否真的经济实惠且容易获得,同样是一个问题。

0ca84178c9d2d722f72eb2b482347b47.png

还有一波网友的观点是,最大的问题不是技术上能不能运行,而是背后的法律问题。

这个问题同样引发了广泛的讨论,不过目前也暂未有定论。

有人认为SCALE像ZLUDA(另一种在AMD上运行CUDA程序的方式)一样存在法律疑点,有可能招致英伟达的诉讼。

002de3834d7462cb0a24e8c16b7e4513.png

具体来说,根据英伟达的EULA条款,CUDA SDK只允许开发在N卡上运行的应用程序,这可能禁止了类似SCALE这样的兼容实现。

f7410f3578d80831117aa30787c3110f.png

但立马就又有网友说,SCALE并没有用到英伟达的“SDK”,何谈SDK使用协议呢?

dd8b042a1b365b72c36f215abfe38da4.png

总之,无论是技术不足还是法律问题,对于这个新工具的讨论都还在持续进行。

至于到底好不好用,这就要由开发者用脚投票了。

参考链接:
[1]https://docs.scale-lang.com/
[2]https://news.ycombinator.com/item?id=40970560

量子位年度AI主题策划正在征集中!

欢迎投稿专题 一千零一个AI应365行AI落地方案

或与我们分享你在寻找的AI产品,或发现的AI新动向

2918ec223d933d8bf5e489988fcaa5ba.png

点这里👇关注我,记得标星哦~

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值