Sync stream in ellpack format. (#10374)

2024-06-04 12:58:26 +08:00
parent c4ec64d409
commit 0808e50ae8
1 changed files with 1 additions and 0 deletions
--- 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;
  }