A shared compilation stack for distributed-memory parallelism in stencil DSLs
Proceedings of the 29th ACM International Conference on Architectural Support for Programming Languages and Operating Systems, Volume 3(2024)
摘要
Domain Specific Languages (DSLs) increase programmer productivity and provide
high performance. Their targeted abstractions allow scientists to express
problems at a high level, providing rich details that optimizing compilers can
exploit to target current- and next-generation supercomputers. The convenience
and performance of DSLs come with significant development and maintenance
costs. The siloed design of DSL compilers and the resulting inability to
benefit from shared infrastructure cause uncertainties around longevity and the
adoption of DSLs at scale. By tailoring the broadly-adopted MLIR compiler
framework to HPC, we bring the same synergies that the machine learning
community already exploits across their DSLs (e.g. Tensorflow, PyTorch) to the
finite-difference stencil HPC community. We introduce new HPC-specific
abstractions for message passing targeting distributed stencil computations. We
demonstrate the sharing of common components across three distinct HPC
stencil-DSL compilers: Devito, PSyclone, and the Open Earth Compiler, showing
that our framework generates high-performance executables based upon a shared
compiler ecosystem.
更多查看译文
AI 理解论文
溯源树
样例
生成溯源树,研究论文发展脉络
Chat Paper
正在生成论文摘要