├── .flake8 ├── .gitignore ├── CITATION.cff ├── LICENSE ├── README.md ├── backends.py ├── benchmarks ├── __init__.py ├── equation_of_state │ ├── README.md │ ├── __init__.py │ ├── eos_aesara.py │ ├── eos_cupy.py │ ├── eos_jax.py │ ├── eos_numba.py │ ├── eos_numpy.py │ ├── eos_pytorch.py │ ├── eos_taichi.py │ └── eos_tensorflow.py ├── isoneutral_mixing │ ├── README.md │ ├── __init__.py │ ├── isoneutral_aesara.py │ ├── isoneutral_cupy.py │ ├── isoneutral_jax.py │ ├── isoneutral_numba.py │ ├── isoneutral_numpy.py │ ├── isoneutral_pytorch.py │ └── isoneutral_taichi.py └── turbulent_kinetic_energy │ ├── README.md │ ├── __init__.py │ ├── tke_jax.py │ ├── tke_numba.py │ ├── tke_numpy.py │ └── tke_pytorch.py ├── environment-cpu.yml ├── environment-gpu.yml ├── plot.py ├── results ├── aws-plots │ ├── bench-equation_of_state-CPU.png │ ├── bench-equation_of_state-GPU.png │ ├── bench-isoneutral_mixing-CPU.png │ ├── bench-isoneutral_mixing-GPU.png │ ├── bench-turbulent_kinetic_energy-CPU.png │ └── bench-turbulent_kinetic_energy-GPU.png ├── aws.md ├── colab.md ├── magni-plots │ ├── bench-equation_of_state-CPU.png │ ├── bench-equation_of_state-GPU.png │ ├── bench-isoneutral_mixing-CPU.png │ ├── bench-isoneutral_mixing-GPU.png │ ├── bench-turbulent_kinetic_energy-CPU.png │ └── bench-turbulent_kinetic_energy-GPU.png ├── magni-run-all.sh ├── magni.md └── pyhpc_benchmarks_colab.ipynb ├── run.py └── utilities.py /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/.flake8 -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/.gitignore -------------------------------------------------------------------------------- /CITATION.cff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/CITATION.cff -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/README.md -------------------------------------------------------------------------------- /backends.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/backends.py -------------------------------------------------------------------------------- /benchmarks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /benchmarks/equation_of_state/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/benchmarks/equation_of_state/README.md -------------------------------------------------------------------------------- /benchmarks/equation_of_state/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/benchmarks/equation_of_state/__init__.py -------------------------------------------------------------------------------- /benchmarks/equation_of_state/eos_aesara.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/benchmarks/equation_of_state/eos_aesara.py -------------------------------------------------------------------------------- /benchmarks/equation_of_state/eos_cupy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/benchmarks/equation_of_state/eos_cupy.py -------------------------------------------------------------------------------- /benchmarks/equation_of_state/eos_jax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/benchmarks/equation_of_state/eos_jax.py -------------------------------------------------------------------------------- /benchmarks/equation_of_state/eos_numba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/benchmarks/equation_of_state/eos_numba.py -------------------------------------------------------------------------------- /benchmarks/equation_of_state/eos_numpy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/benchmarks/equation_of_state/eos_numpy.py -------------------------------------------------------------------------------- /benchmarks/equation_of_state/eos_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/benchmarks/equation_of_state/eos_pytorch.py -------------------------------------------------------------------------------- /benchmarks/equation_of_state/eos_taichi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/benchmarks/equation_of_state/eos_taichi.py -------------------------------------------------------------------------------- /benchmarks/equation_of_state/eos_tensorflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/benchmarks/equation_of_state/eos_tensorflow.py -------------------------------------------------------------------------------- /benchmarks/isoneutral_mixing/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/benchmarks/isoneutral_mixing/README.md -------------------------------------------------------------------------------- /benchmarks/isoneutral_mixing/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/benchmarks/isoneutral_mixing/__init__.py -------------------------------------------------------------------------------- /benchmarks/isoneutral_mixing/isoneutral_aesara.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/benchmarks/isoneutral_mixing/isoneutral_aesara.py -------------------------------------------------------------------------------- /benchmarks/isoneutral_mixing/isoneutral_cupy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/benchmarks/isoneutral_mixing/isoneutral_cupy.py -------------------------------------------------------------------------------- /benchmarks/isoneutral_mixing/isoneutral_jax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/benchmarks/isoneutral_mixing/isoneutral_jax.py -------------------------------------------------------------------------------- /benchmarks/isoneutral_mixing/isoneutral_numba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/benchmarks/isoneutral_mixing/isoneutral_numba.py -------------------------------------------------------------------------------- /benchmarks/isoneutral_mixing/isoneutral_numpy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/benchmarks/isoneutral_mixing/isoneutral_numpy.py -------------------------------------------------------------------------------- /benchmarks/isoneutral_mixing/isoneutral_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/benchmarks/isoneutral_mixing/isoneutral_pytorch.py -------------------------------------------------------------------------------- /benchmarks/isoneutral_mixing/isoneutral_taichi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/benchmarks/isoneutral_mixing/isoneutral_taichi.py -------------------------------------------------------------------------------- /benchmarks/turbulent_kinetic_energy/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/benchmarks/turbulent_kinetic_energy/README.md -------------------------------------------------------------------------------- /benchmarks/turbulent_kinetic_energy/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/benchmarks/turbulent_kinetic_energy/__init__.py -------------------------------------------------------------------------------- /benchmarks/turbulent_kinetic_energy/tke_jax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/benchmarks/turbulent_kinetic_energy/tke_jax.py -------------------------------------------------------------------------------- /benchmarks/turbulent_kinetic_energy/tke_numba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/benchmarks/turbulent_kinetic_energy/tke_numba.py -------------------------------------------------------------------------------- /benchmarks/turbulent_kinetic_energy/tke_numpy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/benchmarks/turbulent_kinetic_energy/tke_numpy.py -------------------------------------------------------------------------------- /benchmarks/turbulent_kinetic_energy/tke_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/benchmarks/turbulent_kinetic_energy/tke_pytorch.py -------------------------------------------------------------------------------- /environment-cpu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/environment-cpu.yml -------------------------------------------------------------------------------- /environment-gpu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/environment-gpu.yml -------------------------------------------------------------------------------- /plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/plot.py -------------------------------------------------------------------------------- /results/aws-plots/bench-equation_of_state-CPU.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/results/aws-plots/bench-equation_of_state-CPU.png -------------------------------------------------------------------------------- /results/aws-plots/bench-equation_of_state-GPU.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/results/aws-plots/bench-equation_of_state-GPU.png -------------------------------------------------------------------------------- /results/aws-plots/bench-isoneutral_mixing-CPU.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/results/aws-plots/bench-isoneutral_mixing-CPU.png -------------------------------------------------------------------------------- /results/aws-plots/bench-isoneutral_mixing-GPU.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/results/aws-plots/bench-isoneutral_mixing-GPU.png -------------------------------------------------------------------------------- /results/aws-plots/bench-turbulent_kinetic_energy-CPU.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/results/aws-plots/bench-turbulent_kinetic_energy-CPU.png -------------------------------------------------------------------------------- /results/aws-plots/bench-turbulent_kinetic_energy-GPU.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/results/aws-plots/bench-turbulent_kinetic_energy-GPU.png -------------------------------------------------------------------------------- /results/aws.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/results/aws.md -------------------------------------------------------------------------------- /results/colab.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/results/colab.md -------------------------------------------------------------------------------- /results/magni-plots/bench-equation_of_state-CPU.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/results/magni-plots/bench-equation_of_state-CPU.png -------------------------------------------------------------------------------- /results/magni-plots/bench-equation_of_state-GPU.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/results/magni-plots/bench-equation_of_state-GPU.png -------------------------------------------------------------------------------- /results/magni-plots/bench-isoneutral_mixing-CPU.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/results/magni-plots/bench-isoneutral_mixing-CPU.png -------------------------------------------------------------------------------- /results/magni-plots/bench-isoneutral_mixing-GPU.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/results/magni-plots/bench-isoneutral_mixing-GPU.png -------------------------------------------------------------------------------- /results/magni-plots/bench-turbulent_kinetic_energy-CPU.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/results/magni-plots/bench-turbulent_kinetic_energy-CPU.png -------------------------------------------------------------------------------- /results/magni-plots/bench-turbulent_kinetic_energy-GPU.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/results/magni-plots/bench-turbulent_kinetic_energy-GPU.png -------------------------------------------------------------------------------- /results/magni-run-all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/results/magni-run-all.sh -------------------------------------------------------------------------------- /results/magni.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/results/magni.md -------------------------------------------------------------------------------- /results/pyhpc_benchmarks_colab.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/results/pyhpc_benchmarks_colab.ipynb -------------------------------------------------------------------------------- /run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/run.py -------------------------------------------------------------------------------- /utilities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dionhaefner/pyhpc-benchmarks/HEAD/utilities.py --------------------------------------------------------------------------------