Move into cpu predictor.

dmlc · Jun 1, 2021 · 4f444c2 · 4f444c2
1 parent c3c82e4
commit 4f444c2
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 14 deletions.
diff --git a/src/predictor/cpu_predictor.cc b/src/predictor/cpu_predictor.cc
@@ -28,6 +28,20 @@ namespace predictor {
 
 DMLC_REGISTRY_FILE_TAG(cpu_predictor);
 
+template <bool has_missing, bool has_categorical>
+bst_node_t GetLeafIndex(RegTree const &tree, const RegTree::FVec &feat,
+                        RegTree::CategoricalSplitMatrix const& cats) {
+  bst_node_t nid = 0;
+  while (!tree[nid].IsLeaf()) {
+    unsigned split_index = tree[nid].SplitIndex();
+    auto fvalue = feat.GetFvalue(split_index);
+    auto nodes = common::Span<RegTree::Node const>{tree.GetNodes()};
+    nid = GetNextNode<has_missing, has_categorical>(
+        nodes, nid, fvalue, has_missing && feat.IsMissing(split_index), cats);
+  }
+  return nid;
+}
+
 bst_float PredValue(const SparsePage::Inst &inst,
                     const std::vector<std::unique_ptr<RegTree>> &trees,
                     const std::vector<int> &tree_info, int bst_group,

diff --git a/src/predictor/predict_fn.h b/src/predictor/predict_fn.h
@@ -31,20 +31,6 @@ inline XGBOOST_DEVICE bst_node_t GetNextNode(
   }
   return nid;
 }
-
-template <bool has_missing, bool has_categorical>
-bst_node_t GetLeafIndex(RegTree const &tree, const RegTree::FVec &feat,
-                        RegTree::CategoricalSplitMatrix const& cats) {
-  bst_node_t nid = 0;
-  while (!tree[nid].IsLeaf()) {
-    unsigned split_index = tree[nid].SplitIndex();
-    auto fvalue = feat.GetFvalue(split_index);
-    auto nodes = common::Span<RegTree::Node const>{tree.GetNodes()};
-    nid = GetNextNode<has_missing, has_categorical>(
-        nodes, nid, fvalue, has_missing && feat.IsMissing(split_index), cats);
-  }
-  return nid;
-}
 }  // namespace predictor
 }  // namespace xgboost
 #endif  // XGBOOST_PREDICTOR_PREDICT_FN_H_