From e036d6df8940261abe1872cb197f95c126bf1a9a Mon Sep 17 00:00:00 2001
From: chenyu <chenyu@fastmail.com>
Date: Thu, 1 Jan 2026 18:08:23 -0500
Subject: [PATCH] properly fix DiskDevice reuse (#13961)

---
 test/unit/test_disk_tensor.py | 23 +++++++++++++++++++++++
 tinygrad/runtime/ops_disk.py  |  2 +-
 2 files changed, 24 insertions(+), 1 deletion(-)

diff --git a/test/unit/test_disk_tensor.py b/test/unit/test_disk_tensor.py
index 5462792282..36f087221c 100644
--- a/test/unit/test_disk_tensor.py
+++ b/test/unit/test_disk_tensor.py
@@ -360,6 +360,29 @@ class TestDiskTensor(unittest.TestCase):
     x = Tensor.empty(size + len(test), dtype=dtypes.uint8, device=f"disk:{fn}").to("CPU").realize()
     assert x[size:].data().tobytes() == test
 
+  def test_disk_device_reuse(self):
+    from tinygrad.runtime.ops_disk import DiskDevice
+    fn = pathlib.Path(temp("dt_device_reuse"))
+    fn.unlink(missing_ok=True)
+    fn.write_bytes(bytes(range(256)))
+    # create first tensor and realize it
+    t1 = Tensor.empty(128, device=f"disk:{fn}", dtype=dtypes.uint8)
+    t1.to("CPU").realize()
+    # get the DiskDevice and check internal state
+    disk_device = Device[f"DISK:{fn}"]
+    assert isinstance(disk_device, DiskDevice)
+    assert disk_device.count == 1
+    assert hasattr(disk_device, "mem")
+    first_fd = disk_device.fd
+    # create second tensor on same file - should reuse the device, not re-open
+    t2 = Tensor.empty(64, device=f"disk:{fn}", dtype=dtypes.uint8)
+    t2.to("CPU").realize()
+    assert disk_device.count == 2
+    assert disk_device.fd == first_fd, "file descriptor changed - file was unnecessarily re-opened"
+    # verify data is correct
+    np.testing.assert_equal(t1.numpy(), np.arange(128, dtype=np.uint8))
+    np.testing.assert_equal(t2.numpy(), np.arange(64, dtype=np.uint8))
+
 class TestPathTensor(unittest.TestCase):
   def setUp(self):
     self.temp_dir = tempfile.TemporaryDirectory()
diff --git a/tinygrad/runtime/ops_disk.py b/tinygrad/runtime/ops_disk.py
index 5f1dc144c4..2725b6d103 100644
--- a/tinygrad/runtime/ops_disk.py
+++ b/tinygrad/runtime/ops_disk.py
@@ -18,7 +18,7 @@ class DiskDevice(Compiled):
     super().__init__(device, DiskAllocator(self), None, None)
   def _might_open(self, size:int):
     assert self.size is None or size <= self.size, f"can't reopen Disk tensor with larger size, opened with {self.size}, tried to open with {size}"
-    if self.size is not None and hasattr(self.device, "mem"):
+    if self.size is not None and hasattr(self, "mem"):
       self.count += 1
       return
     filename = self.device[len("disk:"):]