通过 AMD AI 引擎无缝加速 Fortran 内建函数
分布式、并行与集群计算
2025-04-15 v1 新兴技术
性能
摘要
HPC 社区面临的一个重大挑战是如何在满足可持续运行日益增长的关注的同时，继续为科学程序员提供所要求的性能。专用架构，如 FPGA 和 AMD 的 AI 引擎（AIE），已被证明能够提供显著的能效优势，然而一个主要挑战是，要最有效地编程这些架构需要大量的专业知识和时间投入，这是主要的障碍。Fortran 是科学计算的语言，在本文中我们探索通过 AMD Ryzen AI CPU 中的 AIE 自动加速 Fortran 内建函数。利用开源 Flang 编译器和 MLIR 生态系统，我们描述了一种将 MLIR 线性代数方言降低到 AMD AIE 方言的方法，并证明对于合适的工作负载，AIE 可以在无需程序员进行任何代码修改的情况下提供显著的性能优势。
引用
@article{arxiv.2502.10254,
title = {Seamless acceleration of Fortran intrinsics via AMD AI engines},
author = {Nick Brown and Gabriel Rodríguez Canal},
journal= {arXiv preprint arXiv:2502.10254},
year = {2025}
}
备注
Preprint