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

fix residency set for real (#2032)

Browse Source
This commit is contained in:
Awni Hannun 2025-04-01 12:59:48 -07:00 committed by GitHub
parent de5f38fd48
commit f98ce25ab9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -48,7 +48,7 @@ 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()) { if (!holder->buf->heap()) {
residency_set_.erase(holder->buf); residency_set_.erase(holder->buf);
} }
holder->buf->release(); holder->buf->release();