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 { if (!fi->Read(&impl->base_rowid)) { return false; } + dh::DefaultStream().Sync(); return true; }