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:
inline bool has_init_score() const { return has_init_score_; }
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)
for (int64_t i = 0; i < num_data_; ++i) {
for (int i = 0; i < num_data_; ++i) {
score_[offset + i] += val;
}
}
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)
for (int64_t i = 0; i < num_data_; ++i) {
for (int i = 0; i < num_data_; ++i) {
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