diff --git a/arraycontext/impl/pyopencl/fake_numpy.py b/arraycontext/impl/pyopencl/fake_numpy.py index a379f29006d6d66270872e894169f13f153d8e5f..6044e0c9e46e8f6455b798a471046055a200920e 100644 --- a/arraycontext/impl/pyopencl/fake_numpy.py +++ b/arraycontext/impl/pyopencl/fake_numpy.py @@ -107,6 +107,9 @@ class PyOpenCLFakeNumpyNamespace(LoopyBasedFakeNumpyNamespace): return self._array_context._rec_map_container(_copy, ary) + def arange(self, *args, **kwargs): + return cl_array.arange(self._array_context.queue, *args, **kwargs) + # }}} # {{{ array manipulation routines @@ -360,7 +363,6 @@ class PyOpenCLFakeNumpyNamespace(LoopyBasedFakeNumpyNamespace): # }}} - # }}}