Using the Graphcore IPU for Traditional HPC Applications

EasyChair Preprint no. 4896

Abstract

The increase in machine learning workloads means that AI accelerators are expected to become common in supercomputers, evoking considerable interest in the scientific high- performance computing (HPC) community about how these devices might also be exploited for traditional HPC workloads. In this paper, we report our early results using the Graphcore Intelligence Processing Unit (IPU) for stencil computations on structured grid problems, which are used for solvers for differential equations in domains such as computational fluid dynamics. We characterise the IPU’s performance by presenting both STREAM memory bandwidth benchmark results and a Roofline performance model. Using two example applications (the Gaussian Blur filter and a 2D Lattice Boltzmann fluid simulation), we discuss the challenges encountered during this first known IPU implementation of structured grid stencils. We demonstrate that the IPU and its low-level programming framework, Poplar, expose sufficient programmability to express these HPC problems, and achieve performance comparable to that of modern GPUs.

Keyphrases: Accelerator, differential equation, heterogeneous computing, HPC, Roofline Model, Stencil, stencil computation, structured grid

@Booklet{EasyChair:4896,
  author = {Thorben Louw and Simon McIntosh-Smith},
  title = {Using the Graphcore IPU for Traditional HPC Applications},
  howpublished = {EasyChair Preprint no. 4896},

  year = {EasyChair, 2021}}