Skip to content
Snippets Groups Projects
Commit 18eb23bc authored by Stephan Seitz's avatar Stephan Seitz
Browse files

Add pystencils.autodiff

parent 2e2bc712
No related branches found
No related tags found
No related merge requests found
...@@ -55,6 +55,7 @@ All options: ...@@ -55,6 +55,7 @@ All options:
- `alltrafos`: pulls in additional dependencies for loop simplification e.g. libisl - `alltrafos`: pulls in additional dependencies for loop simplification e.g. libisl
- `bench_db`: functionality to store benchmark result in object databases - `bench_db`: functionality to store benchmark result in object databases
- `interactive`: installs dependencies to work in Jupyter including image I/O, plotting etc. - `interactive`: installs dependencies to work in Jupyter including image I/O, plotting etc.
- `autodiff`: enable derivation of adjoint kernels and generation of Torch/Tensorflow operations
- `doc`: packages to build documentation - `doc`: packages to build documentation
Options can be combined e.g. Options can be combined e.g.
......
...@@ -13,6 +13,13 @@ from .simp import AssignmentCollection ...@@ -13,6 +13,13 @@ from .simp import AssignmentCollection
from .slicing import make_slice from .slicing import make_slice
from .sympyextensions import SymbolCreator from .sympyextensions import SymbolCreator
try:
import pystencils_autodiff
autodiff = pystencils_autodiff
except Exception:
pass
__all__ = ['Field', 'FieldType', 'fields', __all__ = ['Field', 'FieldType', 'fields',
'TypedSymbol', 'TypedSymbol',
'make_slice', 'make_slice',
......
"""
Provides tools for generation of auto-differentiable operations.
See https://github.com/theHamsta/pystencils_autodiff
Installation:
.. code-block:: bash
pip install pyconrad-autodiff
"""
import os
if 'CI' not in os.environ:
raise NotImplementedError('pystencils-autodiff is not installed. Run `pip install pyconrad-autodiff`')
...@@ -108,6 +108,7 @@ setup(name='pystencils', ...@@ -108,6 +108,7 @@ setup(name='pystencils',
'alltrafos': ['islpy', 'py-cpuinfo'], 'alltrafos': ['islpy', 'py-cpuinfo'],
'bench_db': ['blitzdb', 'pymongo', 'pandas'], 'bench_db': ['blitzdb', 'pymongo', 'pandas'],
'interactive': ['matplotlib', 'ipy_table', 'imageio', 'jupyter', 'pyevtk'], 'interactive': ['matplotlib', 'ipy_table', 'imageio', 'jupyter', 'pyevtk'],
'autodiff': ['pystencils_autodiff'],
'doc': ['sphinx', 'sphinx_rtd_theme', 'nbsphinx', 'doc': ['sphinx', 'sphinx_rtd_theme', 'nbsphinx',
'sphinxcontrib-bibtex', 'sphinx_autodoc_typehints', 'pandoc'], 'sphinxcontrib-bibtex', 'sphinx_autodoc_typehints', 'pandoc'],
}, },
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment