Skip to content

Commit c8de58d

Browse files
authored
Small refactor for the GPU predictor. (#11735)
1 parent e0622eb commit c8de58d

File tree

2 files changed

+270
-270
lines changed

2 files changed

+270
-270
lines changed

src/data/ellpack_page.cuh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -159,6 +159,7 @@ struct EllpackAccessorImpl {
159159
return this->null_value_ & ((Ind() << NullShift()) - Ind());
160160
}
161161
[[nodiscard]] XGBOOST_HOST_DEV_INLINE bst_idx_t NumBins() const { return gidx_fvalue_map.size(); }
162+
[[nodiscard]] XGBOOST_HOST_DEV_INLINE bst_idx_t NumRows() const { return n_rows; }
162163
[[nodiscard]] XGBOOST_HOST_DEV_INLINE size_t NumFeatures() const { return min_fvalue.size(); }
163164
};
164165

0 commit comments

Comments
 (0)