Revert "Switch to per-thread default stream (#9396)" (#9413)

This reverts commit f7f673b00c.
This commit is contained in:
Jiaming Yuan
2023-07-25 03:03:28 +08:00
committed by GitHub
parent 1b657a5513
commit 3a9996173e
8 changed files with 35 additions and 25 deletions

View File

@@ -73,7 +73,7 @@ void TestSortPositionBatch(const std::vector<int>& ridx_in, const std::vector<Se
dh::device_vector<int8_t> tmp;
SortPositionBatch<uint32_t, decltype(op), int>(dh::ToSpan(d_batch_info), dh::ToSpan(ridx),
dh::ToSpan(ridx_tmp), dh::ToSpan(counts),
total_rows, op, &tmp);
total_rows, op, &tmp, nullptr);
auto op_without_data = [=] __device__(auto ridx) { return ridx % 2 == 0; };
for (size_t i = 0; i < segments.size(); i++) {