From ceff388f3d861d1e973130ae5618ef106b61c28d Mon Sep 17 00:00:00 2001
From: nimlgen <138685161+nimlgen@users.noreply.github.com>
Date: Thu, 18 Dec 2025 11:20:43 +0300
Subject: [PATCH] am: extend va space (#13745)

---
 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 476ca84b13..ca470a79c3 100644
--- a/tinygrad/runtime/support/am/amdev.py
+++ b/tinygrad/runtime/support/am/amdev.py
@@ -134,7 +134,7 @@ class AMPageTableEntry:
   def supports_huge_page(self, paddr:int): return self.lv >= am.AMDGPU_VM_PDB2
 
 class AMMemoryManager(MemoryManager):
-  va_allocator = TLSFAllocator(512 * (1 << 30), base=0x200000000000) # global for all devices.
+  va_allocator = TLSFAllocator((1 << 44), base=0x200000000000) # global for all devices.
 
   def on_range_mapped(self):
     # Invalidate TLB after mappings.