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

hcq: remove redunt syncs and fix typing (#11096)

Browse Source 
Before this patch the code could issues reduntdant syncs because of
the typing issue. Current tests should cover all correctness checks.
This commit is contained in:
nimlgen
2025-07-04 21:49:47 +03:00
committed by GitHub
parent 41aa54eb5a
commit 577afc9f05
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -92,9 +92,9 @@ class HCQGraph(MultiGraphRunner):
if (qa:=queue_access[enqueue_queue][dep_queue]) is None or qa < dep_val:
opt_deps.append((self.signals[dep_queue], dep_val))
queue_access[enqueue_queue][dep_queue] = dep_val
dev_access[enqueue_queue].update(dev_access[dep_queue])
# Ensure device is ready for use in current context: the graph has initialized the device and it's safe to operate on it within this graph.
for dep_queue, _ in opt_deps: dev_access[enqueue_queue].update(dev_access[dep_queue])
sync_signals = [(self.signals[d], self.kickoff_var) for b in ji.bufs if (d:=Device[cast(Buffer, b).device]) not in dev_access[enqueue_queue]]
dev_access[enqueue_queue].update(cast(HCQCompiled, Device[cast(Buffer, b).device]) for b in ji.bufs)