diff --git a/src/data/ellpack_page_raw_format.cu b/src/data/ellpack_page_raw_format.cu index 8316368ba885..62b29640c334 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 { if (!fi->Read(&impl->base_rowid)) { return false; } + dh::DefaultStream().Sync(); return true; }