AMD发布AMD ROCm 5.6版,为AI应用带来最佳性能

AMD发布AMD ROCm 5.6版,为AI应用带来最佳性能

AMD针对AI应用发布全新AMD ROCm 5.6开放软件平台,AMD人工智能事业群资深副总裁Vamsi Boppana于部落格文章中重点介绍ROCm 5.6的新功能。

AMD ROCm 5.6版正式发布

ROCm 5.6 为人工智能(AI) 与大型语言模型 (LLM) 工作负载带来:

  • 将 Hugging Face 单元测试套件集成至 ROCm QA 中。
  • 在 PyTorch 2.0 传感器模式下对 OpenAI Triton 的增量支持。
  • 通过 ROCm 对 PyTorch、TensorFlow 和 JAX 提供 OpenXLA 支持,支持更广泛的社群。

对于 HPC 应用程序,ROCm 5.6 引入对快速傅立叶变换 (Fast Fourier Transform,FFT)、基础线性代数程式集 (Basic Linear Algebra Subprograms,BLAS) 与计算机 (solver) 等多个数学函式库的效能提升。 这些数学库是 HPC 应用程序的基础,并增强了 ROCm 开发和部署工具,包括安装、ROCgdb (CPU-GPU 集成调试器)、ROCm 分析工具和文件。

为提供最卓越的性能,AMD也将于未来数月内对框架和后端编译器作进一步的优化,其中包括MLIR基础架构改进,巩固AMD对OpenAI Triton与OpenXLA编译器的支持。 客户将在 Hugging Face 的 AMD Hub 上看到更多为 AMD 解决方案作出优化的开源 AI 模型。

此外,AMD将于今年晚些时候正式在采用AMD RDNA 3架构的特定显卡上释出AMD ROCm支持。 Linux 预计将于今年秋季正式支持基于 RDNA 3 架构的显卡,首先是 48GB Radeon PRO W7900 绘图卡与 24GB Radeon RX 7900 XTX 显示卡,并将陆续发布更多支持显卡和扩充功能。

(1)
老哥老哥

相关推荐

发表回复

您的电子邮箱地址不会被公开。 必填项已用 * 标注