Add google test for a column sampling, restore metainfo tests (#3637)

* Add google test for a column sampling, restore metainfo tests

* Update metainfo test for visual studio

* Fix multi-GPU bug introduced in #3635
This commit is contained in:
Rory Mitchell
2018-08-28 16:10:26 +12:00
committed by GitHub
parent 7ef2b599c7
commit 78bea0d204
3 changed files with 102 additions and 3 deletions

View File

@@ -77,9 +77,9 @@ struct HostDeviceVectorImpl {
void LazySyncHost() {
dh::safe_cuda(cudaSetDevice(device_));
dh::safe_cuda(
cudaMemcpy(vec_->data_h_.data(), data_.data().get() + start_,
data_.size() * sizeof(T), cudaMemcpyDeviceToHost));
dh::safe_cuda(cudaMemcpy(vec_->data_h_.data() + start_,
data_.data().get(), data_.size() * sizeof(T),
cudaMemcpyDeviceToHost));
on_d_ = false;
}