pyhpc-benchmarks
pyhpc-benchmarks copied to clipboard
A suite of benchmarks for CPU and GPU performance of the most popular high-performance libraries for Python :rocket:
Fixes #9 Notes: 1. Calling the setup function multiple times in a benchmark should be avoided 2. Equation of state benchmark was easy to implement 3. Isoneutral benchmark has some...
https://github.com/fluiddyn/transonic can accelerate Python/Numpy code and run on accelerators
https://github.com/weld-project/weld can accelerate Python/numpy code by optimizing data movement.
The Tensor Algebra Compiler (https://github.com/tensor-compiler/taco) seems to be good at sparse/dense linear algebra and has Python frontend: http://tensor-compiler.org/docs/pycomputations/index.html
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...
DaCe (https://github.com/spcl/dace) is a parallel computing framework that also support Numpy frontend, similar to JAX and Numba. It runs on CPU/GPU/FPGA. Would be interesting to add it for comparison!