mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages 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
commit d147e2a549
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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):