global_config.h

Go to the documentation of this file.

 
#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_