File tree Expand file tree Collapse file tree 6 files changed +15
-6
lines changed
Expand file tree Collapse file tree 6 files changed +15
-6
lines changed Original file line number Diff line number Diff line change @@ -16,6 +16,7 @@ code to work with all of them? No problem! Comes with pre-made array context
1616implementations for:
1717
1818- numpy
19+ - cupy
1920- `PyOpenCL <https://documen.tician.de/pyopencl/array.html>`__
2021- `JAX <https://jax.readthedocs.io/en/latest/>`__
2122- `Pytato <https://documen.tician.de/pytato>`__ (for lazy/deferred evaluation)
Original file line number Diff line number Diff line change 1- from __future__ import annotations
2-
3-
41"""
52.. currentmodule:: arraycontext
63
7- A mod :`cupy`-based array context.
4+ A : mod:`cupy`-based array context.
85
96.. autoclass:: CupyArrayContext
107"""
118
9+ from __future__ import annotations
10+
11+
1212__copyright__ = """
1313Copyright (C) 2024 University of Illinois Board of Trustees
1414"""
Original file line number Diff line number Diff line change 33
44# You can set these variables from the command line, and also
55# from the environment for the first two.
6- SPHINXOPTS ?=
6+ SPHINXOPTS ?= -W -n
77SPHINXBUILD ?= python $(shell which sphinx-build)
88SOURCEDIR = .
99BUILDDIR = _build
Original file line number Diff line number Diff line change 2323 "pytest": ("https://docs.pytest.org/en/latest/", None),
2424 "python": ("https://docs.python.org/3/", None),
2525 "pytools": ("https://documen.tician.de/pytools", None),
26+ "cupy": ("https://docs.cupy.dev/en/stable/", None),
2627}
2728
2829# Some modules need to import things just so that sphinx can resolve symbols in
Original file line number Diff line number Diff line change @@ -13,6 +13,12 @@ Array context based on :mod:`numpy`
1313
1414.. automodule:: arraycontext.impl.numpy
1515
16+
17+ Array context based on :mod:`cupy`
18+ --------------------------------------------
19+
20+ .. automodule:: arraycontext.impl.cupy
21+
1622Array context based on :mod:`pyopencl.array`
1723--------------------------------------------
1824
Original file line number Diff line number Diff line change @@ -6,14 +6,15 @@ code to work with all of them? No problem! Comes with pre-made array context
66implementations for:
77
88- :mod:`numpy`
9+ - :mod:`cupy`
910- :mod:`pyopencl`
1011- :mod:`jax.numpy`
1112- :mod:`pytato` (for lazy/deferred evaluation)
1213- Debugging
1314- Profiling
1415
1516:mod:`arraycontext` started life as an array abstraction for use with the
16- :mod:`meshmode` unstrucuted discretization package.
17+ :mod:`meshmode` unstructured discretization package.
1718
1819Design Guidelines
1920-----------------
You can’t perform that action at this time.
0 commit comments