e8a962575a
- New parameter `on_host`. - Abstract format creation and stream creation into policy classes.
31 lines
859 B
C++
31 lines
859 B
C++
/**
|
|
* Copyright 2021-2024, XGBoost Contributors
|
|
*/
|
|
#include "sparse_page_source.h"
|
|
|
|
#include <filesystem> // for exists
|
|
#include <string> // for string
|
|
#include <cstdio> // for remove
|
|
#include <numeric> // for partial_sum
|
|
|
|
namespace xgboost::data {
|
|
void Cache::Commit() {
|
|
if (!written) {
|
|
std::partial_sum(offset.begin(), offset.end(), offset.begin());
|
|
written = true;
|
|
}
|
|
}
|
|
|
|
void TryDeleteCacheFile(const std::string& file) {
|
|
// Don't throw, this is called in a destructor.
|
|
auto exists = std::filesystem::exists(file);
|
|
if (!exists) {
|
|
LOG(WARNING) << "External memory cache file " << file << " is missing.";
|
|
}
|
|
if (std::remove(file.c_str()) != 0) {
|
|
LOG(WARNING) << "Couldn't remove external memory cache file " << file
|
|
<< "; you may want to remove it manually";
|
|
}
|
|
}
|
|
} // namespace xgboost::data
|