tinygrad/tinygrad
2
0
Fork 3
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-08-29 10:36:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix checking DiskBuffer is opened [pr] (#8547)

Browse Source 
`assert self.device.mem is not None` did not assert because `.mem` triggers AttributeError first
This commit is contained in:
chenyu
2025-01-09 18:58:36 -05:00
committed by GitHub
parent 62447c253d
commit 88661cd96f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/runtime/ops_disk.py
+1 -1
View File
@@ -67,7 +67,7 @@ class DiskBuffer:
self.device, self.size, self.offset = device, size, offset
def __repr__(self): return f"<DiskBuffer size={self.size} offset={self.offset}>"
def _buf(self) -> memoryview:
assert self.device.mem is not None, "DiskBuffer wasn't opened"
assert hasattr(self.device, "mem"), "DiskBuffer wasn't opened"
return memoryview(self.device.mem)[self.offset:self.offset+self.size]
MAP_LOCKED, MAP_POPULATE = 0 if OSX else 0x2000, getattr(mmap, "MAP_POPULATE", 0 if OSX else 0x008000)