Remove omp_get_max_threads in gbm and linear. (#7537)

* Use ctx in gbm. * Use ctx threads in gbm and linear.
2022-01-05 03:28:52 +08:00
parent eea094e1bc
commit 28af6f9abb
12 changed files with 124 additions and 135 deletions
--- a/include/xgboost/gbm.h
+++ b/include/xgboost/gbm.h
@@ -38,7 +38,7 @@ class PredictionContainer;
 */
 class GradientBooster : public Model, public Configurable {
 protected:
-  GenericParameter const* generic_param_;
+  GenericParameter const* ctx_;

 public:
  /*! \brief virtual destructor */
--- a/include/xgboost/linear_updater.h
+++ b/include/xgboost/linear_updater.h
@@ -29,7 +29,7 @@ class GBLinearModel;
 */
 class LinearUpdater : public Configurable {
 protected:
-  GenericParameter const* learner_param_;
+  GenericParameter const* ctx_;

 public:
  /*! \brief virtual destructor */