github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-22 13:28:06 -05:00
Code Issues Packages Projects Releases Wiki Activity

hcq fix timestamp around kernel (#5837)

Browse Source
This commit is contained in:
nimlgen
2024-07-31 16:12:27 +03:00
committed by GitHub
parent 2e087ca8e4
commit 431749dc21
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tinygrad/runtime/graph/hcq.py
View File

@@ -107,11 +107,11 @@ class HCQGraph(MultiGraphRunner):
self.copy_to_devs[Device[dest.device]].add(Device[src.device])
self.op_cmd_idx[j] = (enqueue_queue, len(enqueue_queue) - 1)
if signal_val is not None: enqueue_queue.signal(signal, signal_val)
# Encode finish profile timestamp (if needed).
if prof_info: enqueue_queue.timestamp(prof_info[1])
if signal_val is not None: enqueue_queue.signal(signal, signal_val)
for dev in self.devices:
for dep_dev in list(self.copy_to_devs[dev]) + [dev]:
if (last_j:=self.last_ji[self.copy_queues[dep_dev]]) is None: continue