Merge pull request #45043 from mmusich/mm_fix_44923_in_14_0_X

[14.0.X] `PhotonXGBoostEstimator` : set `XGBoost` to use one thread to avoid spawning hundreds of OpenMP threads
cms-sw · May 27, 2024 · f40fd9a · f40fd9a
1 parent 77c6c38
commit f40fd9a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/RecoEgamma/PhotonIdentification/src/PhotonXGBoostEstimator.cc b/RecoEgamma/PhotonIdentification/src/PhotonXGBoostEstimator.cc
@@ -3,6 +3,9 @@
 
 PhotonXGBoostEstimator::PhotonXGBoostEstimator(const edm::FileInPath& weightsFile, int best_ntree_limit) {
   XGBoosterCreate(NULL, 0, &booster_);
+  // Set number of threads to 1, to avoid spawning hundreds of OpenMP threads
+  // See https://github.com/cms-sw/cmssw/issues/44923 for details
+  XGBoosterSetParam(booster_, "nthread", "1");
   XGBoosterLoadModel(booster_, weightsFile.fullPath().c_str());
   best_ntree_limit_ = best_ntree_limit;