diff --git a/python/setup.py b/python/setup.py index afa674c..2e0a59b 100644 --- a/python/setup.py +++ b/python/setup.py @@ -20,7 +20,8 @@ except Exception: nthreads = 0 -include_dirs = [os.path.dirname(sysconfig.get_path("include")),] +include_dirs = [os.path.dirname(sysconfig.get_path("include")), + os.path.join(os.path.dirname(os.path.abspath(__file__)), "..", "c", "include")] if os.getenv("CUDA_HOME"): include_dirs.insert(0, os.path.join(os.environ["CUDA_HOME"], "include")) library_dirs = [get_python_lib()]