Sync stream in ellpack format. (#10374)

dmlc · Jun 4, 2024 · 0808e50 · 0808e50
1 parent c4ec64d
commit 0808e50
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;
   }