Make sure packages related to pyopencl are fully installed.

vasole · Apr 9, 2018 · 86e8fcf · 86e8fcf
1 parent 59692e5
commit 86e8fcf
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/cx_setup.py b/cx_setup.py
@@ -233,7 +233,13 @@ def dummy(*var, **kw):
     special_modules.append(os.path.dirname(h5py.__file__))
 if OPENCL:
     special_modules.append(os.path.dirname(pyopencl.__file__))
-    includes.append("pytools")
+    import mako
+    import cffi
+    import pytools
+    special_modules.append(os.path.dirname(mako.__file__))
+    special_modules.append(os.path.dirname(cffi.__file__))
+    special_modules.append(os.path.dirname(pytools.__file__))
+    #includes.append("pytools")
     includes.append("decorator")
 else:
     excludes.append("pyopencl")