zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-24 06:28:08 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix residency set clearing (#2027)

Browse Source
This commit is contained in:
Awni Hannun
2025-03-30 16:27:26 -07:00
committed by GitHub
parent fe597e141c
commit b2d2b37888
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
mlx/backend/metal/allocator.cpp
View File

@@ -48,6 +48,9 @@ int BufferCache::clear() {
int n_release = 0; int n_release = 0;
for (auto& [size, holder] : buffer_pool_) { for (auto& [size, holder] : buffer_pool_) {
if (holder->buf) { if (holder->buf) {
if (holder->buf->heap()) {
residency_set_.erase(holder->buf);
}
holder->buf->release(); holder->buf->release();
n_release++; n_release++;
} }