mirror da
https://github.com/rspamd/rspamd.git
synced 2024-07-29 20:17:47 +02:00
52 righe
1.3 KiB
CMake
52 righe
1.3 KiB
CMake
LIST(APPEND CMAKE_MODULE_PATH "${PROJECT_SOURCE_DIR}")
|
|
|
|
SET(src
|
|
init.c
|
|
hash_map.c
|
|
)
|
|
SET(luasrc
|
|
_env.lua
|
|
benchmark.lua
|
|
CartNode.lua
|
|
CartTrainer.lua
|
|
CartTree.lua
|
|
DataSet.lua
|
|
DecisionForest.lua
|
|
DecisionForestTrainer.lua
|
|
DecisionTree.lua
|
|
DFD.lua
|
|
GiniState.lua
|
|
GradientBoostState.lua
|
|
GradientBoostTrainer.lua
|
|
init.lua
|
|
LogitBoostCriterion.lua
|
|
math.lua
|
|
MSECriterion.lua
|
|
RandomForestTrainer.lua
|
|
Sparse2Dense.lua
|
|
SparseTensor.lua
|
|
test.lua
|
|
TreeState.lua
|
|
utils.lua
|
|
WorkPool.lua
|
|
)
|
|
|
|
IF (WITH_OPENMP)
|
|
FIND_PACKAGE(OpenMP)
|
|
IF(OPENMP_FOUND)
|
|
MESSAGE(STATUS "Compiling with OpenMP support")
|
|
SET(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} ${OpenMP_C_FLAGS}")
|
|
SET(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${OpenMP_CXX_FLAGS}")
|
|
SET(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} ${OpenMP_EXE_LINKER_FLAGS}")
|
|
ENDIF(OPENMP_FOUND)
|
|
ENDIF (WITH_OPENMP)
|
|
|
|
ADD_TORCH_PACKAGE(decisiontree "${src}" "${luasrc}" "A decision tree library, for Torch")
|
|
INCLUDE_DIRECTORIES(${CMAKE_CURRENT_SOURCE_DIR})
|
|
### Torch packages supposes libraries prefix is "lib"
|
|
SET_TARGET_PROPERTIES(decisiontree PROPERTIES
|
|
PREFIX "lib"
|
|
IMPORT_PREFIX "lib")
|
|
TARGET_LINK_LIBRARIES(decisiontree ${TH_LIBRARIES})
|
|
INSTALL(TARGETS decisiontree DESTINATION ${RSPAMD_LIBDIR})
|