08ce495b5d
- Pass context from booster to DMatrix. - Use context instead of integer for `n_threads`. - Check the consistency configuration for `max_bin`. - Test for all combinations of initialization options.
34 lines
993 B
C++
34 lines
993 B
C++
/**
|
|
* Copyright 2023, XGBoost Contributors
|
|
*/
|
|
#ifndef XGBOOST_DATA_BATCH_UTILS_H_
|
|
#define XGBOOST_DATA_BATCH_UTILS_H_
|
|
|
|
#include "xgboost/data.h" // for BatchParam
|
|
|
|
namespace xgboost::data::detail {
|
|
// At least one batch parameter is initialized.
|
|
inline void CheckEmpty(BatchParam const& l, BatchParam const& r) {
|
|
if (!l.Initialized()) {
|
|
CHECK(r.Initialized()) << "Batch parameter is not initialized.";
|
|
}
|
|
}
|
|
|
|
/**
|
|
* \brief Should we regenerate the gradient index?
|
|
*
|
|
* \param old Parameter stored in DMatrix.
|
|
* \param p New parameter passed in by caller.
|
|
*/
|
|
inline bool RegenGHist(BatchParam old, BatchParam p) {
|
|
// Parameter is renewed or caller requests a regen
|
|
if (!p.Initialized()) {
|
|
// Empty parameter is passed in, don't regenerate so that we can use gindex in
|
|
// predictor, which doesn't have any training parameter.
|
|
return false;
|
|
}
|
|
return p.regen || old.ParamNotEqual(p);
|
|
}
|
|
} // namespace xgboost::data::detail
|
|
#endif // XGBOOST_DATA_BATCH_UTILS_H_
|