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

hcq better timout haandling (#8269)

Browse Source
This commit is contained in:
nimlgen
2024-12-16 13:44:55 +03:00
committed by GitHub
parent f05fd118a2
commit a2a4ff30dc
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tinygrad/runtime/support/hcq.py
View File

@@ -210,10 +210,9 @@ class HCQSignal(Generic[DeviceType]):
timeout: Maximum time to wait in milliseconds. Defaults to 10s.
"""
start_time = int(time.time() * 1000)
while (time_spent:=int(time.time() * 1000) - start_time) < timeout:
if self.value >= value: return
while self.value < value and (time_spent:=int(time.time() * 1000) - start_time) < timeout:
self._sleep(time_spent)
raise RuntimeError(f"Wait timeout: {timeout} ms! (the signal is not set to {value}, but {self.value})")
if self.value < value: raise RuntimeError(f"Wait timeout: {timeout} ms! (the signal is not set to {value}, but {self.value})")
@contextlib.contextmanager
def hcq_profile(dev:HCQCompiled, enabled, desc, queue_type:Optional[Type[HWQueue]]=None, queue:Optional[HWQueue]=None):