File tree Expand file tree Collapse file tree 4 files changed +9
-17
lines changed Expand file tree Collapse file tree 4 files changed +9
-17
lines changed Original file line number Diff line number Diff line change @@ -36,7 +36,7 @@ void ReaderBase::Start() {
3636 }
3737}
3838
39- ReaderBase::~ReaderBase () {}
39+ ReaderBase::~ReaderBase () { Shutdown (); }
4040
4141} // namespace framework
4242} // namespace paddle
Original file line number Diff line number Diff line change @@ -39,9 +39,9 @@ class ReaderBase {
3939 protected:
4040 virtual void ReadNextImpl (std::vector<LoDTensor>* out) = 0;
4141
42- virtual void ShutdownImpl () = 0;
42+ virtual void ShutdownImpl () {}
4343
44- virtual void StartImpl () = 0;
44+ virtual void StartImpl () {}
4545
4646 ReaderStatus status_{kRunning };
4747};
@@ -61,15 +61,8 @@ class DecoratedReader : public ReaderBase {
6161 std::shared_ptr<ReaderBase> reader_;
6262};
6363
64- class FileReader : public ReaderBase {
65- public:
66- FileReader () : ReaderBase() {}
67-
68- protected:
69- void ShutdownImpl () override {}
70-
71- void StartImpl () override {}
72- };
64+ // FileReader is just a conceptual class.
65+ class FileReader : public ReaderBase {};
7366
7467// The ReaderHolder is used as reader' unified wrapper,
7568// making it easier to access different type reader in Variables.
Original file line number Diff line number Diff line change @@ -56,8 +56,8 @@ class CreatePyReaderOp : public framework::OperatorBase {
5656
5757 const std::string& queue_name = Input (" blocking_queue" );
5858 auto * queue_holder_var = scope.FindVar (queue_name);
59- PADDLE_ENFORCE (
60- queue_holder_var != nullptr ,
59+ PADDLE_ENFORCE_NOT_NULL (
60+ queue_holder_var,
6161 " No LoDTensorBlockingQueueHolder variable with name %s found" ,
6262 queue_name);
6363 auto * queue_holder =
Original file line number Diff line number Diff line change @@ -22,8 +22,7 @@ template <bool ThreadSafe>
2222class RecordIOFileReader : public framework ::FileReader {
2323 public:
2424 explicit RecordIOFileReader (const std::string& filename)
25- : FileReader(),
26- scanner_(filename),
25+ : scanner_(filename),
2726 dev_ctx_(*platform::DeviceContextPool::Instance ().Get(
2827 platform::CPUPlace ())) {
2928 if (ThreadSafe) {
@@ -42,7 +41,7 @@ class RecordIOFileReader : public framework::FileReader {
4241 }
4342 }
4443
45- void ShutdownImpl () override { scanner_.Reset (); }
44+ void StartImpl () override { scanner_.Reset (); }
4645
4746 private:
4847 std::unique_ptr<std::mutex> mutex_;
You can’t perform that action at this time.
0 commit comments