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

split ranges (#12411)

Browse Source 
* split ranges

* simpler

* split ranges

* range str

* fix test

* oops

* faster

* no group 2

* tests

* dont_sub_ranges_for_image

* revert that
This commit is contained in:
George Hotz
2025-10-02 12:57:22 +08:00
committed by GitHub
parent 6fc6b51b59
commit 583553f467
11 changed files with 81 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/unit/test_simplify_valid_idx.py
View File

@@ -198,11 +198,11 @@ class TestValidIdxSimplification(unittest.TestCase):
"((((r0+r1)<1)!=True)&(((r2+r3)<1)!=True))")
def test_valid_with_non_const_rhs(self):
ridx0 = Range(0, 2**16)
ridx0 = Range(0, 1024)
ridx1 = Range(1, 4)
ridx2 = Range(2, 4)
valid = (ridx0<(ridx1*4 + ridx2))&(ridx0<-1).ne(True)
idx = ridx0%1024
idx = ridx0
load = get_gated_load_uop(valid, idx)
self.check(load,
"r0",