omp schedule

mathias-von-ottenbreit · mathias-von-ottenbreit · commit 57f018f9f261 · 2024-02-17T14:01:04.000+01:00
diff --git a/cpp/APLRRegressor.h b/cpp/APLRRegressor.h
@@ -1133,7 +1133,7 @@ std::vector<size_t> APLRRegressor::find_terms_eligible_current_indexes_for_a_bas
 void APLRRegressor::estimate_split_point_for_each_term(std::vector<Term> &terms, std::vector<size_t> &terms_indexes)
 {
     bool multithreading{n_jobs != 1 && terms_indexes.size() > 1};
-#pragma omp parallel for schedule(auto) if (multithreading)
+#pragma omp parallel for schedule(guided) if (multithreading)
     for (size_t i = 0; i < terms_indexes.size(); ++i)
     {
         terms[terms_indexes[i]].estimate_split_point(X_train, neg_gradient_current, sample_weight_train, bins, v, min_observations_in_split);

Original file line number	Diff line number	Diff line change
`@@ -1133,7 +1133,7 @@ std::vector<size_t> APLRRegressor::find_terms_eligible_current_indexes_for_a_bas`
`1133`	`1133`	`void APLRRegressor::estimate_split_point_for_each_term(std::vector<Term> &terms, std::vector<size_t> &terms_indexes)`
`1134`	`1134`	`{`
`1135`	`1135`	`bool multithreading{n_jobs != 1 && terms_indexes.size() > 1};`
`1136`		`-#pragma omp parallel for schedule(auto) if (multithreading)`
	`1136`	`+#pragma omp parallel for schedule(guided) if (multithreading)`
`1137`	`1137`	`for (size_t i = 0; i < terms_indexes.size(); ++i)`
`1138`	`1138`	`{`
`1139`	`1139`	`terms[terms_indexes[i]].estimate_split_point(X_train, neg_gradient_current, sample_weight_train, bins, v, min_observations_in_split);`