Skip to content

Commit 6fb4e86

Browse files
committed
Resolve.
1 parent ff052f2 commit 6fb4e86

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/tree/param.h

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -272,7 +272,7 @@ XGBOOST_DEVICE inline T CalcGainGivenWeight(const TrainingParams &p,
272272

273273
// calculate the cost of loss function
274274
template <typename TrainingParams, typename T>
275-
XGBOOST_DEVICE inline T CalcGain(const TrainingParams &p, T sum_grad, T sum_hess) {
275+
XGBOOST_DEVICE inline T CalcGain(const TrainingParams &p, T const& sum_grad, T const& sum_hess) {
276276
if (sum_hess < p.min_child_weight) {
277277
return T(0.0);
278278
}
@@ -295,8 +295,8 @@ XGBOOST_DEVICE inline T CalcGain(const TrainingParams &p, T sum_grad, T sum_hess
295295
}
296296

297297
template <typename TrainingParams,
298-
typename StatT, typename T = decltype(StatT().GetHess())>
299-
XGBOOST_DEVICE inline T CalcGain(const TrainingParams &p, StatT stat) {
298+
typename StatT, typename T = typename std::remove_reference< decltype(StatT().GetHess()) >::type>
299+
XGBOOST_DEVICE inline T CalcGain(const TrainingParams &p, StatT const& stat) {
300300
return CalcGain(p, stat.GetGrad(), stat.GetHess());
301301
}
302302

0 commit comments

Comments
 (0)