Fix CUDA and mdlp library issues

This commit is contained in:
2024-11-20 21:02:56 +01:00
parent 9a323cd7a3
commit f0f3d9ad6e
20 changed files with 40 additions and 493 deletions

View File

@@ -4,9 +4,10 @@ include_directories(
${BayesNet_SOURCE_DIR}/lib/json/include
${BayesNet_SOURCE_DIR}
${CMAKE_BINARY_DIR}/configured_files/include
${FImdlp_INCLUDE_DIRS}
)
file(GLOB_RECURSE Sources "*.cc")
add_library(BayesNet ${Sources})
target_link_libraries(BayesNet mdlp "${TORCH_LIBRARIES}")
target_link_libraries(BayesNet ${FImdlp} "${TORCH_LIBRARIES}")

View File

@@ -9,7 +9,7 @@
#include <string>
#include <map>
#include <torch/torch.h>
#include <CPPFImdlp.h>
#include <fimdlp/CPPFImdlp.h>
#include "bayesnet/network/Network.h"
#include "Classifier.h"