Skip to content

Commit fbe99bc

Browse files
[Storage]Add deleter for mmap_storage get_slice (PaddlePaddle#75966)
* add deleter for get_slice * add destructor for mmapstorage * add destructor for mmapstorage
1 parent 70b14ac commit fbe99bc

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

paddle/fluid/pybind/pybind.cc

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1216,6 +1216,16 @@ struct MmapStorage {
12161216
}
12171217
#endif
12181218
}
1219+
~MmapStorage() {
1220+
if (base_ptr_) {
1221+
#if defined(WIN32) || defined(_WIN32) || defined(WIN64) || defined(_WIN6)
1222+
UnmapViewOfFile(base_ptr_);
1223+
#else
1224+
munmap(base_ptr_, size);
1225+
#endif
1226+
base_ptr_ = nullptr;
1227+
}
1228+
}
12191229
void *base_ptr_;
12201230
int64_t size;
12211231
};

0 commit comments

Comments
 (0)