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

remove pyint [pr] (#7016)

Browse Source 
* remove pyint

* bump time on tp [pr]

* dont truncate in const fold

* remove dead code

* Revert "dont truncate in const fold"

This reverts commit 29c81db0f7.

* remove define_var
This commit is contained in:
George Hotz
2024-10-12 22:36:24 +08:00
committed by GitHub
parent 38d45dfba5
commit 85a45164fb
14 changed files with 74 additions and 89 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/unit/test_shapetracker.py
View File

@@ -11,7 +11,7 @@ from tinygrad.codegen.uopgraph import sym
from itertools import product
def shapetracker_getitem(st:ShapeTracker, val:int):
idx, valid = st.reshape((st.size,)).to_indexed_uops([UOp.const(dtypes.pyint, val)])
idx, valid = st.reshape((st.size,)).to_indexed_uops([UOp.const(dtypes.int, val)])
idx, valid = graph_rewrite(idx, sym), graph_rewrite(valid, sym)
assert idx.op is UOps.CONST and valid.op is UOps.CONST
return idx.arg, valid.arg