github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-12 07:35:16 -05:00
Code Issues Packages Projects Releases Wiki Activity

hotfix: that mem_used was in the wrong place

Browse Source
This commit is contained in:
George Hotz
2024-03-28 17:09:04 -07:00
parent 1fa0351acb
commit d085837179
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/buffer.py
View File

@@ -18,7 +18,6 @@ class Buffer:
if isinstance(dtype, ImageDType): options = BufferOptions(image=dtype) # TODO: image hack shouldn't be here. where should it be?
self.device, self.size, self.dtype, self.options = device, size, dtype, options
if opaque is not None: self.allocate(opaque)
if not self.device.startswith("DISK"): GlobalCounters.mem_used += self.nbytes
if initial_value is not None:
self.allocate()
self.copyin(memoryview(initial_value))
@@ -27,6 +26,7 @@ class Buffer:
from tinygrad.device import Device
self.allocator = Device[self.device].allocator
self._buf = opaque if opaque is not None else self.allocator.alloc(self.nbytes, self.options)
if not self.device.startswith("DISK"): GlobalCounters.mem_used += self.nbytes
return self
def __reduce__(self):
buf = None