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

update test_nested_after_contiguous_store (#15763)

Browse Source 
add kernel counts and some TODOs
This commit is contained in:
chenyu
2026-04-16 09:59:26 -04:00
committed by GitHub
parent 126cda45f8
commit d147e2a549
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/unit/test_assign.py
View File

@@ -633,10 +633,12 @@ class TestAssign(unittest.TestCase):
def test_nested_after_contiguous_store(self):
# Mirrors the nested contiguous-write-then-assign-back shape from torch backend view updates.
base = Tensor.empty(3, dtype=dtypes.int64)
base.assign(Tensor([1, 2, 3], dtype=dtypes.int64))
base.assign(Tensor([1, 2, 3], dtype=dtypes.int64)) # TODO: cycle error if commented this line out
contig = base.contiguous()
contig.assign(Tensor([1, 4, 3], dtype=dtypes.int64))
GlobalCounters.reset()
base.assign(contig).realize()
self.assertEqual(GlobalCounters.kernel_count, 3) # TODO: first copy is not required
self.assertEqual(base.tolist(), [1,4,3])
class TestAssignOrdering(unittest.TestCase):