From a44dc4bdb086a2a0b655c7f54b5fc5242f247530 Mon Sep 17 00:00:00 2001
From: Awni Hannun <awni@apple.com>
Date: Mon, 24 Feb 2025 13:57:59 -0800
Subject: [PATCH] fix leaking objc (#1898)

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

diff --git a/mlx/backend/metal/device.cpp b/mlx/backend/metal/device.cpp
index a987ac4b3..7a82e2fb3 100644
--- a/mlx/backend/metal/device.cpp
+++ b/mlx/backend/metal/device.cpp
@@ -249,6 +249,7 @@ Device::~Device() {
   for (auto& l : library_map_) {
     l.second->release();
   }
+  stream_map_.clear();
   device_->release();
 }