From 757ceab2a2d0c4993f008c9d6aeb4909d45896cf Mon Sep 17 00:00:00 2001 From: nimlgen <138685161+nimlgen@users.noreply.github.com> Date: Wed, 5 Nov 2025 19:12:59 +0800 Subject: [PATCH] system: allow using vidmem for uc mem (#13104) --- tinygrad/runtime/support/system.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tinygrad/runtime/support/system.py b/tinygrad/runtime/support/system.py index b0e73b8415..49b1a32a37 100644 --- a/tinygrad/runtime/support/system.py +++ b/tinygrad/runtime/support/system.py @@ -252,7 +252,7 @@ class LNXPCIIfaceBase: def alloc(self, size:int, host=False, uncached=False, cpu_access=False, contiguous=False, force_devmem=False, **kwargs) -> HCQBuffer: # NOTE: logic on macos is different, since bar is small - should_use_sysmem = host or (((uncached or cpu_access) if OSX else (uncached and cpu_access)) and not force_devmem) + should_use_sysmem = host or ((cpu_access if OSX else (uncached and cpu_access)) and not force_devmem) if should_use_sysmem: vaddr = self.dev_impl.mm.alloc_vaddr(size:=round_up(size, mmap.PAGESIZE), align=mmap.PAGESIZE) memview, paddrs = System.alloc_sysmem(size, vaddr=vaddr, contiguous=contiguous)