From dc4dd898b7af053b73bbc0fce07d302e338aae0b Mon Sep 17 00:00:00 2001
From: wozeparrot <wozeparrot@gmail.com>
Date: Fri, 19 Sep 2025 14:09:12 -0700
Subject: [PATCH] fix: close mmap (#12249)

---
 tinygrad/runtime/ops_disk.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tinygrad/runtime/ops_disk.py b/tinygrad/runtime/ops_disk.py
index cae966aa1a..56809dafbc 100644
--- a/tinygrad/runtime/ops_disk.py
+++ b/tinygrad/runtime/ops_disk.py
@@ -39,7 +39,9 @@ class DiskDevice(Compiled):
   def _might_close(self):
     self.count -= 1
     if self.count == 0:
-      if self.fd is not None: os.close(self.fd)
+      if self.fd is not None:
+        os.close(self.fd)
+      if hasattr(self, "mem"): self.mem.close()
       self.size = None
   def _iouring_setup(self):
     DiskDevice._tried_io_uring_init = True