From 9a53fcbde40fb185e13533c2c334fd4e8230df62 Mon Sep 17 00:00:00 2001 From: nimlgen <138685161+nimlgen@users.noreply.github.com> Date: Thu, 13 Nov 2025 03:30:42 +0800 Subject: [PATCH] amd: sqtt on rdna3.5 (#13233) --- tinygrad/runtime/ops_amd.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tinygrad/runtime/ops_amd.py b/tinygrad/runtime/ops_amd.py index 7601516deb..0d59b9f95d 100644 --- a/tinygrad/runtime/ops_amd.py +++ b/tinygrad/runtime/ops_amd.py @@ -594,7 +594,8 @@ class AMDProgram(HCQProgram): self.dev.synchronize() for se, buf in enumerate(self.dev.sqtt_buffers): - wptr = ((self.dev.sqtt_wptrs.cpu_view().view(fmt='I')[se]&0x1FFFFFFF)-(((buf.va_addr//32)&0x1FFFFFFF) if self.dev.target[0] == 11 else 0))*32 + wptr = (self.dev.sqtt_wptrs.cpu_view().view(fmt='I')[se] & 0x1FFFFFFF) * 32 + if self.dev.target[:2] == (11, 0): wptr -= ((buf.va_addr // 32) & 0x1FFFFFFF) * 32 if DEBUG >= 5: print(f'\t{self.dev.device}: SE {se} blob size {wptr:#x}') assert wptr >= 0 and wptr <= buf.size, f"{wptr} > {buf.size}, should never happen"