366f3cb9d8
* Ensure RMM is 0.18 or later * Add use_rmm flag to global configuration * Modify XGBCachingDeviceAllocatorImpl to skip CUB when use_rmm=True * Update the demo * [CI] Pin NumPy to 1.19.4, since NumPy 1.19.5 doesn't work with latest Shap
35 lines
954 B
C++
35 lines
954 B
C++
/*!
|
|
* Copyright 2020 by Contributors
|
|
* \file global_config.h
|
|
* \brief Global configuration for XGBoost
|
|
* \author Hyunsu Cho
|
|
*/
|
|
#ifndef XGBOOST_GLOBAL_CONFIG_H_
|
|
#define XGBOOST_GLOBAL_CONFIG_H_
|
|
|
|
#include <xgboost/parameter.h>
|
|
#include <vector>
|
|
#include <string>
|
|
|
|
namespace xgboost {
|
|
class Json;
|
|
|
|
struct GlobalConfiguration : public XGBoostParameter<GlobalConfiguration> {
|
|
int verbosity { 1 };
|
|
bool use_rmm { false };
|
|
DMLC_DECLARE_PARAMETER(GlobalConfiguration) {
|
|
DMLC_DECLARE_FIELD(verbosity)
|
|
.set_range(0, 3)
|
|
.set_default(1) // shows only warning
|
|
.describe("Flag to print out detailed breakdown of runtime.");
|
|
DMLC_DECLARE_FIELD(use_rmm)
|
|
.set_default(false)
|
|
.describe("Whether to use RAPIDS Memory Manager to allocate GPU memory in XGBoost");
|
|
}
|
|
};
|
|
|
|
using GlobalConfigThreadLocalStore = dmlc::ThreadLocalStore<GlobalConfiguration>;
|
|
} // namespace xgboost
|
|
|
|
#endif // XGBOOST_GLOBAL_CONFIG_H_
|