From 004e85feede3df5b4063503a3774a49627872ef0 Mon Sep 17 00:00:00 2001
From: Alexandru Fikl <alexfikl@gmail.com>
Date: Mon, 18 Oct 2021 16:20:48 -0500
Subject: [PATCH] pyopencl: remove unused astype
---
arraycontext/impl/pyopencl/fake_numpy.py | 5 -----
arraycontext/impl/pytato/fake_numpy.py | 5 +++--
2 files changed, 3 insertions(+), 7 deletions(-)
diff --git a/arraycontext/impl/pyopencl/fake_numpy.py b/arraycontext/impl/pyopencl/fake_numpy.py
index c60f33d..70689fc 100644
--- a/arraycontext/impl/pyopencl/fake_numpy.py
+++ b/arraycontext/impl/pyopencl/fake_numpy.py
@@ -267,11 +267,6 @@ class PyOpenCLFakeNumpyNamespace(BaseFakeNumpyNamespace):
result = result.get()[()]
return result
- def astype(self, a, dtype):
- return rec_map_array_container(
- lambda x: x.astype(dtype, queue=self._array_context.queue),
- a)
-
# }}}
diff --git a/arraycontext/impl/pytato/fake_numpy.py b/arraycontext/impl/pytato/fake_numpy.py
index 62b5e20..9a92695 100644
--- a/arraycontext/impl/pytato/fake_numpy.py
+++ b/arraycontext/impl/pytato/fake_numpy.py
@@ -65,8 +65,9 @@ class PytatoFakeNumpyNamespace(BaseFakeNumpyNamespace):
return super().__getattr__(name)
def reshape(self, a, newshape, order="C"):
- return rec_multimap_array_container(
- partial(pt.reshape, order=order), a, newshape)
+ return rec_map_array_container(
+ lambda ary: pt.reshape(a, newshape, order=order),
+ a)
def transpose(self, a, axes=None):
return rec_multimap_array_container(pt.transpose, a, axes)
--
GitLab