From 6300e4a12a251cff2bf5956cbb79a1b07d19189b Mon Sep 17 00:00:00 2001 From: billmccartney Date: Mon, 18 Dec 2023 16:44:38 -0500 Subject: [PATCH] Fixed wheel building on windows by adding path to included header files. --- python/setup.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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()]