sirhcm/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests 2 Actions 8 Activity

metal: fix incorrect _free on interpreter exit (#11158)

Browse Source
This commit is contained in:
Pyry Kovanen
2025-07-10 14:01:30 +03:00
committed by GitHub
parent 3d610f6d2b
commit 32117402dd
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/runtime/ops_metal.py
+2 -1
View File
@@ -198,7 +198,8 @@ class MetalAllocator(LRUAllocator[MetalDevice]):
ret = msg("newBufferWithLength:options:", objc_id)(self.dev.sysdevice, ctypes.c_ulong(size), MTLResourceOptions.MTLResourceStorageModeShared)
if ret.value is None: raise MemoryError(f"Metal OOM while allocating {size=}")
return MetalBuffer(ret, size)
def _free(self, opaque:MetalBuffer, options): msg("release")(opaque.buf)
def _free(self, opaque:MetalBuffer, options):
if msg is not None and libobjc is not None: msg("release")(opaque.buf)
def _transfer(self, dest:MetalBuffer, src:MetalBuffer, sz:int, src_dev:MetalDevice, dest_dev:MetalDevice):
dest_dev.synchronize()
src_command_buffer = msg("commandBuffer", objc_instance)(src_dev.mtl_queue)