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

arange folding with new ge (#6604)

Browse Source 
* arange folding with new ge

* bump allowed gated

* bump allowed speed
This commit is contained in:
George Hotz
2024-09-19 18:01:28 +08:00
committed by GitHub
parent 224151a958
commit a1a882b006
4 changed files with 20 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_hcq.py
View File

@@ -291,7 +291,7 @@ class TestHCQ(unittest.TestCase):
et = TestHCQ.d0._gpu2cpu_time(sig_en.timestamp, True) - TestHCQ.d0._gpu2cpu_time(sig_st.timestamp, True)
print(f"exec kernel time: {et:.2f} us")
assert 1 <= et <= (2500 if CI else 30)
assert 1 <= et <= (3000 if CI else 30)
def test_speed_copy_bandwidth(self):
if TestHCQ.d0.hw_copy_queue_t is None: self.skipTest("device does not support copy queue")