From 88eb2303261fa2b29b0bc43cebb8a6ad49c55785 Mon Sep 17 00:00:00 2001 From: nimlgen <138685161+nimlgen@users.noreply.github.com> Date: Mon, 29 Dec 2025 14:49:44 +0300 Subject: [PATCH] memory: correct pa allocator size (#13861) --- tinygrad/runtime/support/memory.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tinygrad/runtime/support/memory.py b/tinygrad/runtime/support/memory.py index ec70e61c4c..b8c6179d48 100644 --- a/tinygrad/runtime/support/memory.py +++ b/tinygrad/runtime/support/memory.py @@ -176,7 +176,7 @@ class MemoryManager: self.boot_allocator = TLSFAllocator(boot_size, base=0) self.ptable_allocator = TLSFAllocator(round_up(vram_size // 512, 1 << 20) if self.reserve_ptable else 0, base=self.boot_allocator.size) - self.pa_allocator = TLSFAllocator(vram_size - (64 << 20), base=self.boot_allocator.size + self.ptable_allocator.size) + self.pa_allocator = TLSFAllocator(vram_size - (off_sz:=self.boot_allocator.size + self.ptable_allocator.size) - (64 << 20), base=off_sz) self.root_page_table = pt_t(self.dev, self.palloc(0x1000, zero=not self.dev.smi_dev, boot=True), lv=first_lv) def _frag_size(self, va, sz, must_cover=True):