diff --git a/README.md b/README.md index d079128a539b9c9244d39dc5fc0a6c5e88055227..60ab1dd3e0cebc9ba6948ace01382dc127176711 100644 --- a/README.md +++ b/README.md @@ -52,6 +52,7 @@ Without `[interactive]` you get a minimal version with very little dependencies. All options: - `gpu`: use this if an Nvidia GPU is available and CUDA is installed +- `pyopencl`: basic OpenCL support (experimental) - `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. diff --git a/setup.py b/setup.py index 6959f09c28b6d9508802c4d0b1c81999e1b4463c..3da19ffedeffdc88c2c4b5424a7a1e26983621b8 100644 --- a/setup.py +++ b/setup.py @@ -106,6 +106,7 @@ setup(name='pystencils', }, extras_require={ 'gpu': ['pycuda'], + 'opencl': ['pyopencl'], 'alltrafos': ['islpy', 'py-cpuinfo'], 'bench_db': ['blitzdb', 'pymongo', 'pandas'], 'interactive': ['matplotlib', 'ipy_table', 'imageio', 'jupyter', 'pyevtk'],