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

qcom as_buffer (#6371)

Browse Source
This commit is contained in:
nimlgen
2024-09-05 16:06:25 +03:00
committed by GitHub
parent a1a15b54c9
commit 93cd1fa034
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tinygrad/runtime/ops_qcom.py
View File

@@ -313,6 +313,10 @@ class QCOMAllocator(HCQAllocator):
self.device.synchronize()
ctypes.memmove(from_mv(dest), src.va_addr, dest.nbytes)
def as_buffer(self, src:HCQBuffer) -> memoryview:
self.device.synchronize()
return to_mv(src.va_addr, src.size)
def _free(self, opaque, options:BufferOptions):
self.device.synchronize()
self.device._gpu_free(opaque)