github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-10 23:48:01 -05:00
Code Issues Packages Projects Releases Wiki Activity

wtf (#3944)

Browse Source
This commit is contained in:
uuuvn
2024-03-27 02:49:28 +02:00
committed by GitHub
parent 5530b0cbed
commit d6d902afe9
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/device.py
View File

@@ -170,7 +170,7 @@ class LRUAllocator(Allocator): # pylint: disable=abstract-method
opaques.clear()
def free(self, opaque:Any, size:int, options:Optional[BufferOptions]=None):
if getenv("LRU", 1) and (options is None or not options.signal): self.cache[(size, options)].append(opaque)
else: super().free(size, size, options)
else: super().free(opaque, size, options)
class _MallocAllocator(LRUAllocator):
def _alloc(self, size:int, options:BufferOptions): return (ctypes.c_uint8 * size)()