add name uop (#9149)

* add name uop, TODO: refactor renderer to use

* renderer uses name uop

* fix tests

* render

* ptx
This commit is contained in:
George Hotz
2025-02-18 15:26:58 +08:00
committed by GitHub
parent 2db8b4046a
commit a4dab3ec3f
18 changed files with 49 additions and 31 deletions

View File

@@ -952,7 +952,7 @@ class TestLinearizer(unittest.TestCase):
sink = UOp(Ops.SINK, src=(store,))
lin = Kernel(sink)
lin.linearize()
assert len(lin.uops) <= 9, "too many uops"
assert len(lin.uops) <= 10, "too many uops"
def test_upcast_cse(self):
# when upcasting, within a subtree, there may be common expressions.