Skip to content

Commit 659ebc9

Browse files
[Storage]Add deleter for mmap_storage get_slice (#75966) (#76029)
* add deleter for get_slice * add destructor for mmapstorage * add destructor for mmapstorage
1 parent 38a0e1d commit 659ebc9

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
@@ -1129,6 +1129,16 @@ struct MmapStorage {
11291129
}
11301130
#endif
11311131
}
1132+
~MmapStorage() {
1133+
if (base_ptr_) {
1134+
#if defined(WIN32) || defined(_WIN32) || defined(WIN64) || defined(_WIN6)
1135+
UnmapViewOfFile(base_ptr_);
1136+
#else
1137+
munmap(base_ptr_, size);
1138+
#endif
1139+
base_ptr_ = nullptr;
1140+
}
1141+
}
11321142
void *base_ptr_;
11331143
int64_t size;
11341144
};

0 commit comments

Comments
 (0)