github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-07 03:00:26 -04:00
Code Issues Packages Projects Releases Wiki Activity

qcom correct image size (#6355)

Browse Source
This commit is contained in:
nimlgen
2024-09-04 18:32:35 +03:00
committed by GitHub
parent a86e7d598e
commit dde7a0d79c
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/runtime/ops_qcom.py
View File

@@ -219,7 +219,7 @@ class QCOMProgram(HCQProgram):
self._parse_lib()
# reserve some space after for gpu to use
self.lib_gpu = self.device.allocator.alloc(len(self.image) + 0x20000, options=BufferOptions(cpu_access=True, nolru=True))
self.lib_gpu = self.device.allocator.alloc(self.image_size, options=BufferOptions(cpu_access=True, nolru=True))
to_mv(self.lib_gpu.va_addr, self.image_size)[:] = self.image
self.pvtmem_size_per_item = round_up(self.pvtmem, 512) >> 9