Skip to content

Commit 5d45793

Browse files
committed
hot fix
1 parent 50104f1 commit 5d45793

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

paddle/fluid/framework/reader.h

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -62,9 +62,9 @@ class FileReader : public ReaderBase {
6262
// making it easier to access different type reader in Variables.
6363
class ReaderHolder {
6464
public:
65-
void Reset(ReaderBase* reader) { reader_.reset(reader); }
65+
void Reset(ReaderBase* reader) { reader_ = reader; }
6666

67-
ReaderBase* Get() const { return reader_.get(); }
67+
ReaderBase* Get() const { return reader_; }
6868

6969
void ReadNext(std::vector<LoDTensor>* out) {
7070
PADDLE_ENFORCE_NOT_NULL(reader_);
@@ -76,7 +76,7 @@ class ReaderHolder {
7676
}
7777

7878
private:
79-
std::unique_ptr<ReaderBase> reader_;
79+
ReaderBase* reader_;
8080
};
8181

8282
} // namespace framework

0 commit comments

Comments
 (0)