zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-22 21:48:08 +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
1 changed files with 1 additions and 1 deletions
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;
for (auto& [size, holder] : buffer_pool_) {
if (holder->buf) {
if (holder->buf->heap()) {
if (!holder->buf->heap()) {
residency_set_.erase(holder->buf);
}
holder->buf->release();