Skip to content

Commit 3ed5354

Browse files
committed
change thrust env
1 parent 436f3ae commit 3ed5354

1 file changed

Lines changed: 1 addition & 2 deletions

File tree

src/FedTree/Tree/hist_tree_builder.cpp

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,6 @@
2020
#include <iterator>
2121
#include <algorithm>
2222
#include <random>
23-
#define THRUST_HOST_SYSTEM THRUST_HOST_SYSTEM_OMP
2423

2524
using namespace thrust;
2625

@@ -694,7 +693,7 @@ void HistTreeBuilder::compute_histogram_in_a_level(int level, int n_max_splits,
694693

695694
this->build_n_hist++;
696695
if (n_column > 1){
697-
inclusive_scan_by_key(thrust::omp::par, hist_fid, hist_fid + n_split,
696+
inclusive_scan_by_key(thrust::host, hist_fid, hist_fid + n_split,
698697
hist.host_data(), hist.host_data());
699698
}
700699
else{

0 commit comments

Comments
 (0)