Unverified Commit 9b263735 authored by Guolin Ke's avatar Guolin Ke Committed by GitHub
Browse files

fix the omp index in window

parent 1dbe5e99
...@@ -13,6 +13,7 @@ ...@@ -13,6 +13,7 @@
#include <algorithm> #include <algorithm>
#include <chrono> #include <chrono>
#include <cstdio> #include <cstdio>
#include <cstdint>
#include <fstream> #include <fstream>
#include <vector> #include <vector>
...@@ -59,9 +60,9 @@ class GOSS: public GBDT { ...@@ -59,9 +60,9 @@ class GOSS: public GBDT {
if (gradients != nullptr) { if (gradients != nullptr) {
// use customized objective function // use customized objective function
CHECK(hessians != nullptr && objective_function_ == nullptr); CHECK(hessians != nullptr && objective_function_ == nullptr);
size_t total_size = static_cast<size_t>(num_data_) * num_tree_per_iteration_; int64_t total_size = static_cast<int64_t>(num_data_) * num_tree_per_iteration_;
#pragma omp parallel for schedule(static) #pragma omp parallel for schedule(static)
for (size_t i = 0; i < total_size; ++i) { for (int64_t i = 0; i < total_size; ++i) {
gradients_[i] = gradients[i]; gradients_[i] = gradients[i];
hessians_[i] = hessians[i]; hessians_[i] = hessians[i];
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment