From 94b66a2c018d0e458706b3cfb5ad881bb0ee03ca Mon Sep 17 00:00:00 2001 From: Alexandru Fikl Date: Fri, 22 Oct 2021 16:06:18 -0500 Subject: [PATCH] add astype to pyopencl context --- arraycontext/impl/pyopencl/fake_numpy.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/arraycontext/impl/pyopencl/fake_numpy.py b/arraycontext/impl/pyopencl/fake_numpy.py index c8c1715a..0d955e57 100644 --- a/arraycontext/impl/pyopencl/fake_numpy.py +++ b/arraycontext/impl/pyopencl/fake_numpy.py @@ -267,6 +267,11 @@ def rec_equal(x, y): result = result.get()[()] return result + def astype(self, a, dtype): + return rec_map_array_container( + lambda ary: ary.astype(dtype, queue=self._array_context.queue), + a) + # }}}