Skip to content

Commit b24260d

Browse files
kaushikcfdinducer
authored andcommitted
register real/imag as pytato functions
1 parent 55fa4b0 commit b24260d

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

arraycontext/impl/pytato/fake_numpy.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ class PytatoFakeNumpyNamespace(LoopyBasedFakeNumpyNamespace):
5353
_pt_funcs = frozenset({
5454
"sin", "cos", "tan", "arcsin", "arccos", "arctan",
5555
"sinh", "cosh", "tanh", "exp", "log", "log10",
56-
"sqrt", "abs", "isnan"
56+
"sqrt", "abs", "isnan", "real", "imag", "conj",
5757
})
5858

5959
def _get_fake_numpy_linalg_namespace(self):
@@ -209,9 +209,6 @@ def _pt_sum(ary):
209209

210210
return rec_map_reduce_array_container(sum, _pt_sum, a)
211211

212-
def conj(self, x):
213-
return rec_multimap_array_container(pt.conj, x)
214-
215212
def maximum(self, x, y):
216213
return rec_multimap_array_container(pt.maximum, x, y)
217214

0 commit comments

Comments
 (0)