Architecture and performance of Devito, a system for automated stencil computation
read the original abstract
Stencil computations are a key part of many high-performance computing applications, such as image processing, convolutional neural networks, and finite-difference solvers for partial differential equations. Devito is a framework capable of generating highly-optimized code given symbolic equations expressed in Python, specialized in, but not limited to, affine (stencil) codes. The lowering process---from mathematical equations down to C++ code---is performed by the Devito compiler through a series of intermediate representations. Several performance optimizations are introduced, including advanced common sub-expressions elimination, tiling and parallelization. Some of these are obtained through well-established stencil optimizers, integrated in the back-end of the Devito compiler. The architecture of the Devito compiler, as well as the performance optimizations that are applied when generating code, are presented. The effectiveness of such performance optimizations is demonstrated using operators drawn from seismic imaging applications.
This paper has not been read by Pith yet.
Forward citations
Cited by 5 Pith papers
-
Joint elastic full waveform inversion of multi-component geophone and distributed acoustic sensing data
A VSS-based joint FWI framework enables direct multi-deployment inversion of geophone and DAS data, yielding more accurate elastic parameter recovery than single-sensor cases on Marmousi benchmarks when sensors provid...
-
Stencil Computations on Cerebras Wafer-Scale Engine
CStencil on the WSE-3 achieves up to 342x speedup for 2D stencils versus an adapted single-precision GPU solver and saturates both compute and on-chip memory bandwidth.
-
Automatic Differentiation for Adjoint Stencil Loops
A method for adjoint differentiation of stencil loops that preserves their structure and parallelizability via combined AD and loop transformations, released as the PerforAD tool with seismic and CFD test cases.
-
Arbitrary high order shaped stencils for time domain finite difference schemes in seismic wave propagation
A general mathematical framework is proposed for deriving arbitrary high-order shaped stencils in finite difference schemes for acoustic wave propagation, with tests indicating benefits for dispersion-optimized cross ...
-
Investigating the OPS intermediate representation to target GPUs in the Devito DSL
Integration of OPS intermediate representation as a GPU backend in Devito DSL yields speedups over the core backend for structured-mesh finite-difference applications.
discussion (0)
Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.