From 7ddc888ad5be36a2b59c019d8015bc88ec966953 Mon Sep 17 00:00:00 2001 From: nimlgen <138685161+nimlgen@users.noreply.github.com> Date: Mon, 16 Feb 2026 22:48:07 +0300 Subject: [PATCH] am: 48bit for gfx950 (#14799) --- tinygrad/runtime/support/am/ip.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tinygrad/runtime/support/am/ip.py b/tinygrad/runtime/support/am/ip.py index 4ef90d3297..2b7171f726 100644 --- a/tinygrad/runtime/support/am/ip.py +++ b/tinygrad/runtime/support/am/ip.py @@ -70,7 +70,7 @@ class AM_GMC(AM_IP): self.trans_futher = self.adev.ip_ver[am.GC_HWIP] < (10, 0, 0) # 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.address_space_mask = (1 << (48 if self.adev.ip_ver[am.GC_HWIP][:2] in {(9,4), (9,5)} 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))