This commit is contained in:
Jiaming Yuan
2021-11-15 01:28:11 +08:00
committed by GitHub
parent a7057fa64c
commit d4274bc556
6 changed files with 17 additions and 18 deletions

View File

@@ -233,7 +233,7 @@ struct GPUHistMakerDevice {
// thread safe
void Reset(HostDeviceVector<GradientPair>* dh_gpair, DMatrix* dmat, int64_t num_columns) {
auto const& info = dmat->Info();
this->column_sampler.Init(num_columns, info.feature_weigths.HostVector(),
this->column_sampler.Init(num_columns, info.feature_weights.HostVector(),
param.colsample_bynode, param.colsample_bylevel,
param.colsample_bytree);
dh::safe_cuda(cudaSetDevice(device_id));