SISSO方法在CPU和GPU上的高性能与可移植实现
性能
2025-02-28 v1
摘要
SISSO( sure-independence screening and sparsifying operator)是一种基于符号回归和压缩感知的人工智能方法，广泛应用于材料科学研究。SISSO++是其C++实现，采用MPI和OpenMP进行并行化，适合高性能计算(HPC)环境。随着异构硬件在HPC和AI领域的主流化，我们选择使用Kokkos性能可移植库将SISSO++代码移植到GPU上。Kokkos允许我们维持针对Nvidia和AMD GPU的单一代码库，显著减少维护工作。在本工作中，我们总结了实现硬件和性能可移植性所需的代码更改，并附带在Nvidia和AMD GPU上的性能基准。我们展示了在代码三个最耗时的部分获得的加速。
引用
@article{arxiv.2502.20072,
title = {A high-performance and portable implementation of the SISSO method for CPUs and GPUs},
author = {Sebastian Eibl and Yi Yao and Matthias Scheffler and Markus Rampp and Luca M. Ghiringhelli and Thomas A. R. Purcell},
journal= {arXiv preprint arXiv:2502.20072},
year = {2025}
}