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

add failing tests for merge views (#8306)

Browse Source 
* add failing tests for merge views

* assert is not none

* make linter happy
This commit is contained in:
Jyotirmaya Mahanta
2024-12-18 11:57:38 +05:30
committed by GitHub
parent 801e199196
commit 45f2fb82d5
1 changed files with 50 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

63
test/unit/test_view.py
View File

@@ -6,38 +6,38 @@ from tinygrad.shape.view import View, merge_dims
class TestView(unittest.TestCase):
def test_canonicalize_empty_mask(self):
v = View.create(shape=(2,2,2), strides=(4,2,1), mask=((0,2),(0,2),(0,2)))
assert v.mask is None
self.assertIsNone(v.mask)
v = View.create(shape=(4,3,2), strides=(1,4,10), mask=((0,4),(0,3),(0,2)))
assert v.mask is None
self.assertIsNone(v.mask)
def test_minify_zero_strided_dims(self):
target = View.create(shape=(2,2), strides=(30,2), offset=7, mask=None)
v = View.create(shape=(2,1,2), strides=(30,0,2), offset=7, mask=None)
assert v.minify() == target
self.assertEqual(v.minify(), target)
v = View.create(shape=(1,2,2), strides=(0,30,2), offset=7, mask=None)
assert v.minify() == target
self.assertEqual(v.minify(), target)
v = View.create(shape=(2,2,1), strides=(30,2,0), offset=7, mask=None)
assert v.minify() == target
self.assertEqual(v.minify(), target)
v = View.create(shape=(2,1,1,2), strides=(30,0,0,2), offset=7, mask=None)
assert v.minify() == target
self.assertEqual(v.minify(), target)
v = View.create(shape=(1,1,2,2), strides=(0,0,30,2), offset=7, mask=None)
assert v.minify() == target
self.assertEqual(v.minify(), target)
v = View.create(shape=(2,2,1,1), strides=(30,2,0,0), offset=7, mask=None)
assert v.minify() == target
self.assertEqual(v.minify(), target)
v = View.create(shape=(1,2,2,1), strides=(0,30,2,0), offset=7, mask=None)
assert v.minify() == target
self.assertEqual(v.minify(), target)
v = View.create(shape=(1,2,1,2), strides=(0,30,0,2), offset=7, mask=None)
assert v.minify() == target
self.assertEqual(v.minify(), target)
def test_empty_mask_contiguous(self):
v1 = View.create(shape=(2,2,2), strides=(4,2,1), mask=None)
v2 = View.create(shape=(2,2,2), strides=(4,2,1), mask=((0,2),(0,2),(0,2)))
assert v1.contiguous == v2.contiguous
self.assertEqual(v1.contiguous, v2.contiguous)
v1 = View.create(shape=(1,1,1,4), strides=(0,0,0,1), offset=0, mask=None)
v2 = View.create(shape=(1,1,1,4), strides=(0,0,0,1), offset=0, mask=((0,1),(0,1),(0,1),(0,4)))
assert v1.contiguous == v2.contiguous
self.assertEqual(v1.contiguous, v2.contiguous)
v = View.create(shape=(2,3,4), mask=((0,2),(0,3),(0,4)))
assert v.contiguous
self.assertTrue(v.contiguous)
class TestMergeDims(unittest.TestCase):
def test_contiguous(self):
@@ -125,5 +125,42 @@ class TestMergeViews(unittest.TestCase):
self.assertIsNotNone(v)
self.assertEqual(v, View(shape=(1, 1, 6, 3), strides=(0, 0, 5, -1), offset=-2, mask=((0, 1), (0, 1), (1, 6), (0, 3)), contiguous=False))
@unittest.expectedFailure # TODO: fix these
def test_merges_from_fuzzer(self):
v0 = View(shape=(2, 4), strides=(2, 1), offset=-2, mask=((0, 2), (2, 4)), contiguous=False)
v1 = View(shape=(2, 4, 2, 2), strides=(4, 0, -2, -1), offset=3, mask=None, contiguous=False)
target = View(shape=(2, 4, 2, 2), strides=(2, 0, 0, -1), offset=1, mask=((0, 2), (0, 4), (0, 1), (0, 2)), contiguous=False)
v = v0 + v1
self.assertIsNotNone(v)
self.assertEqual(v, target)
v0 = View(shape=(5, 10, 12), strides=(100, 1, 10), offset=-20, mask=((0, 5), (0, 10), (2, 12)), contiguous=False)
v1 = View(shape=(10, 6, 5, 2, 2), strides=(12, 2, 120, 1, 0), offset=0, mask=None, contiguous=False)
target = View(shape=(10, 6, 5, 2, 2), strides=(1, 20, 100, 10, 0), offset=-20, mask=((0, 10), (1, 6), (0, 5), (0, 2), (0, 2)), contiguous=False)
v = v0 + v1
self.assertIsNotNone(v)
self.assertEqual(v, target)
v0 = View(shape=(8, 7, 3), strides=(1, 12, -4), offset=6, mask=((2, 6), (0, 7), (0, 3)), contiguous=False)
v1 = View(shape=(4, 2, 6, 2, 1), strides=(42, 21, 3, 1, 0), offset=4, mask=None, contiguous=False)
target = View(shape=(4, 2, 6, 2, 1), strides=(2, 1, 12, -4, 0), offset=14, mask=((1, 3), (0, 2), (0, 6), (0, 2), (0, 1)), contiguous=False)
v = v0 + v1
self.assertIsNotNone(v)
self.assertEqual(v, target)
v0 = View(shape=(7, 21, 3), strides=(54, 3, 1), offset=-9, mask=((0, 6), (3, 21), (0, 3)), contiguous=False)
v1 = View(shape=(5, 3, 3, 7), strides=(63, 1, 3, 9), offset=63, mask=None, contiguous=False)
target = View(shape=(5, 3, 3, 7), strides=(54, 1, 3, 9), offset=45, mask=((0, 5), (0, 3), (0, 3), (1, 7)), contiguous=False)
v = v0 + v1
self.assertIsNotNone(v)
self.assertEqual(v, target)
v0 = View(shape=(5, 1, 24), strides=(20, 0, 1), offset=-2, mask=((0, 5), (0, 1), (2, 22)), contiguous=False)
v1 = View(shape=(12, 2, 5, 2, 1), strides=(2, 1, 24, 0, 0), offset=0, mask=None, contiguous=False)
target = View(shape=(12, 2, 5, 2, 1), strides=(2, 1, 20, 0, 0), offset=-2, mask=((1, 11), (0, 2), (0, 5), (0, 2), (0, 1)), contiguous=False)
v = v0 + v1
self.assertIsNotNone(v)
self.assertEqual(v, target)
if __name__ == '__main__':
unittest.main()