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) {
26
26
27
27
void ReaderBase::InsertDecoratedReader (
28
28
const std::shared_ptr<ReaderBase> &decorated_reader) {
29
- std::lock_guard<std::mutex> guard (mu_)) ;
29
+ std::lock_guard<std::mutex> guard (mu_);
30
30
decorated_readers_.emplace_back (decorated_reader);
31
31
}
32
32
Original file line number Diff line number Diff line change @@ -104,7 +104,13 @@ class ReaderHolder {
104
104
}
105
105
106
106
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
+ }
108
114
}
109
115
110
116
void Shutdown () {
Original file line number Diff line number Diff line change @@ -21,13 +21,12 @@ class StubDecoratedReader : public paddle::framework::DecoratedReader {
21
21
explicit StubDecoratedReader (const std::shared_ptr<ReaderBase> &reader)
22
22
: DecoratedReader(reader) {}
23
23
24
- void ReadNext (std::vector<paddle::framework::LoDTensor> *out) override {}
24
+ void ReadNextImpl (std::vector<paddle::framework::LoDTensor> *out) override {}
25
25
};
26
26
27
27
class StubRootReader : public paddle ::framework::ReaderBase {
28
28
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 {}
31
30
};
32
31
33
32
TEST (READER, decorate_chain) {
Original file line number Diff line number Diff line change @@ -375,9 +375,6 @@ def open_recordio_file(filename,
375
375
if pass_num > 1 :
376
376
main_prog_var = multi_pass (reader = main_prog_var , pass_num = pass_num )
377
377
378
- if for_parallel :
379
- main_prog_var = parallel (reader = main_prog_var )
380
-
381
378
return monkey_patch_reader_methods (main_prog_var )
382
379
383
380
You can’t perform that action at this time.
0 commit comments