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

x//a<b -> x <a*b for positive a (#6622)

Browse Source 
openpilot valids 47 -> 37
This commit is contained in:
chenyu
2024-09-20 04:38:47 -04:00
committed by GitHub
parent 72c7087420
commit 5707503048
4 changed files with 12 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/unit/test_image_valid.py
View File

@@ -69,6 +69,9 @@ class TestValidSimplification(unittest.TestCase):
gidx1 = Special("gidx1", 32)
self.assertEqual(render((10, 10, 4), (gidx1).lt(10), UOp(UOps.VECTORIZE, dtypes.int.vec(2), (gidx0, gidx1))),
"read_imagef(data0, smp, (int2)(gidx0,gidx1))")
# same thing, valid has a div
self.assertEqual(render((10, 10, 4), (gidx1//2).lt(5), UOp(UOps.VECTORIZE, dtypes.int.vec(2), (gidx0, gidx1))),
"read_imagef(data0, smp, (int2)(gidx0,gidx1))")
# 10x20 image, not out of bound
self.assertEqual(render((20, 10, 4), (gidx1).lt(10), UOp(UOps.VECTORIZE, dtypes.int.vec(2), (gidx0, gidx1))),
"((gidx1<10)?read_imagef(data0, smp, (int2)(gidx0,gidx1)):(float4)(0.0f,0.0f,0.0f,0.0f))")