Skip to content

Commit f78c4cd

Browse files
committed
Cleanup.
1 parent c90ec77 commit f78c4cd

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

src/tree/gpu_hist/multi_evaluate_splits.cuh

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -52,8 +52,7 @@ class MultiHistEvaluator {
5252
}
5353
[[nodiscard]] common::Span<GradientPairInt64 const> GetNodeSum(bst_node_t nidx,
5454
bst_target_t n_targets) const {
55-
auto offset = nidx * n_targets;
56-
return dh::ToSpan(this->node_sums_).subspan(offset, n_targets);
55+
return GetNodeSumImpl(dh::ToSpan(this->node_sums_), nidx, n_targets);
5756
}
5857

5958
// Track the child gradient sum.

0 commit comments

Comments
 (0)