global_config.h

前往此文件文档。

 
#ifndef XGBOOST_GLOBAL_CONFIG_H_
#define XGBOOST_GLOBAL_CONFIG_H_
 
#include <dmlc/thread_local.h> // 用于 ThreadLocalStore
#include <xgboost/parameter.h> // 用于 XGBoostParameter
 
#include <cstdint> // 用于 int32_t
 
namespace xgboost {
struct GlobalConfiguration : public XGBoostParameter<GlobalConfiguration> {
 std::int32_t verbosity{1};
 bool use_rmm{false};
 // 这不是一个dmlc参数，以避免与上下文类冲突。
 std::int32_t nthread{0};
 DMLC_DECLARE_PARAMETER(GlobalConfiguration) {
 DMLC_DECLARE_FIELD(verbosity)
 .set_range(0, 3)
 .set_default(1) // 仅显示警告
 .describe("用于打印详细运行时分解的标志。");
 DMLC_DECLARE_FIELD(use_rmm).set_default(false).describe(
 "是否使用RAPIDS内存管理器在XGBoost中分配GPU内存");
  }
};
 
using GlobalConfigThreadLocalStore = dmlc::ThreadLocalStore<GlobalConfiguration>;
 
struct InitNewThread {
 GlobalConfiguration config;
 std::int32_t device{-1};
 
 void operator()() const;
 InitNewThread();
};
} // namespace xgboost
 
#endif // XGBOOST_GLOBAL_CONFIG_H_