From 500573ac388c7a7ad60a4a1262a3a6088810cae1 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Sun, 17 Sep 2023 19:42:40 -0500
Subject: [PATCH] Add arange to pyopencl actx
---
arraycontext/impl/pyopencl/fake_numpy.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/arraycontext/impl/pyopencl/fake_numpy.py b/arraycontext/impl/pyopencl/fake_numpy.py
index a379f29..6044e0c 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):
# }}}
-
# }}}
--
GitLab