github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-06 21:53:53 -05:00
Code Issues Packages Projects Releases Wiki Activity

am: mi300 48bit address space (#14004)

Browse Source 
* am: mi300 48bit address space

* fix
This commit is contained in:
nimlgen
2026-01-04 15:19:25 +03:00
committed by GitHub
parent 606786e152
commit bf356ae996
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tinygrad/runtime/support/am/ip.py
View File

@@ -57,8 +57,8 @@ class AM_GMC(AM_IP):
self.trans_futher = self.adev.ip_ver[am.GC_HWIP] < (10, 0, 0)
# GFX11/GFX12 has 44-bit address space
self.address_space_mask = (1 << 44) - 1
# mi3xx has 48-bit, others have 44-bit address space
self.address_space_mask = (1 << (48 if self.adev.ip_ver[am.GC_HWIP][:2] == (9,4) else 44)) - 1
self.memscratch_xgmi_paddr = self.adev.paddr2xgmi(self.adev.mm.palloc(0x1000, zero=False, boot=True))
self.dummy_page_xgmi_paddr = self.adev.paddr2xgmi(self.adev.mm.palloc(0x1000, zero=False, boot=True))