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

hcq update waits and signals in place (#4984)

Browse Source 
* hcq update waits and signals in place

* start amd

* amd works

* prettier

* test

* normal messages

* linetr

* linter 2
This commit is contained in:
nimlgen
2024-06-17 17:19:07 +03:00
committed by GitHub
parent 603a4a0ce1
commit 794acefbf3
5 changed files with 107 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/external/external_test_hcq.py vendored
View File

@@ -98,9 +98,9 @@ class TestHCQ(unittest.TestCase):
def test_update_exec(self):
q = TestHCQ.compute_queue()
exec_ptr = q.ptr()
exec_cmd_idx = len(q)
q.exec(TestHCQ.runner.clprg, TestHCQ.d0.kernargs_ptr, TestHCQ.runner.p.global_size, TestHCQ.runner.p.local_size)
q.update_exec(exec_ptr, (1,1,1), (1,1,1))
q.update_exec(exec_cmd_idx, (1,1,1), (1,1,1))
q.signal(TestHCQ.d0.timeline_signal, TestHCQ.d0.timeline_value).submit(TestHCQ.d0)
TestHCQ.d0._wait_signal(TestHCQ.d0.timeline_signal, TestHCQ.d0.timeline_value)
TestHCQ.d0.timeline_value += 1