Refine Shutdown Impl

guochaorong-patch-1
yuyang18 7 years ago
parent 7ca8528031
commit ee7d8b4d66
No known key found for this signature in database
GPG Key ID: 6DFF29878217BE5F

@ -52,12 +52,10 @@ class DoubleBufferReader : public framework::DecoratedReader {
void ReadNextImpl(std::vector<framework::LoDTensor>* out) override;
~DoubleBufferReader() { EndPrefetcher(); }
private:
void ShutdownImpl() override {
EndPrefetcher();
reader_->Shutdown();
EndPrefetcher();
}
void StartImpl() override {

@ -48,9 +48,9 @@ class ShuffleReader : public framework::DecoratedReader {
private:
void ShutdownImpl() override {
reader_->Shutdown();
buffer_.clear();
iteration_pos_ = 0;
reader_->Shutdown();
}
void StartImpl() override {

@ -36,8 +36,6 @@ class MultiFileReader : public framework::ReaderBase {
void ReadNextImpl(std::vector<framework::LoDTensor>* out) override;
~MultiFileReader() { EndScheduler(); }
private:
void ShutdownImpl() override { EndScheduler(); }

Loading…
Cancel
Save