From e033caa3ba62f87cbc12f16af19c2777e73c5679 Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Wed, 23 Sep 2020 07:59:33 +0800 Subject: [PATCH] Remove linking RMM library. (#6146) * Remove linking RMM library. * RMM is now header only. * Remove remaining reference. --- plugin/CMakeLists.txt | 12 +++--------- 1 file changed, 3 insertions(+), 9 deletions(-) diff --git a/plugin/CMakeLists.txt b/plugin/CMakeLists.txt index dc54c2509ad5..5e1b412885d8 100644 --- a/plugin/CMakeLists.txt +++ b/plugin/CMakeLists.txt @@ -13,13 +13,8 @@ if (PLUGIN_DENSE_PARSER) endif (PLUGIN_DENSE_PARSER) if (PLUGIN_RMM) - find_path(RMM_INCLUDE "rmm" - HINTS "$ENV{RMM_ROOT}/include") - - find_library(RMM_LIBRARY "rmm" - HINTS "$ENV{RMM_ROOT}/lib" "$ENV{RMM_ROOT}/build") - - if ((NOT RMM_LIBRARY) OR (NOT RMM_INCLUDE)) + find_path(RMM_INCLUDE "rmm" HINTS "$ENV{RMM_ROOT}/include") + if (NOT RMM_INCLUDE) message(FATAL_ERROR "Could not locate RMM library") endif () @@ -27,13 +22,12 @@ if (PLUGIN_RMM) message(STATUS "RMM: RMM_INCLUDE set to ${RMM_INCLUDE}") target_include_directories(objxgboost PUBLIC ${RMM_INCLUDE}) - target_link_libraries(objxgboost PUBLIC ${RMM_LIBRARY} cuda) target_compile_definitions(objxgboost PUBLIC -DXGBOOST_USE_RMM=1) endif (PLUGIN_RMM) if (PLUGIN_UPDATER_ONEAPI) add_library(oneapi_plugin OBJECT - ${xgboost_SOURCE_DIR}/plugin/updater_oneapi/regression_obj_oneapi.cc + ${xgboost_SOURCE_DIR}/plugin/updater_oneapi/regression_obj_oneapi.cc ${xgboost_SOURCE_DIR}/plugin/updater_oneapi/predictor_oneapi.cc) target_include_directories(oneapi_plugin PRIVATE