Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
SHVETS, KIRILL committed Apr 30, 2020
1 parent 6793b03 commit ddb786b
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions include/xgboost/base.h
Original file line number Diff line number Diff line change
Expand Up @@ -141,22 +141,22 @@ class GradientPairInternal {
public:
using ValueT = T;

inline void Add(const GradientPairInternal& b) {
XGBOOST_DEVICE void Add(const GradientPairInternal& b) {
grad_ += b.grad_;
hess_ += b.hess_;
}

inline void Add(const ValueT& grad, const ValueT& hess) {
XGBOOST_DEVICE void Add(const ValueT& grad, const ValueT& hess) {
grad_ += grad;
hess_ += hess;
}

inline void SetSubstract(const GradientPairInternal& a, const GradientPairInternal& b) {
XGBOOST_DEVICE void SetSubstract(const GradientPairInternal& a, const GradientPairInternal& b) {
grad_ = a.grad_ - b.grad_;
hess_ = a.hess_ - b.hess_;
}

inline static void Reduce(GradientPairInternal& a, const GradientPairInternal& b) { // NOLINT(*)
XGBOOST_DEVICE static void Reduce(GradientPairInternal& a, const GradientPairInternal& b) { // NOLINT
a.Add(b);
}

Expand Down

0 comments on commit ddb786b

Please sign in to comment.