github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-10 14:45:35 -05:00
Code Issues Packages Projects Releases Wiki Activity

default threefry (#6116)

Browse Source
This commit is contained in:
wozeparrot
2024-09-25 17:45:13 +08:00
committed by GitHub
parent 992cde05d7
commit c100f3d406
15 changed files with 82 additions and 82 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
viz/test_viz.py
View File

@@ -42,8 +42,8 @@ class TestViz(unittest.TestCase):
def test_ctx_groups(self):
contexts.clear()
schedule1 = Tensor.randn(4, 1).contiguous().schedule()
schedule2 = Tensor.randn(4, 4).contiguous().schedule()
schedule1 = Tensor.zeros(4, 1).contiguous().exp().schedule()
schedule2 = Tensor.zeros(4, 1).contiguous().exp().schedule()
list(lower_schedule(schedule1))
list(lower_schedule(schedule2))
ret = load_kernels(contexts)
@@ -118,8 +118,8 @@ class TestViz(unittest.TestCase):
def test_dedup_ast(self):
contexts.clear()
a = Tensor.randn(4, 4)+2
b = Tensor.randn(4, 4)+2
a = Tensor.empty(4, 4).contiguous().realize()+2
b = Tensor.empty(4, 4).contiguous().realize()+2
Tensor.schedule(a, b)
kernels = load_kernels(contexts)
self.assertEqual(len(kernels), 1)