Fix inplace predict missing value. (#6787)

2021-03-27 05:36:10 +08:00
parent 5c87c2bba8
commit a59c7323b4
8 changed files with 97 additions and 33 deletions
--- a/src/data/device_adapter.cuh
+++ b/src/data/device_adapter.cuh
@@ -16,9 +16,14 @@ namespace xgboost {
 namespace data {

 struct IsValidFunctor : public thrust::unary_function<Entry, bool> {
-  explicit IsValidFunctor(float missing) : missing(missing) {}
-
  float missing;
+
+  XGBOOST_DEVICE explicit IsValidFunctor(float missing) : missing(missing) {}
+
+  __device__ bool operator()(float value) const {
+    return !(common::CheckNAN(value) || value == missing);
+  }
+
  __device__ bool operator()(const data::COOTuple& e) const {
    if (common::CheckNAN(e.value) || e.value == missing) {
      return false;