HD.

dmlc · Dec 22, 2021 · c3656ca · c3656ca
1 parent b12c4e1
commit c3656ca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/common/categorical.h b/src/common/categorical.h
@@ -33,7 +33,7 @@ inline XGBOOST_DEVICE bool IsCat(Span<FeatureType const> ft, bst_feature_t fidx)
 }
 
 template <typename SizeT>
-bool InvalidCat(float cat, SizeT n) {
+XGBOOST_DEVICE bool InvalidCat(float cat, SizeT n) {
   return cat < 0 || cat > static_cast<float>(std::numeric_limits<bst_cat_t>::max()) || cat >= n;
 }