diff --git a/CMakeLists.txt b/CMakeLists.txt index 3b8bbd2e0272..20ade5db13d8 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -732,7 +732,12 @@ install(TARGETS ${MXNET_INSTALL_TARGETS} ARCHIVE DESTINATION ${CMAKE_INSTALL_LIBDIR} ) +# NOTE: Public headers will be installed into ${CMAKE_INSTALL_PREFIX}/${CMAKE_INSTALL_INCLUDEDIR}, see +# https://cmake.org/cmake/help/v3.0/variable/CMAKE_INSTALL_PREFIX.html +# https://cmake.org/cmake/help/v3.0/module/GNUInstallDirs.html + install(DIRECTORY include/ DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}) +install(DIRECTORY 3rdparty/tvm/nnvm/include/ DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}) if (INSTALL_EXAMPLES) install(DIRECTORY example DESTINATION ${CMAKE_INSTALL_DATADIR}/${PROJECT_NAME}) endif() diff --git a/cpp-package/CMakeLists.txt b/cpp-package/CMakeLists.txt index f7fbc77e1a5e..5d2977279d74 100644 --- a/cpp-package/CMakeLists.txt +++ b/cpp-package/CMakeLists.txt @@ -20,4 +20,6 @@ if(USE_CPP_PACKAGE) add_subdirectory(example) endif() + install(DIRECTORY include/ DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}) + endif()