Skip to content

Commit b94f24d

Browse files
committed
Move StartPrefetcher and EndPrefetcher to private
1 parent 597c845 commit b94f24d

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

paddle/fluid/operators/reader/create_double_buffer_reader_op.cc

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,9 @@ class DoubleBufferReader : public framework::DecoratedReader {
6666
void ReadNext(std::vector<framework::LoDTensor>* out) override;
6767
void ReInit() override;
6868

69+
~DoubleBufferReader() { EndPrefetcher(); }
70+
71+
private:
6972
void StartPrefetcher() {
7073
channel_ = framework::MakeChannel<Item>(kChannelSize);
7174
prefetcher_ = std::thread([this] { PrefetchThreadFunc(); });
@@ -80,9 +83,6 @@ class DoubleBufferReader : public framework::DecoratedReader {
8083
channel_ = nullptr;
8184
}
8285

83-
~DoubleBufferReader() { EndPrefetcher(); }
84-
85-
private:
8686
void PrefetchThreadFunc();
8787

8888
std::thread prefetcher_;

0 commit comments

Comments
 (0)