github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-15 09:05:40 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix jit realize issue (#3258)

Browse Source
This commit is contained in:
George Hotz
2024-01-26 18:27:35 -08:00
committed by GitHub
parent 4197ef17c4
commit c4d870db0d
3 changed files with 20 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tinygrad/runtime/ops_hip.py
View File

@@ -160,7 +160,7 @@ class HIPSyncEvent(JITRunner):
to_mv(rawbufs[0]._buf, 4).cast("I")[0] = 0
hip_set_device(self.device.device)
check(hip.hipStreamWriteValue32(None, rawbufs[0]._buf, 1, 0))
update_stats(colored("sync", "red"), 0, 0, {}, None, 1, device=self.dname)
update_stats(colored("sync", "red"), 0, 0, {}, None, 1, jit, device=self.dname)
class HIPWaitEvent(JITRunner):
def __init__(self, device):
@@ -169,4 +169,4 @@ class HIPWaitEvent(JITRunner):
def __call__(self, rawbufs:List[Buffer], var_vals, wait=False, jit=False):
hip_set_device(self.device.device)
check(hip.hipStreamWaitValue32(None, rawbufs[0]._buf, 1, 1, 0xFFFFFFFF))
update_stats(colored("wait", "RED"), 0, 0, {}, None, 1, device=self.dname)
update_stats(colored("wait", "RED"), 0, 0, {}, None, 1, jit, device=self.dname)