sirhcm/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests 2 Actions 8 Activity

am: remove sleep from wait_reg (#10139)

Browse Source 
* am: remove sleep from wait_reg

* fst

* ooops
This commit is contained in:
nimlgen
2025-05-02 15:46:29 +03:00
committed by GitHub
parent 45bf7c5b81
commit 81410befc2
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/runtime/support/am/amdev.py
+2 -2
View File
@@ -347,9 +347,9 @@ class AMDev:
self.reg("regBIF_BX_PF0_RSMU_DATA").write(val)
def wait_reg(self, reg:AMRegister, value:int, mask=0xffffffff, timeout=10000) -> int:
for _ in range(timeout):
start_time = int(time.perf_counter() * 1000)
while int(time.perf_counter() * 1000) - start_time < timeout:
if ((rval:=reg.read()) & mask) == value: return rval
time.sleep(0.001)
raise RuntimeError(f'wait_reg timeout reg=0x{reg.addr:X} mask=0x{mask:X} value=0x{value:X} last_val=0x{rval}')
def _read_vram(self, addr, size) -> bytes: