Update CMakeLists

This commit is contained in:
2024-04-10 12:55:23 +02:00
parent a660c66c01
commit 1e88222775
2 changed files with 3 additions and 2 deletions

View File

@@ -65,7 +65,8 @@ endif (ENABLE_CLANG_TIDY)
# External libraries - dependencies of PyClassifiers # External libraries - dependencies of PyClassifiers
# -------------------------------------------------- # --------------------------------------------------
find_library(BayesNet NAMES libBayesNet BayesNet) find_library(BayesNet NAMES libBayesNet BayesNet libBayesNet.a PATHS ${Platform_SOURCE_DIR}/../lib/lib REQUIRED)
find_path(Bayesnet_INCLUDE_DIRS REQUIRED NAMES bayesnet PATHS ${Platform_SOURCE_DIR}/../lib/include)
# Subdirectories # Subdirectories
# -------------- # --------------

View File

@@ -2,7 +2,7 @@ include_directories(
${Python3_INCLUDE_DIRS} ${Python3_INCLUDE_DIRS}
${TORCH_INCLUDE_DIRS} ${TORCH_INCLUDE_DIRS}
${PyClassifiers_SOURCE_DIR}/lib/json/include ${PyClassifiers_SOURCE_DIR}/lib/json/include
/usr/local/include ${Bayesnet_INCLUDE_DIRS}
) )
add_library(PyClassifiers ODTE.cc STree.cc SVC.cc RandomForest.cc XGBoost.cc PyClassifier.cc PyWrap.cc) add_library(PyClassifiers ODTE.cc STree.cc SVC.cc RandomForest.cc XGBoost.cc PyClassifier.cc PyWrap.cc)
target_link_libraries(PyClassifiers ${Python3_LIBRARIES} "${TORCH_LIBRARIES}" ${LIBTORCH_PYTHON} Boost::boost Boost::python Boost::numpy) target_link_libraries(PyClassifiers ${Python3_LIBRARIES} "${TORCH_LIBRARIES}" ${LIBTORCH_PYTHON} Boost::boost Boost::python Boost::numpy)