Implement Empty method for host device vector. (#5781)

* Fix accessing nullptr.
This commit is contained in:
Jiaming Yuan
2020-06-13 19:02:26 +08:00
committed by GitHub
parent 306e38ff31
commit 1fa84b61c1
3 changed files with 11 additions and 1 deletions

View File

@@ -59,7 +59,9 @@ class HostDeviceVectorImpl {
}
}
size_t Size() const { return HostCanRead() ? data_h_.size() : data_d_->size(); }
size_t Size() const {
return HostCanRead() ? data_h_.size() : data_d_ ? data_d_->size() : 0;
}
int DeviceIdx() const { return device_; }