github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix hcq sync (#5062)

Browse Source 
* fix hcq sync

* rewrite

* linter + comment

* fix profiler

* no default dict

* correct sync of unjitted transfer

* fix test
This commit is contained in:
nimlgen
2024-06-26 17:50:37 +03:00
committed by GitHub
parent 3604642847
commit 16405b973a
3 changed files with 77 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_graph.py
View File

@@ -146,7 +146,6 @@ class TestGraph(unittest.TestCase):
helper_test_graphs(Device[d0].graph, graphs)
@unittest.skipUnless(Device.DEFAULT in {"CUDA", "NV", "AMD"}, "mutidevice graph required")
@unittest.skipIf(Device.DEFAULT in {"NV", "AMD"}, "Broken #5050")
def test_copies_2_devs(self):
d0, d1 = Device.DEFAULT, f"{Device.DEFAULT}:1"
b0 = [helper_alloc_rawbuffer(d0, fill=True) for _ in range(3)]
@@ -159,7 +158,6 @@ class TestGraph(unittest.TestCase):
helper_test_graphs(Device[d0].graph, graphs)
@unittest.skipUnless(Device.DEFAULT in {"CUDA", "NV", "AMD"}, "mutidevice graph required")
@unittest.skipIf(Device.DEFAULT in {"NV", "AMD"}, "Broken #5050")
def test_copies_after_graph_global(self):
d0, d1, d2, d3 = Device.DEFAULT, f"{Device.DEFAULT}:1", f"{Device.DEFAULT}:2", f"{Device.DEFAULT}:3"
b0 = [helper_alloc_rawbuffer(d0, fill=True) for _ in range(8)]
@@ -207,7 +205,6 @@ class TestGraph(unittest.TestCase):
helper_test_graphs(Device[d0].graph, graphs)
@unittest.skipUnless(Device.DEFAULT in {"CUDA", "NV", "AMD"}, "mutidevice graph required")
@unittest.skipIf(Device.DEFAULT in {"NV", "AMD"}, "Broken #5050")
def test_graph_after_copies_devs(self):
d0, d1, d2, d3 = Device.DEFAULT, f"{Device.DEFAULT}:1", f"{Device.DEFAULT}:2", f"{Device.DEFAULT}:3"
b0 = [helper_alloc_rawbuffer(d0, fill=True) for _ in range(8)]