Fix GPU categorical split memory allocation. (#9529)

2023-08-29 10:06:03 +08:00
parent be6a552956
commit 942b957eef
5 changed files with 67 additions and 64 deletions
--- a/src/common/categorical.h
+++ b/src/common/categorical.h
@@ -52,7 +52,7 @@ inline XGBOOST_DEVICE bool InvalidCat(float cat) {
 *
 *   Go to left if it's NOT the matching category, which matches one-hot encoding.
 */
-inline XGBOOST_DEVICE bool Decision(common::Span<uint32_t const> cats, float cat) {
+inline XGBOOST_DEVICE bool Decision(common::Span<CatBitField::value_type const> cats, float cat) {
  KCatBitField const s_cats(cats);
  if (XGBOOST_EXPECT(InvalidCat(cat), false)) {
    return true;