CMakeLists.txt 674 Bytes
Newer Older
Guolin Ke's avatar
Guolin Ke committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
include_directories(${LightGBM_HEADER_DIR})

if(USE_MPI)
  include_directories(${MPI_CXX_INCLUDE_PATH})
endif()

AUX_SOURCE_DIRECTORY(./application/ APPLICATION_SRC) 
AUX_SOURCE_DIRECTORY(./boosting/ BOOSTING_SRC) 
AUX_SOURCE_DIRECTORY(./io/ IO_SRC) 
AUX_SOURCE_DIRECTORY(./metric/ METRIC_SRC) 
AUX_SOURCE_DIRECTORY(./objective/ OBJECTIVE_SRC) 
AUX_SOURCE_DIRECTORY(./network/ NETWORK_SRC) 
AUX_SOURCE_DIRECTORY(./treelearner/ TREELEARNER_SRC) 

Guolin Ke's avatar
Guolin Ke committed
15
add_executable(lightgbm main.cpp ${APPLICATION_SRC} ${BOOSTING_SRC} ${IO_SRC} ${METRIC_SRC} ${OBJECTIVE_SRC} ${NETWORK_SRC} ${TREELEARNER_SRC})
Guolin Ke's avatar
Guolin Ke committed
16
17

if(USE_MPI)
Guolin Ke's avatar
Guolin Ke committed
18
  TARGET_LINK_LIBRARIES(lightgbm ${MPI_CXX_LIBRARIES})
Guolin Ke's avatar
Guolin Ke committed
19
20
endif(USE_MPI)