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

hot fix

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