Skip to content

Commit be1429c

Browse files
alexfiklinducer
alexfikl
authored and
inducer
committed
update numpy coverage
1 parent 12e173f commit be1429c

File tree

2 files changed

+104
-0
lines changed

2 files changed

+104
-0
lines changed

‎doc/make_numpy_coverage_table.py‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,9 @@ def initialize_contexts():
6262

6363
return [
6464
arraycontext.PyOpenCLArrayContext(queue, force_device_scalars=True),
65+
arraycontext.EagerJAXArrayContext(),
6566
arraycontext.PytatoPyOpenCLArrayContext(queue),
67+
arraycontext.PytatoJAXArrayContext(),
6668
]
6769

6870

‎doc/numpy_coverage.rst‎

Lines changed: 102 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,22 +15,34 @@ Array creation routines
1515

1616
* - Function
1717
- :class:`~arraycontext.PyOpenCLArrayContext`
18+
- :class:`~arraycontext.EagerJAXArrayContext`
1819
- :class:`~arraycontext.PytatoPyOpenCLArrayContext`
20+
- :class:`~arraycontext.PytatoJAXArrayContext`
1921
* - :func:`numpy.empty_like`
2022
- :green:`Yes`
2123
- :green:`Yes`
24+
- :green:`Yes`
25+
- :green:`Yes`
2226
* - :func:`numpy.ones_like`
2327
- :green:`Yes`
2428
- :green:`Yes`
29+
- :green:`Yes`
30+
- :green:`Yes`
2531
* - :func:`numpy.zeros_like`
2632
- :green:`Yes`
2733
- :green:`Yes`
34+
- :green:`Yes`
35+
- :green:`Yes`
2836
* - :func:`numpy.full_like`
2937
- :green:`Yes`
3038
- :green:`Yes`
39+
- :green:`Yes`
40+
- :green:`Yes`
3141
* - :func:`numpy.copy`
42+
- :green:`Yes`
3243
- :green:`Yes`
3344
- :red:`No`
45+
- :red:`No`
3446

3547
Array manipulation routines
3648
~~~~~~~~~~~~~~~~~~~~~~~~~~~
@@ -40,25 +52,39 @@ Array manipulation routines
4052

4153
* - Function
4254
- :class:`~arraycontext.PyOpenCLArrayContext`
55+
- :class:`~arraycontext.EagerJAXArrayContext`
4356
- :class:`~arraycontext.PytatoPyOpenCLArrayContext`
57+
- :class:`~arraycontext.PytatoJAXArrayContext`
4458
* - :func:`numpy.reshape`
4559
- :green:`Yes`
4660
- :green:`Yes`
61+
- :green:`Yes`
62+
- :green:`Yes`
4763
* - :func:`numpy.ravel`
4864
- :green:`Yes`
4965
- :green:`Yes`
66+
- :green:`Yes`
67+
- :green:`Yes`
5068
* - :func:`numpy.transpose`
5169
- :red:`No`
5270
- :green:`Yes`
71+
- :green:`Yes`
72+
- :green:`Yes`
5373
* - :func:`numpy.broadcast_to`
5474
- :red:`No`
5575
- :green:`Yes`
76+
- :green:`Yes`
77+
- :green:`Yes`
5678
* - :func:`numpy.concatenate`
5779
- :green:`Yes`
5880
- :green:`Yes`
81+
- :green:`Yes`
82+
- :green:`Yes`
5983
* - :func:`numpy.stack`
6084
- :green:`Yes`
6185
- :green:`Yes`
86+
- :green:`Yes`
87+
- :green:`Yes`
6288

6389
Linear algebra
6490
~~~~~~~~~~~~~~
@@ -68,10 +94,14 @@ Linear algebra
6894

6995
* - Function
7096
- :class:`~arraycontext.PyOpenCLArrayContext`
97+
- :class:`~arraycontext.EagerJAXArrayContext`
7198
- :class:`~arraycontext.PytatoPyOpenCLArrayContext`
99+
- :class:`~arraycontext.PytatoJAXArrayContext`
72100
* - :func:`numpy.vdot`
101+
- :green:`Yes`
73102
- :green:`Yes`
74103
- :red:`No`
104+
- :red:`No`
75105

76106
Logic Functions
77107
~~~~~~~~~~~~~~~
@@ -81,31 +111,49 @@ Logic Functions
81111

82112
* - Function
83113
- :class:`~arraycontext.PyOpenCLArrayContext`
114+
- :class:`~arraycontext.EagerJAXArrayContext`
84115
- :class:`~arraycontext.PytatoPyOpenCLArrayContext`
116+
- :class:`~arraycontext.PytatoJAXArrayContext`
85117
* - :func:`numpy.all`
86118
- :green:`Yes`
87119
- :green:`Yes`
120+
- :green:`Yes`
121+
- :green:`Yes`
88122
* - :func:`numpy.any`
89123
- :green:`Yes`
90124
- :green:`Yes`
125+
- :green:`Yes`
126+
- :green:`Yes`
91127
* - :data:`numpy.greater`
92128
- :green:`Yes`
93129
- :green:`Yes`
130+
- :green:`Yes`
131+
- :green:`Yes`
94132
* - :data:`numpy.greater_equal`
95133
- :green:`Yes`
96134
- :green:`Yes`
135+
- :green:`Yes`
136+
- :green:`Yes`
97137
* - :data:`numpy.less`
98138
- :green:`Yes`
99139
- :green:`Yes`
140+
- :green:`Yes`
141+
- :green:`Yes`
100142
* - :data:`numpy.less_equal`
101143
- :green:`Yes`
102144
- :green:`Yes`
145+
- :green:`Yes`
146+
- :green:`Yes`
103147
* - :data:`numpy.equal`
104148
- :green:`Yes`
105149
- :green:`Yes`
150+
- :green:`Yes`
151+
- :green:`Yes`
106152
* - :data:`numpy.not_equal`
107153
- :green:`Yes`
108154
- :green:`Yes`
155+
- :green:`Yes`
156+
- :green:`Yes`
109157

110158
Mathematical functions
111159
~~~~~~~~~~~~~~~~~~~~~~
@@ -115,82 +163,136 @@ Mathematical functions
115163

116164
* - Function
117165
- :class:`~arraycontext.PyOpenCLArrayContext`
166+
- :class:`~arraycontext.EagerJAXArrayContext`
118167
- :class:`~arraycontext.PytatoPyOpenCLArrayContext`
168+
- :class:`~arraycontext.PytatoJAXArrayContext`
119169
* - :data:`numpy.sin`
120170
- :green:`Yes`
121171
- :green:`Yes`
172+
- :green:`Yes`
173+
- :green:`Yes`
122174
* - :data:`numpy.cos`
123175
- :green:`Yes`
124176
- :green:`Yes`
177+
- :green:`Yes`
178+
- :green:`Yes`
125179
* - :data:`numpy.tan`
126180
- :green:`Yes`
127181
- :green:`Yes`
182+
- :green:`Yes`
183+
- :green:`Yes`
128184
* - :data:`numpy.arcsin`
129185
- :green:`Yes`
130186
- :green:`Yes`
187+
- :green:`Yes`
188+
- :green:`Yes`
131189
* - :data:`numpy.arccos`
132190
- :green:`Yes`
133191
- :green:`Yes`
192+
- :green:`Yes`
193+
- :green:`Yes`
134194
* - :data:`numpy.arctan`
135195
- :green:`Yes`
136196
- :green:`Yes`
197+
- :green:`Yes`
198+
- :green:`Yes`
137199
* - :data:`numpy.arctan2`
138200
- :green:`Yes`
139201
- :green:`Yes`
202+
- :green:`Yes`
203+
- :green:`Yes`
140204
* - :data:`numpy.sinh`
141205
- :green:`Yes`
142206
- :green:`Yes`
207+
- :green:`Yes`
208+
- :green:`Yes`
143209
* - :data:`numpy.cosh`
144210
- :green:`Yes`
145211
- :green:`Yes`
212+
- :green:`Yes`
213+
- :green:`Yes`
146214
* - :data:`numpy.tanh`
147215
- :green:`Yes`
148216
- :green:`Yes`
217+
- :green:`Yes`
218+
- :green:`Yes`
149219
* - :data:`numpy.floor`
150220
- :green:`Yes`
151221
- :green:`Yes`
222+
- :green:`Yes`
223+
- :green:`Yes`
152224
* - :data:`numpy.ceil`
153225
- :green:`Yes`
154226
- :green:`Yes`
227+
- :green:`Yes`
228+
- :green:`Yes`
155229
* - :func:`numpy.sum`
156230
- :green:`Yes`
157231
- :green:`Yes`
232+
- :green:`Yes`
233+
- :green:`Yes`
158234
* - :data:`numpy.exp`
159235
- :green:`Yes`
160236
- :green:`Yes`
237+
- :green:`Yes`
238+
- :green:`Yes`
161239
* - :data:`numpy.log`
162240
- :green:`Yes`
163241
- :green:`Yes`
242+
- :green:`Yes`
243+
- :green:`Yes`
164244
* - :data:`numpy.log10`
165245
- :green:`Yes`
166246
- :green:`Yes`
247+
- :green:`Yes`
248+
- :green:`Yes`
167249
* - :func:`numpy.real`
168250
- :green:`Yes`
169251
- :green:`Yes`
252+
- :green:`Yes`
253+
- :green:`Yes`
170254
* - :func:`numpy.imag`
171255
- :green:`Yes`
172256
- :green:`Yes`
257+
- :green:`Yes`
258+
- :green:`Yes`
173259
* - :data:`numpy.conjugate`
174260
- :green:`Yes`
175261
- :green:`Yes`
262+
- :green:`Yes`
263+
- :green:`Yes`
176264
* - :data:`numpy.maximum`
177265
- :green:`Yes`
178266
- :green:`Yes`
267+
- :green:`Yes`
268+
- :green:`Yes`
179269
* - :func:`numpy.amax`
180270
- :green:`Yes`
181271
- :green:`Yes`
272+
- :green:`Yes`
273+
- :green:`Yes`
182274
* - :data:`numpy.minimum`
183275
- :green:`Yes`
184276
- :green:`Yes`
277+
- :green:`Yes`
278+
- :green:`Yes`
185279
* - :func:`numpy.amin`
186280
- :green:`Yes`
187281
- :green:`Yes`
282+
- :green:`Yes`
283+
- :green:`Yes`
188284
* - :data:`numpy.sqrt`
189285
- :green:`Yes`
190286
- :green:`Yes`
287+
- :green:`Yes`
288+
- :green:`Yes`
191289
* - :data:`numpy.absolute`
192290
- :green:`Yes`
193291
- :green:`Yes`
292+
- :green:`Yes`
293+
- :green:`Yes`
194294
* - :data:`numpy.fabs`
195295
- :green:`Yes`
196296
- :green:`Yes`
297+
- :green:`Yes`
298+
- :green:`Yes`

0 commit comments

Comments
 (0)