Skip to content

Commit

Permalink
update numpy coverage
Browse files Browse the repository at this point in the history
  • Loading branch information
@alexfikl @inducer
alexfikl authored and inducer committed Jun 26, 2022
1 parent 12e173f commit be1429c
Show file tree
Hide file tree
Showing 2 changed files with 104 additions and 0 deletions.
2 changes: 2 additions & 0 deletions doc/make_numpy_coverage_table.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -62,7 +62,9 @@ def initialize_contexts():

return [
arraycontext.PyOpenCLArrayContext(queue, force_device_scalars=True),
arraycontext.EagerJAXArrayContext(),
arraycontext.PytatoPyOpenCLArrayContext(queue),
arraycontext.PytatoJAXArrayContext(),
]


Expand Down
102 changes: 102 additions & 0 deletions doc/numpy_coverage.rst
View file
Original file line number Diff line number Diff line change
Expand Up @@ -15,22 +15,34 @@ Array creation routines

* - Function
- :class:`~arraycontext.PyOpenCLArrayContext`
- :class:`~arraycontext.EagerJAXArrayContext`
- :class:`~arraycontext.PytatoPyOpenCLArrayContext`
- :class:`~arraycontext.PytatoJAXArrayContext`
* - :func:`numpy.empty_like`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :func:`numpy.ones_like`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :func:`numpy.zeros_like`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :func:`numpy.full_like`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :func:`numpy.copy`
- :green:`Yes`
- :green:`Yes`
- :red:`No`
- :red:`No`

Array manipulation routines
~~~~~~~~~~~~~~~~~~~~~~~~~~~
Expand All @@ -40,25 +52,39 @@ Array manipulation routines

* - Function
- :class:`~arraycontext.PyOpenCLArrayContext`
- :class:`~arraycontext.EagerJAXArrayContext`
- :class:`~arraycontext.PytatoPyOpenCLArrayContext`
- :class:`~arraycontext.PytatoJAXArrayContext`
* - :func:`numpy.reshape`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :func:`numpy.ravel`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :func:`numpy.transpose`
- :red:`No`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :func:`numpy.broadcast_to`
- :red:`No`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :func:`numpy.concatenate`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :func:`numpy.stack`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`

Linear algebra
~~~~~~~~~~~~~~
Expand All @@ -68,10 +94,14 @@ Linear algebra

* - Function
- :class:`~arraycontext.PyOpenCLArrayContext`
- :class:`~arraycontext.EagerJAXArrayContext`
- :class:`~arraycontext.PytatoPyOpenCLArrayContext`
- :class:`~arraycontext.PytatoJAXArrayContext`
* - :func:`numpy.vdot`
- :green:`Yes`
- :green:`Yes`
- :red:`No`
- :red:`No`

Logic Functions
~~~~~~~~~~~~~~~
Expand All @@ -81,31 +111,49 @@ Logic Functions

* - Function
- :class:`~arraycontext.PyOpenCLArrayContext`
- :class:`~arraycontext.EagerJAXArrayContext`
- :class:`~arraycontext.PytatoPyOpenCLArrayContext`
- :class:`~arraycontext.PytatoJAXArrayContext`
* - :func:`numpy.all`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :func:`numpy.any`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :data:`numpy.greater`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :data:`numpy.greater_equal`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :data:`numpy.less`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :data:`numpy.less_equal`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :data:`numpy.equal`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :data:`numpy.not_equal`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`

Mathematical functions
~~~~~~~~~~~~~~~~~~~~~~
Expand All @@ -115,82 +163,136 @@ Mathematical functions

* - Function
- :class:`~arraycontext.PyOpenCLArrayContext`
- :class:`~arraycontext.EagerJAXArrayContext`
- :class:`~arraycontext.PytatoPyOpenCLArrayContext`
- :class:`~arraycontext.PytatoJAXArrayContext`
* - :data:`numpy.sin`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :data:`numpy.cos`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :data:`numpy.tan`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :data:`numpy.arcsin`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :data:`numpy.arccos`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :data:`numpy.arctan`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :data:`numpy.arctan2`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :data:`numpy.sinh`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :data:`numpy.cosh`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :data:`numpy.tanh`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :data:`numpy.floor`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :data:`numpy.ceil`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :func:`numpy.sum`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :data:`numpy.exp`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :data:`numpy.log`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :data:`numpy.log10`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :func:`numpy.real`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :func:`numpy.imag`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :data:`numpy.conjugate`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :data:`numpy.maximum`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :func:`numpy.amax`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :data:`numpy.minimum`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :func:`numpy.amin`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :data:`numpy.sqrt`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :data:`numpy.absolute`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
* - :data:`numpy.fabs`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`
- :green:`Yes`

0 comments on commit be1429c

Please sign in to comment.