support cuda 10.1 (#4223)

* support cuda 10.1

* add cuda 10.1 to jenkins build matrix
This commit is contained in:
Rong Ou
2019-03-08 12:22:12 +13:00
committed by Rory Mitchell
parent 0944360416
commit 9837b09b20
4 changed files with 10 additions and 3 deletions
+2 -1
View File
@@ -53,7 +53,8 @@ struct HostDeviceVectorImpl {
if (vec_ == nullptr) { vec_ = vec; }
CHECK_EQ(vec, vec_);
device_ = device;
LazyResize(vec_->Size());
// TODO(rongou): remove pointer dereference once CUDA 10.1 is fixed.
LazyResize((*vec_).Size());
perm_d_ = vec_->perm_h_.Complementary();
}