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

fix offset in score_updater.

parent b01d00aa
...@@ -52,17 +52,17 @@ public: ...@@ -52,17 +52,17 @@ public:
inline bool has_init_score() const { return has_init_score_; } inline bool has_init_score() const { return has_init_score_; }
inline void AddScore(double val, int cur_tree_id) { inline void AddScore(double val, int cur_tree_id) {
int64_t offset = cur_tree_id * num_data_; const size_t offset = static_cast<size_t>(num_data_) * cur_tree_id;
#pragma omp parallel for schedule(static) #pragma omp parallel for schedule(static)
for (int64_t i = 0; i < num_data_; ++i) { for (int i = 0; i < num_data_; ++i) {
score_[offset + i] += val; score_[offset + i] += val;
} }
} }
inline void MultiplyScore(double val, int cur_tree_id) { inline void MultiplyScore(double val, int cur_tree_id) {
int64_t offset = cur_tree_id * num_data_; const size_t offset = static_cast<size_t>(num_data_) * cur_tree_id;
#pragma omp parallel for schedule(static) #pragma omp parallel for schedule(static)
for (int64_t i = 0; i < num_data_; ++i) { for (int i = 0; i < num_data_; ++i) {
score_[offset + i] *= val; score_[offset + i] *= val;
} }
} }
......
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