Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

jenkins build for cuda 10.0 (#4281)

Browse Source 
* jenkins build for cuda 10.0

* yum install nccl2 for cuda 10.0
This commit is contained in:
Rong Ou
2019-03-22 22:35:18 -07:00
committed by Philip Hyunsu Cho
parent 263e2038e9
commit 5aa42b5f11
4 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/common/host_device_vector.cu
View File

@@ -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();
}