From 0808e50ae8f1fb87881a5039e4884fef4a81515c Mon Sep 17 00:00:00 2001
From: Jiaming Yuan <jm.yuan@outlook.com>
Date: Tue, 4 Jun 2024 12:58:26 +0800
Subject: [PATCH] Sync stream in ellpack format. (#10374)

---
 src/data/ellpack_page_raw_format.cu | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/data/ellpack_page_raw_format.cu b/src/data/ellpack_page_raw_format.cu
index 8316368ba..62b29640c 100644
--- a/src/data/ellpack_page_raw_format.cu
+++ b/src/data/ellpack_page_raw_format.cu
@@ -36,6 +36,7 @@ class EllpackPageRawFormat : public SparsePageFormat<EllpackPage> {
     if (!fi->Read(&impl->base_rowid)) {
       return false;
     }
+    dh::DefaultStream().Sync();
     return true;
   }