| 632 | |
| 633 | template <typename T> |
| 634 | void DatasetImpl<T>::WaitPreLoadDone() { |
| 635 | VLOG(3) << "DatasetImpl<T>::WaitPreLoadDone() begin"; |
| 636 | for (std::thread& t : preload_threads_) { |
| 637 | t.join(); |
| 638 | } |
| 639 | input_channel_->Close(); |
| 640 | int64_t in_chan_size = input_channel_->Size(); |
| 641 | input_channel_->SetBlockSize(in_chan_size / thread_num_ + 1); |
| 642 | VLOG(3) << "DatasetImpl<T>::WaitPreLoadDone() end"; |
| 643 | } |
| 644 | |
| 645 | // release memory data |
| 646 | template <typename T> |