Commit 90554d08 by 健美猫 Committed by Tianqi Chen

Install nnvm lib and haders. Offer choice to build static or shared lib. (#146)

parent f5bb0e02
...@@ -17,6 +17,10 @@ set(nnvm_LINKER_LIBS "") ...@@ -17,6 +17,10 @@ set(nnvm_LINKER_LIBS "")
add_definitions(-DNNVM_EXPORTS) add_definitions(-DNNVM_EXPORTS)
# Build a shared lib (libnnvm.so) by default
option(BUILD_SHARED_NNVM "Build a shared nnvm lib" ON)
option(BUILD_STATIC_NNVM "Build a static nnvm lib" OFF)
# compile # compile
if(MSVC) if(MSVC)
add_definitions(-DDMLC_USE_CXX11) add_definitions(-DDMLC_USE_CXX11)
...@@ -64,8 +68,15 @@ elseif(DMLC_CORE_PATH) ...@@ -64,8 +68,15 @@ elseif(DMLC_CORE_PATH)
include_directories(${DMLC_CORE_PATH}/include) include_directories(${DMLC_CORE_PATH}/include)
endif() endif()
add_library(nnvm SHARED ${SOURCE}) if(BUILD_SHARED_NNVM)
target_link_libraries(nnvm ${nnvm_LINKER_LIBS}) add_library(nnvm SHARED ${SOURCE})
target_link_libraries(nnvm ${nnvm_LINKER_LIBS})
endif()
if(BUILD_STATIC_NNVM)
add_library(nnvm_static STATIC ${SOURCE})
target_link_libraries(nnvm_static ${nnvm_LINKER_LIBS})
set_target_properties(nnvm_static PROPERTIES OUTPUT_NAME "nnvm")
endif()
if(INSTALL_INCLUDE_DIR) if(INSTALL_INCLUDE_DIR)
add_custom_command(TARGET nnvm POST_BUILD add_custom_command(TARGET nnvm POST_BUILD
...@@ -74,6 +85,16 @@ if(INSTALL_INCLUDE_DIR) ...@@ -74,6 +85,16 @@ if(INSTALL_INCLUDE_DIR)
) )
endif() endif()
# ---[ Install lib, header and docs
if(BUILD_SHARED_NNVM)
install(TARGETS nnvm LIBRARY DESTINATION lib)
endif()
if(BUILD_STATIC_NNVM)
install(TARGETS nnvm_static ARCHIVE DESTINATION lib)
endif()
install(DIRECTORY include DESTINATION .)
install(DIRECTORY docs DESTINATION .)
# ---[ Linter target # ---[ Linter target
if(MSVC) if(MSVC)
find_package(PythonInterp 2) find_package(PythonInterp 2)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment