专栏名称: 猿大侠
猿大侠,既然选择了,就一定成为大侠! 小程序、小游戏、Google、苹果、职场、前沿技术分享,一起成长。
目录
相关文章推荐
人民网舆情数据中心  ·  广东吴川一中学生疑遭欺凌、官方回应网传浙江一 ... ·  10 小时前  
人民网舆情数据中心  ·  从农村聚餐不得使用四季豆事件分析基层移风易俗 ... ·  10 小时前  
人民网舆情数据中心  ·  《哪吒2》登顶全球动画电影票房榜、河北邢台立 ... ·  3 天前  
人民网舆情数据中心  ·  金融机构接入DeepSeek,这些舆情风险需 ... ·  3 天前  
51好读  ›  专栏  ›  猿大侠

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

猿大侠  · 公众号  ·  · 2024-07-18 13:00

正文

作者:克雷西
转自:量子位 | 公众号 QbitAI

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

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

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

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

目前已有大模型框架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等库时均采用这种方式。

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应用。

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

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

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

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

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

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

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

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

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

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

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

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







请到「今天看啥」查看全文