Check.

dmlc · Jan 13, 2022 · cce83a3 · cce83a3
1 parent 417f88d
commit cce83a3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/common/quantile.cu b/src/common/quantile.cu
@@ -588,7 +588,7 @@ struct InvalidCatOp {
 
   XGBOOST_DEVICE bool operator()(size_t i) {
     auto fidx = dh::SegmentId(ptrs, i);
-    return InvalidCat(values[i], ptrs[fidx + 1] - ptrs[fidx]);
+    return IsCat(ft, fidx) && InvalidCat(values[i], ptrs[fidx + 1] - ptrs[fidx]);
   }
 };
 }  // anonymous namespace