diff --git a/src/pystencils/jit/cpu/cpu_pybind11.py b/src/pystencils/jit/cpu/cpu_pybind11.py index d0368a39089e53330b96235fbf3347c5b969686f..0229c5c4d671a412fe3ba3c26711c703459a9d5d 100644 --- a/src/pystencils/jit/cpu/cpu_pybind11.py +++ b/src/pystencils/jit/cpu/cpu_pybind11.py @@ -44,17 +44,22 @@ class CpuJit(JitBase): # Include Directories import pybind11 as pb11 - self._pybind11_include = pb11.get_include() + pybind11_include = pb11.get_include() import sysconfig - self._py_include = sysconfig.get_path("include") + python_include = sysconfig.get_path("include") + + from ...include import get_pystencils_include_path + + pystencils_include = get_pystencils_include_path() self._cxx_fixed_flags = [ "-shared", "-fPIC", - f"-I{self._py_include}", - f"-I{self._pybind11_include}", + f"-I{python_include}", + f"-I{pybind11_include}", + f"-I{pystencils_include}" ] @property