* Add bindings for serialization. * Change `xgb.save.raw' into full serialization instead of simple model. * Add `xgb.load.raw' for unserialization. * Run devtools.
39 lines
1.2 KiB
CMake
39 lines
1.2 KiB
CMake
find_package(LibR REQUIRED)
|
|
message(STATUS "LIBR_CORE_LIBRARY " ${LIBR_CORE_LIBRARY})
|
|
|
|
file(GLOB_RECURSE R_SOURCES
|
|
${CMAKE_CURRENT_LIST_DIR}/src/*.cc
|
|
${CMAKE_CURRENT_LIST_DIR}/src/*.c)
|
|
# Use object library to expose symbols
|
|
add_library(xgboost-r OBJECT ${R_SOURCES})
|
|
|
|
set(R_DEFINITIONS
|
|
-DXGBOOST_STRICT_R_MODE=1
|
|
-DXGBOOST_CUSTOMIZE_GLOBAL_PRNG=1
|
|
-DDMLC_LOG_BEFORE_THROW=0
|
|
-DDMLC_DISABLE_STDIN=1
|
|
-DDMLC_LOG_CUSTOMIZE=1
|
|
-DRABIT_CUSTOMIZE_MSG_
|
|
-DRABIT_STRICT_CXX98_)
|
|
target_compile_definitions(xgboost-r
|
|
PRIVATE ${R_DEFINITIONS})
|
|
target_include_directories(xgboost-r
|
|
PRIVATE
|
|
${LIBR_INCLUDE_DIRS}
|
|
${PROJECT_SOURCE_DIR}/include
|
|
${PROJECT_SOURCE_DIR}/dmlc-core/include
|
|
${PROJECT_SOURCE_DIR}/rabit/include)
|
|
set_target_properties(
|
|
xgboost-r PROPERTIES
|
|
CXX_STANDARD 11
|
|
CXX_STANDARD_REQUIRED ON
|
|
POSITION_INDEPENDENT_CODE ON)
|
|
|
|
set(XGBOOST_DEFINITIONS "${XGBOOST_DEFINITIONS};${R_DEFINITIONS}" PARENT_SCOPE)
|
|
set(XGBOOST_OBJ_SOURCES $<TARGET_OBJECTS:xgboost-r> PARENT_SCOPE)
|
|
set(LINKED_LIBRARIES_PRIVATE ${LINKED_LIBRARIES_PRIVATE} ${LIBR_CORE_LIBRARY} PARENT_SCOPE)
|
|
|
|
if (USE_OPENMP)
|
|
target_link_libraries(xgboost-r PRIVATE OpenMP::OpenMP_CXX)
|
|
endif ()
|