pyhpc-benchmarks
pyhpc-benchmarks copied to clipboard
Compare with an MLIR-based stencil DSL
This project https://github.com/spcl/open-earth-compiler/ provides a DSL frontend for stencil/PDE programs, and rely on MLIR & LLVM to run on NVIDIA and AMD GPUs. It is not a Python frontend, but can be called from Python I think (see https://arxiv.org/abs/2005.13014)
If there's interest for this, I would be happy to accept a PR implementing one or more benchmarks with this (and run the benchmarks on my hardware).