File tree Expand file tree Collapse file tree 4 files changed +10
-8
lines changed 
python/paddle/fluid/layers Expand file tree Collapse file tree 4 files changed +10
-8
lines changed Original file line number Diff line number Diff line change @@ -26,7 +26,7 @@ void ReaderBase::ReadNext(std::vector<LoDTensor> *out) {
2626
2727void  ReaderBase::InsertDecoratedReader (
2828    const  std::shared_ptr<ReaderBase> &decorated_reader) {
29-   std::lock_guard<std::mutex> guard (mu_)) ;
29+   std::lock_guard<std::mutex> guard (mu_);
3030  decorated_readers_.emplace_back (decorated_reader);
3131}
3232
Original file line number Diff line number Diff line change @@ -104,7 +104,13 @@ class ReaderHolder {
104104  }
105105
106106  void  ResetAll () {
107-     //  TODO(fengjiayi): The interface of reseting all.
107+     auto  end_readers = reader_->GetEndPoints ();
108+     for  (auto * reader : end_readers) {
109+       reader->Shutdown ();
110+     }
111+     for  (auto * reader : end_readers) {
112+       reader->Start ();
113+     }
108114  }
109115
110116  void  Shutdown () {
Original file line number Diff line number Diff line change @@ -21,13 +21,12 @@ class StubDecoratedReader : public paddle::framework::DecoratedReader {
2121  explicit  StubDecoratedReader (const  std::shared_ptr<ReaderBase> &reader)
2222      : DecoratedReader(reader) {}
2323
24-   void  ReadNext (std::vector<paddle::framework::LoDTensor> *out) override  {}
24+   void  ReadNextImpl (std::vector<paddle::framework::LoDTensor> *out) override  {}
2525};
2626
2727class  StubRootReader  : public  paddle ::framework::ReaderBase {
2828 public: 
29-   void  ReadNext (std::vector<paddle::framework::LoDTensor> *out) override  {}
30-   void  ReInit () override  {}
29+   void  ReadNextImpl (std::vector<paddle::framework::LoDTensor> *out) override  {}
3130};
3231
3332TEST (READER, decorate_chain) {
Original file line number Diff line number Diff line change @@ -375,9 +375,6 @@ def open_recordio_file(filename,
375375    if  pass_num  >  1 :
376376        main_prog_var  =  multi_pass (reader = main_prog_var , pass_num = pass_num )
377377
378-     if  for_parallel :
379-         main_prog_var  =  parallel (reader = main_prog_var )
380- 
381378    return  monkey_patch_reader_methods (main_prog_var )
382379
383380
    
 
   
 
     
   
   
          
     
  
    
     
 
    
      
     
 
     
    You can’t perform that action at this time.
  
 
    
  
     
    
      
        
     
 
       
      
     
   
 
    
    
  
 
  
 
     
    
0 commit comments