From 3ed53541ef808248baf265fdd3f6712674c23b56 Mon Sep 17 00:00:00 2001
From: Qinbin Li <liqinbin1998@gmail.com>
Date: Thu, 26 Jan 2023 20:43:52 +0800
Subject: [PATCH] change thrust env

---
 src/FedTree/Tree/hist_tree_builder.cpp | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/FedTree/Tree/hist_tree_builder.cpp b/src/FedTree/Tree/hist_tree_builder.cpp
index 4ba043ec..b2e7a26a 100644
--- a/src/FedTree/Tree/hist_tree_builder.cpp
+++ b/src/FedTree/Tree/hist_tree_builder.cpp
@@ -20,7 +20,6 @@
 #include <iterator>
 #include <algorithm>
 #include <random>
-#define THRUST_HOST_SYSTEM THRUST_HOST_SYSTEM_OMP
 
 using namespace thrust;
 
@@ -694,7 +693,7 @@ void HistTreeBuilder::compute_histogram_in_a_level(int level, int n_max_splits,
 
     this->build_n_hist++;
     if (n_column > 1){
-        inclusive_scan_by_key(thrust::omp::par, hist_fid, hist_fid + n_split,
+        inclusive_scan_by_key(thrust::host, hist_fid, hist_fid + n_split,
                               hist.host_data(), hist.host_data());
     }
     else{