Merge pull request #413 from CNugteren/CLBlast-412-python-runtime-libs

Add library dir on Linux for pyclblast

1 files changed, 8 insertions, 0 deletions

diff --git a/src/pyclblast/setup.py b/src/pyclblast/setup.py
index 3da1e454..3ae8d414 100644
--- a/src/pyclblast/setup.py
+++ b/src/pyclblast/setup.py
@@ -10,12 +10,20 @@ from setuptools import setup
 from distutils.extension import Extension
 from Cython.Distutils import build_ext
 
+runtime_library_dirs = list()
+if platform.system() == "Linux":
+    runtime_library_dirs.append("/usr/local/lib")
+elif platform.system() == "Windows":
+    runtime_library_dirs.append("C:\\Program Files\\CLBlast\\lib")
+    runtime_library_dirs.append("C:\\Program Files (x86)\\CLBlast\\lib")
+
 ext_modules = list()
 ext_modules.append(
     Extension(
         "pyclblast",
         ["src/pyclblast.pyx"],
         libraries=["clblast"],
+        runtime_library_dirs=runtime_library_dirs,
         language="c++"
     )
 )