File tree Expand file tree Collapse file tree 8 files changed +18
-9
lines changed Expand file tree Collapse file tree 8 files changed +18
-9
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"""
@@ -64,7 +64,7 @@ class CupyNonObjectArray(metaclass=CupyNonObjectArrayMetaclass):
6464
6565class CupyArrayContext (ArrayContext ):
6666 """
67- A :class:`ArrayContext` that uses :class:`cupy.ndarray` to represent arrays.
67+ An :class:`ArrayContext` that uses :class:`cupy.ndarray` to represent arrays.
6868
6969 .. automethod:: __init__
7070 """
Original file line number Diff line number Diff line change @@ -63,7 +63,7 @@ class NumpyNonObjectArray(metaclass=NumpyNonObjectArrayMetaclass):
6363
6464class NumpyArrayContext (ArrayContext ):
6565 """
66- A :class:`ArrayContext` that uses :class:`numpy.ndarray` to represent arrays.
66+ An :class:`ArrayContext` that uses :class:`numpy.ndarray` to represent arrays.
6767
6868 .. automethod:: __init__
6969 """
Original file line number Diff line number Diff line change 5858
5959class PyOpenCLArrayContext (ArrayContext ):
6060 """
61- A :class:`ArrayContext` that uses :class:`pyopencl.array.Array` instances
61+ An :class:`ArrayContext` that uses :class:`pyopencl.array.Array` instances
6262 for its base array class.
6363
6464 .. attribute:: context
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