Call this at the last iteration (it == n_batches).
| 402 | [[nodiscard]] bool AtEnd() const override { return at_end_; } |
| 403 | // Call this at the last iteration (it == n_batches). |
| 404 | virtual void EndIter() { |
| 405 | this->cache_info_->Commit(); |
| 406 | if (this->cache_info_->Size() != 0) { |
| 407 | CHECK_EQ(this->count_, this->cache_info_->Size()); |
| 408 | } |
| 409 | CHECK_GE(this->count_, 1); |
| 410 | this->count_ = 0; |
| 411 | } |
| 412 | |
| 413 | virtual void Reset(BatchParam const& param) { |
| 414 | TryLockGuard guard{single_threaded_}; |
no test coverage detected