github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-11 23:25:04 -05:00
Code Issues Packages Projects Releases Wiki Activity

hotfix: make KFD timings right

Browse Source
This commit is contained in:
George Hotz
2024-04-04 05:55:29 +00:00
parent 58d162315c
commit 3e72d745ea
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/runtime/ops_kfd.py
View File

@@ -230,7 +230,7 @@ class KFDProgram:
if wait:
self.device._wait_on(self.device.completion_signal.event_id)
assert (wp:=self.device.amd_aql_queue.write_dispatch_id) == (rp:=self.device.amd_aql_queue.read_dispatch_id), f"didn't run {wp} != {rp}"
return (self.device.completion_signal.end_ts-self.device.completion_signal.start_ts)/1e9
return (self.device.completion_signal.end_ts-self.device.completion_signal.start_ts)/1e8
class KFDAllocator(LRUAllocator):
def __init__(self, device:KFDDevice):