From f98ce25ab979be2b8791efd1f9ddeaab3e3e2dbc Mon Sep 17 00:00:00 2001
From: Awni Hannun <awni@apple.com>
Date: Tue, 1 Apr 2025 12:59:48 -0700
Subject: [PATCH] fix residency set for real (#2032)

---
 mlx/backend/metal/allocator.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mlx/backend/metal/allocator.cpp b/mlx/backend/metal/allocator.cpp
index 2c81482a5..d21c66ade 100644
--- a/mlx/backend/metal/allocator.cpp
+++ b/mlx/backend/metal/allocator.cpp
@@ -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();