github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-08 22:48:25 -05:00
Code Issues Packages Projects Releases Wiki Activity

MSTACK little non-functional changes (#10648)

Browse Source
This commit is contained in:
George Hotz
2025-06-05 13:20:22 -07:00
committed by GitHub
parent 79d04d1baf
commit 4c315f8e17
5 changed files with 11 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/test_tensor.py
View File

@@ -518,6 +518,10 @@ class TestTinygrad(unittest.TestCase):
except ValueError:
Tensor.zeros(2, 2).realize()
def test_shrink(self):
t = Tensor.arange(32).contiguous().realize()
self.assertListEqual(t[16:20].tolist(), [16,17,18,19])
@unittest.skip("this test is just flaky, sync issue")
class TestMoveTensor(unittest.TestCase):
d0, d1 = f"{Device.DEFAULT}:0", f"{Device.DEFAULT}:1"