Update nccl (#2494)
This commit is contained in:
parent
e7530bdffc
commit
6bfc472bec
@ -62,7 +62,6 @@ set(LINK_LIBRARIES dmlccore rabit)
|
||||
if(PLUGIN_UPDATER_GPU)
|
||||
find_package(CUDA 7.5 REQUIRED)
|
||||
cmake_minimum_required(VERSION 3.5)
|
||||
add_subdirectory(nccl)
|
||||
|
||||
include_directories(
|
||||
nccl/src
|
||||
@ -76,6 +75,7 @@ if(PLUGIN_UPDATER_GPU)
|
||||
set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS};-Xcompiler -fPIC; -std=c++11")
|
||||
endif()
|
||||
|
||||
add_subdirectory(nccl)
|
||||
cuda_add_library(gpuxgboost ${CUDA_SOURCES} STATIC)
|
||||
target_link_libraries(gpuxgboost nccl)
|
||||
list(APPEND LINK_LIBRARIES gpuxgboost)
|
||||
@ -128,7 +128,6 @@ if(GOOGLE_TEST)
|
||||
|
||||
if(PLUGIN_UPDATER_GPU)
|
||||
file(GLOB_RECURSE CUDA_TEST_SOURCES "plugin/updater_gpu/test/cpp/*.cu")
|
||||
set(CUDA_VERBOSE_BUILD ON)
|
||||
cuda_compile(CUDA_TEST_OBJS ${CUDA_TEST_SOURCES})
|
||||
else()
|
||||
set(CUDA_TEST_OBJS "")
|
||||
|
||||
2
nccl
2
nccl
@ -1 +1 @@
|
||||
Subproject commit 8ec6c27a33a900fb92f7e39acc73cc0f43e8539b
|
||||
Subproject commit 018ff75f78e4d3616051fdcd3da7cf64249b6d59
|
||||
Loading…
x
Reference in New Issue
Block a user