mawei/tinygrad
1
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-13 00:15:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

hotfix: fix bert (#15613)

Browse Source
This commit is contained in:
nimlgen
2026-04-05 23:41:21 +03:00
committed by GitHub
parent e3986a6b74
commit 6a334ceb27
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

@@ -91,7 +91,7 @@ class HCQGraph(MultiGraphRunner):
enqueue_queue = self.copy_queues.setdefault((enqueue_dev, queue_idx),
enqueue_dev.hw_copy_queue_t(queue_idx=queue_idx).wait(self.signals['KICK'], self.kickoff_var))
out_signal = self.signals.setdefault(enqueue_queue, enqueue_dev.new_signal(value=0))
out_signal = self.signals.setdefault(enqueue_queue, self.devices[0].new_signal(value=0))
# Get dependencies based on input and output buffers.
rdeps = self._access_resources(ji.bufs, ji.prg.p.outs if is_exec_prg else [0], (enqueue_queue, j + 1)) #type:ignore