github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-16 01:26:29 -05:00
Code Issues Packages Projects Releases Wiki Activity

hotfix: time.time -> time.perf_counter

Browse Source
This commit is contained in:
George Hotz
2024-12-16 11:32:49 -08:00
parent 2bb298f38d
commit cda34ccadf
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/hcq.py
View File

@@ -209,8 +209,8 @@ class HCQSignal(Generic[DeviceType]):
value: The value to wait for.
timeout: Maximum time to wait in milliseconds. Defaults to 10s.
"""
start_time = int(time.time() * 1000)
while self.value < value and (time_spent:=int(time.time() * 1000) - start_time) < timeout:
start_time = int(time.perf_counter() * 1000)
while self.value < value and (time_spent:=int(time.perf_counter() * 1000) - start_time) < timeout:
self._sleep(time_spent)
if self.value < value: raise RuntimeError(f"Wait timeout: {timeout} ms! (the signal is not set to {value}, but {self.value})")