From bdd4dd92387f4660bba9c45bfd46891802dcbb60 Mon Sep 17 00:00:00 2001
From: nimlgen <138685161+nimlgen@users.noreply.github.com>
Date: Fri, 2 May 2025 12:19:59 +0300
Subject: [PATCH] am: do not expect aligned size in valloc (#10136)

---
 tinygrad/runtime/support/am/amdev.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tinygrad/runtime/support/am/amdev.py b/tinygrad/runtime/support/am/amdev.py
index 40820f03d2..fcea8f7d35 100644
--- a/tinygrad/runtime/support/am/amdev.py
+++ b/tinygrad/runtime/support/am/amdev.py
@@ -206,7 +206,7 @@ class AMMemoryManager:
 
   def valloc(self, size:int, align=0x1000, uncached=False, contigous=False) -> AMMapping:
     # Alloc physical memory and map it to the virtual address
-    va = self.alloc_vaddr(size, align)
+    va = self.alloc_vaddr(size:=round_up(size, 0x1000), align)
 
     if contigous: paddrs = [(self.palloc(size, zero=True), size)]
     else: