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

restore learner

Browse Source
This commit is contained in:
Hui Liu 2023-10-27 11:15:06 -07:00
parent 6762230d9a
commit 6bbca9a8b7
1 changed files with 0 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/learner.cc
View File

@ -846,20 +846,9 @@ class LearnerConfiguration : public Learner {
} }
void InitEstimation(MetaInfo const& info, linalg::Tensor<float, 1>* base_score) { void InitEstimation(MetaInfo const& info, linalg::Tensor<float, 1>* base_score) {
#ifndef XGBOOST_USE_HIP
base_score->Reshape(1); base_score->Reshape(1);
collective::ApplyWithLabels(info, base_score->Data(), collective::ApplyWithLabels(info, base_score->Data(),
[&] { UsePtr(obj_)->InitEstimation(info, base_score); }); [&] { UsePtr(obj_)->InitEstimation(info, base_score); });
#else
if (info.IsVerticalFederated()) {
base_score->Reshape(1);
collective::ApplyWithLabels(info, base_score->Data()->HostPointer(),
sizeof(bst_float) * base_score->Size(),
[&] { UsePtr(obj_)->InitEstimation(info, base_score); });
} else {
UsePtr(obj_)->InitEstimation(info, base_score);
}
#endif
} }
}; };
@ -1478,20 +1467,9 @@ class LearnerImpl : public LearnerIO {
private: private:
void GetGradient(HostDeviceVector<bst_float> const& preds, MetaInfo const& info, void GetGradient(HostDeviceVector<bst_float> const& preds, MetaInfo const& info,
std::int32_t iter, linalg::Matrix<GradientPair>* out_gpair) { std::int32_t iter, linalg::Matrix<GradientPair>* out_gpair) {
#ifndef XGBOOST_USE_HIP
out_gpair->Reshape(info.num_row_, this->learner_model_param_.OutputLength()); out_gpair->Reshape(info.num_row_, this->learner_model_param_.OutputLength());
collective::ApplyWithLabels(info, out_gpair->Data(), collective::ApplyWithLabels(info, out_gpair->Data(),
[&] { obj_->GetGradient(preds, info, iter, out_gpair); }); [&] { obj_->GetGradient(preds, info, iter, out_gpair); });
#else
if (info.IsVerticalFederated()) {
out_gpair->Reshape(info.num_row_, this->learner_model_param_.OutputLength());
collective::ApplyWithLabels(info, out_gpair->Data(),
[&] { obj_->GetGradient(preds, info, iter, out_gpair); });
}
else {
obj_->GetGradient(preds, info, iter, out_gpair);
}
#endif
} }
/*! \brief random number transformation seed. */ /*! \brief random number transformation seed. */