From 2f9c7b9350ba0ae35ce5ce664b6fe2961f0dd8c6 Mon Sep 17 00:00:00 2001 From: Thomas Gibson <gibsonthomas1120@hotmail.com> Date: Fri, 21 May 2021 15:43:03 -0500 Subject: [PATCH] Add actx.np.concatenate --- arraycontext/impl/pyopencl.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/arraycontext/impl/pyopencl.py b/arraycontext/impl/pyopencl.py index 80ea558..1f632a5 100644 --- a/arraycontext/impl/pyopencl.py +++ b/arraycontext/impl/pyopencl.py @@ -129,6 +129,14 @@ class PyOpenCLFakeNumpyNamespace(BaseFakeNumpyNamespace): queue=self._array_context.queue), *arrays) + def concatenate(self, arrays, axis=0): + import pyopencl.array as cla + return cla.concatenate( + arrays, axis, + self._array_context.queue, + self._array_context.allocator + ) + # }}} -- GitLab