From 9a400731d9030cf82fcb4b03a56eb223b903747c Mon Sep 17 00:00:00 2001
From: Jiaming Yuan <jm.yuan@outlook.com>
Date: Sat, 19 Mar 2022 23:22:23 +0800
Subject: [PATCH] Replace device sync with stream sync. (#7737)

---
 src/tree/updater_gpu_hist.cu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/tree/updater_gpu_hist.cu b/src/tree/updater_gpu_hist.cu
index e1fdb11aa..cb7dd9b7e 100644
--- a/src/tree/updater_gpu_hist.cu
+++ b/src/tree/updater_gpu_hist.cu
@@ -665,7 +665,7 @@ struct GPUHistMakerDevice {
           new_candidates[i * 2 + 1] = GPUExpandEntry();
         }
       }
-      dh::safe_cuda(cudaDeviceSynchronize());
+      dh::DefaultStream().Sync();
       driver.Push(new_candidates.begin(), new_candidates.end());
       expand_set = driver.Pop();
     }