CMakeLists.txt 1.11 KB
Newer Older
Guolin Ke's avatar
Guolin Ke committed
1
2
include_directories(${LightGBM_HEADER_DIR})

3
4
5
6
if(APPLE)
  set(CMAKE_SHARED_LIBRARY_SUFFIX ".so")
endif(APPLE)

Guolin Ke's avatar
Guolin Ke committed
7
8
9
10
11
12
13
14
15
16
17
18
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
19
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
20

Guolin Ke's avatar
Guolin Ke committed
21
22
add_library(_lightgbm SHARED c_api.cpp ${APPLICATION_SRC} ${BOOSTING_SRC} ${IO_SRC} ${METRIC_SRC} ${OBJECTIVE_SRC} ${NETWORK_SRC} ${TREELEARNER_SRC})

Guolin Ke's avatar
Guolin Ke committed
23
if(USE_MPI)
Guolin Ke's avatar
Guolin Ke committed
24
  TARGET_LINK_LIBRARIES(lightgbm ${MPI_CXX_LIBRARIES})
Guolin Ke's avatar
Guolin Ke committed
25
  TARGET_LINK_LIBRARIES(_lightgbm ${MPI_CXX_LIBRARIES})
Guolin Ke's avatar
Guolin Ke committed
26
27
endif(USE_MPI)

28
29
30
31
32
33
34
if(WIN32)
  TARGET_LINK_LIBRARIES(lightgbm Ws2_32)
  TARGET_LINK_LIBRARIES(_lightgbm Ws2_32)
  TARGET_LINK_LIBRARIES(lightgbm IPHLPAPI)
  TARGET_LINK_LIBRARIES(_lightgbm IPHLPAPI)
endif(WIN32)