近日,AMD 发布了新的 AMD ROCm 5.6 开放软件平台,AMD AI群组高级副总裁 Vamsi Boppana 在博客中重点介绍了 ROCm 5.6 的一些新功能。
对于 AI 和 LLM 工作负载,ROCm 5.6 带来了:
· 将 Hugging Face 单元测试套件集成到 ROCm QA 中
· PyTorch 2.0 感应器模式下对 OpenAI Triton 的增量支持
· 通过 ROCm 对 PyTorch、TensorFlow 和 JAX 提供 OpenXLA 支持,支持更广泛的社区
对于 HPC 应用程序,ROCm 5.6 引入了对 FFT、BLAS 和求解器等多个数学库的改进,这些数学库构成了 HPC 应用程序的基础,并增强了 ROCm 开发和部署工具,包括安装、ROCgdb(CPU-GPU 集成调试器)、ROCm 分析器 和文档。
在接下来的几个月中,AMD 还将进一步优化框架和后端编译器,以实现更佳性能,其中包括 MLIR 基础设施改进,巩固 AMD 对 OpenAI Triton 和 OpenXLA 编译器的支持。客户还可以在 Hugging Face 的 AMD 中心内期待更多针对 AMD 解决方案优化的开源 AI 模型。
此外,AMD 透露,今年晚些时候,对AMD ROCm 的正式支持将开始在部分基于 AMD RDNA 3 的 GPU 上推出。Linux 上对基于 RDNA 3 的 GPU 的正式支持计划于今年秋季开始推出,首先是 48GB Radeon PRO W7900 和 24GB Radeon RX 7900 XTX,随着时间的推移,还会发布更多卡和扩展功能。
责任编辑:彭菁
-
amd
+关注
关注
25文章
5440浏览量
133921 -
软件
+关注
关注
69文章
4765浏览量
87147 -
编译器
+关注
关注
1文章
1618浏览量
49045
原文标题:AMD 发布针对 AI 优化的 ROCm 5.6
文章出处:【微信号:AMD中国,微信公众号:AMD中国】欢迎添加关注!文章转载请注明出处。
发布评论请先 登录
相关推荐
评论