gpu_hist performance fixes (#5558)

* Remove unnecessary cuda API calls * Fix histogram memory growth
2020-04-19 12:21:13 +12:00
parent e1f22baf8c
commit d6d1035950
7 changed files with 52 additions and 109 deletions
--- a/src/tree/gpu_hist/row_partitioner.cuh
+++ b/src/tree/gpu_hist/row_partitioner.cuh
@@ -108,7 +108,6 @@ class RowPartitioner {
  template <typename UpdatePositionOpT>
  void UpdatePosition(bst_node_t nidx, bst_node_t left_nidx,
                      bst_node_t right_nidx, UpdatePositionOpT op) {
-    dh::safe_cuda(cudaSetDevice(device_idx_));
    Segment segment = ridx_segments_.at(nidx);  // rows belongs to node nidx
    auto d_ridx = ridx_.CurrentSpan();
    auto d_position = position_.CurrentSpan();