add include dir

Signed-off-by: Jinzhe Zeng <[email protected]>
tongzhugroup · Apr 20, 2024 · 408aecc · 408aecc
1 parent feeb1b4
commit 408aecc
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -19,6 +19,7 @@ add_custom_command(
 python_add_library(
   dps MODULE ${CMAKE_CURRENT_BINARY_DIR}/dps.cpp
   ${CMAKE_CURRENT_SOURCE_DIR}/mddatasetbuilder/c_stack.cpp WITH_SOABI)
+target_include_directories(dps PRIVATE ${CMAKE_CURRENT_SOURCE_DIR}/mddatasetbuilder)
 target_compile_definitions(dps PRIVATE CYTHON_LIMITED_API=1
                                        Py_LIMITED_API=0x03070000)
 if(CMAKE_BUILD_TYPE STREQUAL "Debug")