tinygrad/tinygrad
2
0
Fork 3
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-08-29 12:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

regression test for reshape fix #2616 (#2620)

Browse Source
This commit is contained in:
chenyu
2023-12-05 11:46:33 -05:00
committed by GitHub
parent c257a0dd99
commit dd8b4632a4
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/unit/test_shapetracker.py
+6
View File
@@ -476,6 +476,12 @@ class TestComplexShapeTracker(unittest.TestCase):
print(self.st.views)
assert self.st.contiguous
def test_reshape_stable_diffusion(self):
# regression test for https://github.com/tinygrad/tinygrad/pull/2616
st = ShapeTracker((View((2, 1920, 32, 32), (1310720, 1024, 32, 1), 0, ((0, 2), (0, 1280), (0, 32), (0, 32)), False),))
st = st.reshape((2, 32, 240, 256))
assert len(st.views) == 2
class TestSingleShapeTracker(unittest.TestCase):
def setUp(self):
self.st = CheckingShapeTracker((7,4))