首页|PETSc/TAO developments for GPU-based early exascale systems

PETSc/TAO developments for GPU-based early exascale systems

扫码查看
原文链接
  • NETL
  • NSTL
  • Sage
The Portable Extensible Toolkit for Scientific Computation (PETSc) library provides scalable solvers for nonlinear time-dependent differential and algebraic equations and for numerical optimization via the Toolkit for Advanced Optimization (TAO). PETSc is used in dozens of scientific fields and is an important building block for many simulation codes. During the U.S. Department of Energy's Exascale Computing Project, the PETSc team has made substantial efforts to enable efficient utilization of the massive fine-grain parallelism present within exascale compute nodes and to enable performance portability across exascale architectures. We recap some of the challenges that designers of numerical libraries face in such an endeavor, and then discuss the many developments we have made, which include the addition of new GPU backends, features supporting efficient on-device matrix assembly, better support for asynchronicity and GPU kernel concurrency, and new communication infrastructure. We evaluate the performance of these developments on some pre-exascale systems as well as the early exascale systems Frontier and Aurora, using compute kernel, communication layer, solver, and mini-application benchmark studies, and then close with a few observations drawn from our experiences on the tension between portable performance and other goals of numerical libraries.

PETScgraphics processing unitsexascale computing projectperformance portabilityalgebraic solvers

Richard Tran Mills、Mark F. Adams、Satish Balay、Jed Brown、Jacob Faibussowitsch、Toby Isaac、Matthew G. Knepley、Todd Munson、Hansol Sun、Stefano Zampini、Hong Zhang、Junchao Zhang

展开 >

Argonne National Laboratory, Lemont, IL, USA

Lawrence Berkeley National Laboratory, Berkeley, CA, USA

University of Colorado Boulder, Boulder, CO, USA

University at Buffalo, Buffalo, NY, USA

King Abdullah University of Science and Technology, Thuwal, Saudi Arabia

展开 >

2025

10.1177/10943420241303710

International journal of high performance computing applications

International journal of high performance computing applications

ISSN：1094-3420
年,卷(期)：2025.39(2)
  • 43