github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-10 07:28:15 -05:00
Code Issues Packages Projects Releases Wiki Activity

rangeify permuted assign (#12299)

Browse Source 
* enable RANGEIFY=1 test_assign

* work

* rangeify=0 asserts this ast

* remove that

* beta test, it's correct though

* skip multi

* matches torch/np output

* memcopy without memcopy

* can remove this

* rangeify isn't silently wrong anymore

* diff cleanup

* use UOp toposort instead of global tags

* actual assert TestRangeifyAssign

* step

* work

* this isn't optimizing away now

* some todos

* test fusion schedule

* typo

* dedup idxs

* cleaner

* pre

* work

* diff
This commit is contained in:
qazal
2025-09-29 07:27:57 +03:00
committed by GitHub
parent ed90de6583
commit 250cb10e8f
5 changed files with 54 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_rangeify.py
View File

@@ -15,6 +15,8 @@ class TestRangeifyAssign(unittest.TestCase):
print(lst)
print(lst2)
print(lst3)
self.assertListEqual(lst, lst3)
self.assertListEqual(lst2, B.permute(1, 0).tolist())
N = 256