commaai
/
tinygrad
mirror of https://github.com/commaai/tinygrad.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix idx setup in image_valid test_openpilot_conv3 (#6710) 

* fix idx setup in image_valid test_openpilot_conv3

* corrected output and sad
This commit is contained in:
chenyu 2024-09-24 05:49:04 -04:00 committed by GitHub
parent ae3f3fec38
commit 9e51879019
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
test/unit/test_image_valid.py
View File

@ -156,12 +156,12 @@ class TestValidSimplification(unittest.TestCase):
valid = alu2.lt(11)&(alu4.lt(3).ne(True))
shape = (8, 1024, 4)
idx = UOp(UOps.VECTORIZE, dtypes.int.vec(2), (((alu6+832)%1024),(alu2+((idx1+((ridx1+5)/8)+1)/2)+(-4))))
idx = UOp(UOps.VECTORIZE, dtypes.int.vec(2), (((alu6+832)%1024),(alu2+((idx1+((ridx1+5)//8)+1)//2)+(-4))))
# TODO: simplify idx
# alu0 = ((idx2*2)+ridx0)
self.assertEqual(render(shape, valid, idx),
"(((alu0<11)&((((idx1*8)+ridx1)<3)!=1))?read_imagef(data0, smp, (int2)((((idx1*512)+(ridx1*64)+idx0+832)%1024),(alu0+(-4)))):(float4)(0.0f,0.0f,0.0f,0.0f))") # noqa: E501
"(((alu0<11)&((((idx1*8)+ridx1)<3)!=1))?read_imagef(data0, smp, (int2)((((idx1*512)+(ridx1*64)+idx0+832)%1024),(alu0+((idx1+((ridx1+5)//8)+1)//2)+(-4)))):(float4)(0.0f,0.0f,0.0f,0.0f))") # noqa: E501
def test_simplify1(self):
# idx has the form (A % m, A // m + k) and valid has (c0 < A) and (A < c1)