提交 5d457939 编写于 作者: F fengjiayi

hot fix

上级 50104f18
...@@ -62,9 +62,9 @@ class FileReader : public ReaderBase { ...@@ -62,9 +62,9 @@ class FileReader : public ReaderBase {
// making it easier to access different type reader in Variables. // making it easier to access different type reader in Variables.
class ReaderHolder { class ReaderHolder {
public: public:
void Reset(ReaderBase* reader) { reader_.reset(reader); } void Reset(ReaderBase* reader) { reader_ = reader; }
ReaderBase* Get() const { return reader_.get(); } ReaderBase* Get() const { return reader_; }
void ReadNext(std::vector<LoDTensor>* out) { void ReadNext(std::vector<LoDTensor>* out) {
PADDLE_ENFORCE_NOT_NULL(reader_); PADDLE_ENFORCE_NOT_NULL(reader_);
...@@ -76,7 +76,7 @@ class ReaderHolder { ...@@ -76,7 +76,7 @@ class ReaderHolder {
} }
private: private:
std::unique_ptr<ReaderBase> reader_; ReaderBase* reader_;
}; };
} // namespace framework } // namespace framework
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册