github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-08 22:48:25 -05:00
Code Issues Packages Projects Releases Wiki Activity

am: extend va space (#13745)

Browse Source
This commit is contained in:
nimlgen
2025-12-18 11:20:43 +03:00
committed by GitHub
parent 99e667bdcd
commit ceff388f3d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/runtime/support/am/amdev.py
View File

@@ -134,7 +134,7 @@ class AMPageTableEntry:
def supports_huge_page(self, paddr:int): return self.lv >= am.AMDGPU_VM_PDB2 def supports_huge_page(self, paddr:int): return self.lv >= am.AMDGPU_VM_PDB2
class AMMemoryManager(MemoryManager): 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): def on_range_mapped(self):
# Invalidate TLB after mappings. # Invalidate TLB after mappings.