@ -18,9 +18,7 @@ namespace paddle {
namespace framework {
void ReaderBase::ReadNext(std::vector<LoDTensor> *out) {
if (status_ != ReaderStatus::kRunning) {
PADDLE_THROW("The reader is not at the status of 'running'.");
}
PADDLE_ENFORCE_EQ(status_, ReaderStatus::kRunning);
ReadNextImpl(out);
@ -14,7 +14,6 @@
#pragma once
#include <atomic>
#include <memory>
#include <vector>
@ -44,7 +43,7 @@ class ReaderBase {
virtual void StartImpl() = 0;
std::atomic<ReaderStatus> status_{kRunning};
ReaderStatus status_{kRunning};
};
class DecoratedReader : public ReaderBase {