Skip to content
Snippets Groups Projects

pystencils

Run blazingly fast stencil codes on numpy arrays.

alt text

Installation

export PIP_EXTRA_INDEX_URL=https://www.walberla.net/pip
pip install pystencils[interactive]

Without [interactive] you get a minimal version with very little dependencies.

All options:

  • gpu: use this if nVidia GPU is available and CUDA is installed
  • alltrafos: pulls in additional dependencies for loop simplification e.g. libisl
  • bench_db: functionality to store benchmark result in object databases
  • interactive: installs dependencies to work in Jupyter including image I/O, plotting etc.
  • doc: packages to build documentation

Options can be combined e.g.

pip install pystencils[interactive,gpu,doc]

Documentation

Read the docs here