Skip to content

Commit 62c1133

Browse files
committed
Add mutex for decorated_chain
1 parent c48c586 commit 62c1133

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

paddle/fluid/framework/reader.cc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@ ReaderBase::~ReaderBase() {}
2121

2222
void ReaderBase::InsertDecoratedReader(
2323
const std::shared_ptr<ReaderBase> &decorated_reader) {
24+
std::lock_guard<std::mutex> guard(decorated_readers_mtx_);
2425
decorated_readers_.emplace_back(decorated_reader);
2526
}
2627

paddle/fluid/framework/reader.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@ class ReaderBase {
4545
const std::shared_ptr<ReaderBase>& decorated_reader);
4646
// A set of which readers that decorated this reader.
4747
std::vector<std::weak_ptr<ReaderBase>> decorated_readers_;
48+
std::mutex decorated_readers_mtx_;
4849
};
4950

5051
class DecoratedReader : public ReaderBase,

0 commit comments

Comments
 (0)