[backport] Sync stream in ellpack format. (dmlc#10374)

trivialfis · Jun 5, 2024 · 0807a26 · 0807a26
1 parent eed3b3e
commit 0807a26
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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;
   }