Work around nvcc error. (#7673)

2022-02-19 01:41:46 +08:00
parent 3877043d41
commit d625dc2047
2 changed files with 14 additions and 17 deletions
--- a/src/tree/gpu_hist/row_partitioner.cuh
+++ b/src/tree/gpu_hist/row_partitioner.cuh
@@ -120,7 +120,8 @@ class RowPartitioner {

    int64_t* d_left_count = left_counts_.data().get() + nidx;
    // Launch 1 thread for each row
-    dh::LaunchN<1, 128>(segment.Size(), [=] __device__(size_t idx) {
+    dh::LaunchN<1, 128>(segment.Size(), [segment, op, left_nidx, right_nidx, d_ridx, d_left_count,
+                                         d_position] __device__(size_t idx) {
      // LaunchN starts from zero, so we restore the row index by adding segment.begin
      idx += segment.begin;
      RowIndexT ridx = d_ridx[idx];