场方程建模中的颠覆性变革:晶圆级引擎的简易接口
分布式、并行与集群计算
2022-09-30 v2 硬件体系结构
数学软件
性能
摘要
我们提出了一种高层次且易用的应用程序编程接口(API),用于在 Cerebras Systems 晶圆级引擎(WSE)上求解场方程,相较传统分布式计算方法具有超过两个数量级的性能提升。该领域专用 API 称为 WSE 场方程 API(WFA)。WFA 在 NETL 的 Joule 2.0 超级计算机上求解时间比 OpenFOAM 快超过两个数量级。虽然该性能与手工优化的汇编代码一致,但 WFA 提供了易用的高层次 Python 接口,使用户能够轻松构建并求解场方程。我们在此报告 WFA 的编程方法及其在最新一代 WSE(CS-2)上达成的性能。
引用
@article{arxiv.2209.13768,
title = {Disruptive Changes in Field Equation Modeling: A Simple Interface for Wafer Scale Engines},
author = {Mino Woo and Terry Jordan and Robert Schreiber and Ilya Sharapov and Shaheer Muhammad and Abhishek Koneru and Michael James and Dirk Van Essendelft},
journal= {arXiv preprint arXiv:2209.13768},
year = {2022}
}
备注
22 pages, 5 figures, 2 Tables