From 6bf6bda82b4c9545960cf0abdc8f125fe828c8ce Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Thu, 8 Aug 2019 21:04:07 -0500
Subject: [PATCH] Fix numpy import for pypy
---
src/wrap_cl.cpp | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/src/wrap_cl.cpp b/src/wrap_cl.cpp
index 50a48201..b9393ed0 100644
--- a/src/wrap_cl.cpp
+++ b/src/wrap_cl.cpp
@@ -43,7 +43,11 @@ extern void pyopencl_expose_mempool(py::module &m);
static bool import_numpy_helper()
{
- import_array1(false);
+#ifdef PYPY_VERSION
+ import_array();
+#else
+ import_array1(false);
+#endif
return true;
}
--
GitLab