diff --git a/Jenkinsfile b/Jenkinsfile index 215e9102e845..04a918cfc171 100644 --- a/Jenkinsfile +++ b/Jenkinsfile @@ -14,7 +14,7 @@ def dockerRun = 'tests/ci_build/ci_build.sh' def utils def buildMatrix = [ - [ "enabled": true, "os" : "linux", "withGpu": true, "withNccl": true, "withOmp": true, "pythonVersion": "2.7", "cudaVersion": "9.2", "multiGpu": true], + [ "enabled": true, "os" : "linux", "withGpu": true, "withNccl": true, "withOmp": true, "pythonVersion": "2.7", "cudaVersion": "10.0", "multiGpu": true], [ "enabled": true, "os" : "linux", "withGpu": true, "withNccl": true, "withOmp": true, "pythonVersion": "2.7", "cudaVersion": "9.2" ], [ "enabled": true, "os" : "linux", "withGpu": true, "withNccl": true, "withOmp": true, "pythonVersion": "2.7", "cudaVersion": "8.0" ], [ "enabled": true, "os" : "linux", "withGpu": true, "withNccl": false, "withOmp": true, "pythonVersion": "2.7", "cudaVersion": "8.0" ], diff --git a/Jenkinsfile-restricted b/Jenkinsfile-restricted index 99de2c09f544..f55997640d9d 100644 --- a/Jenkinsfile-restricted +++ b/Jenkinsfile-restricted @@ -19,6 +19,7 @@ def commit_id def branch_name def buildMatrix = [ + [ "enabled": true, "os" : "linux", "withGpu": true, "withNccl": true, "withOmp": true, "pythonVersion": "2.7", "cudaVersion": "10.0" ], [ "enabled": true, "os" : "linux", "withGpu": true, "withNccl": true, "withOmp": true, "pythonVersion": "2.7", "cudaVersion": "9.2" ], [ "enabled": true, "os" : "linux", "withGpu": true, "withNccl": true, "withOmp": true, "pythonVersion": "2.7", "cudaVersion": "8.0" ], [ "enabled": true, "os" : "linux", "withGpu": true, "withNccl": false, "withOmp": true, "pythonVersion": "2.7", "cudaVersion": "8.0" ], diff --git a/src/common/host_device_vector.cu b/src/common/host_device_vector.cu index ef3c8eee8f94..55450f3bd819 100644 --- a/src/common/host_device_vector.cu +++ b/src/common/host_device_vector.cu @@ -53,8 +53,7 @@ struct HostDeviceVectorImpl { if (vec_ == nullptr) { vec_ = vec; } CHECK_EQ(vec, vec_); device_ = device; - // TODO(rongou): remove pointer dereference once CUDA 10.1 is fixed. - LazyResize((*vec_).Size()); + LazyResize(vec_->Size()); perm_d_ = vec_->perm_h_.Complementary(); } diff --git a/tests/ci_build/Dockerfile.gpu b/tests/ci_build/Dockerfile.gpu index 15a72c09a598..fa3b914dd205 100644 --- a/tests/ci_build/Dockerfile.gpu +++ b/tests/ci_build/Dockerfile.gpu @@ -26,7 +26,12 @@ RUN \ cp nccl_2.2.13-1+cuda${CUDA_SHORT}_x86_64/include/nccl.h /usr/include && \ cp nccl_2.2.13-1+cuda${CUDA_SHORT}_x86_64/lib/* /usr/lib && \ rm -f nccl_2.2.13-1+cuda${CUDA_SHORT}_x86_64.txz && \ - rm -r nccl_2.2.13-1+cuda${CUDA_SHORT}_x86_64; fi + rm -r nccl_2.2.13-1+cuda${CUDA_SHORT}_x86_64; else \ + wget https://developer.download.nvidia.com/compute/machine-learning/repos/rhel7/x86_64/nvidia-machine-learning-repo-rhel7-1.0.0-1.x86_64.rpm && \ + rpm -i nvidia-machine-learning-repo-rhel7-1.0.0-1.x86_64.rpm && \ + yum -y update && \ + yum install -y libnccl-2.4.2-1+cuda10.0 libnccl-devel-2.4.2-1+cuda10.0 libnccl-static-2.4.2-1+cuda10.0 && \ + rm -f nvidia-machine-learning-repo-rhel7-1.0.0-1.x86_64.rpm; fi ENV PATH=/opt/python/bin:$PATH ENV CC=/opt/rh/devtoolset-2/root/usr/bin/gcc