Skip to content

Commit e8ee9dc

Browse files
committed
Several Polish
1 parent 0d2ccfb commit e8ee9dc

File tree

4 files changed

+9
-17
lines changed

4 files changed

+9
-17
lines changed

paddle/fluid/framework/reader.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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

paddle/fluid/framework/reader.h

Lines changed: 4 additions & 11 deletions
Original file line numberDiff line numberDiff 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.

paddle/fluid/operators/reader/create_py_reader_op.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff 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 =

paddle/fluid/operators/reader/create_recordio_file_reader_op.cc

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,8 +22,7 @@ template <bool ThreadSafe>
2222
class 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_;

0 commit comments

Comments
 (0)