本月早些时候,AMD 谈论了更多关于其统一人工智能软件栈(Unified AI Software Stack)的计划,该计划将在未来几个月内推出,以提供一个统一的软件视图,将人工智能工作无缝卸载到 Ryzen 处理器、AMD 显卡或 AMD Ryzen AI NPU 硬件上。
AMD 统一人工智能软件堆栈的核心部分是使用 LLVM 的 MLIR 中间表示法,然后由这个统一软件决定将人工智能工作卸载到哪个设备上。考虑到 MLIR 的强大功能,它比其他 IR 更适合人工智能,以及他们获得的 MLIR 人才等因素,这样做是有道理的。
AMD 曾在 AMD 不同硬件设备的背景下大谈统一人工智能软件栈。在 LLVM/Clang 19 中,一位 AMD AI 编译器工程师开始将通用的 MLIR 转换为 SPIR-V。AMD 人工智能编译器团队的这位成员开始了上游工作,以便将多层中间表示转换为 SPIR-V,作为 Vulkan 驱动程序以及现代 OpenGL 和 OpenCL 驱动程序可以使用的 IR。
自此以后,MLIR 的通用 “convert-to-spirv “通行证工作一直在继续。例如,就在昨天,LLVM 提出了一项合并请求,要求通过该通道支持矢量解卷。该请求同样来自 AMD AI 编译器工程师。
正在研究的这个 MLIR 到 SPIR-V 传递是否确实适用于统一的 AI 软件堆栈还有待观察,但时机很有趣,而且它与 AMD 的“统一”软件计划非常契合。一旦此过程功能齐全,理论上,通用的 MLIR-to-SPIR-V 过程可用于定位其他 Vulkan 和 OpenCL/OpenGL 支持的显卡/设备。SPIR-V 是一种常见的 IR,并非特定于 AMD。因此,如果此转换为 spirv 的过程变得足够可行,它可能会为 AMD 统一 AI 软件堆栈打开大门,使其适用于其他非 AMD GPU。如果它成为现实,这将非常方便,但其他 AI 支持库等构成 AMD 其余 AI 计算堆栈可能会受到一些限制。
这在 Radeon GPU 支持 Vulkan/SPIR-V 但没有安装 ROCm 计算库或无法通过 SPIR-V 的情况下也可能有用。一旦 MLIR 过渡到 Khronos Group 支持的 IR,还有在 CPU 上执行 SPIR-V 的路线和其他可能性。
无论如何,在最近的 AMD 技术日上讨论统一 AI 软件堆栈之后,再回想 AMD AI 编译器工程师最近一直在为上游 LLVM 开发通用 MLIR 到 SPIR-V 传递,AMD 非常有趣和创新的软件时代可能就在前方。
版权声明:本文内容转自互联网,本文观点仅代表作者本人。本站仅提供信息存储空间服务,所有权归原作者所有。如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至1393616908@qq.com 举报,一经查实,本站将立刻删除。