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

@@ -802,7 +802,7 @@ class TestIdxUpcast(unittest.TestCase):
if s.ast.op is Ops.SINK:
renderer = Device[s.bufs[0].device].renderer
uops = linearize_uop(full_graph_rewrite(rewrite_shapetracker_with_index(s.ast, renderer), renderer))
renderer.render("test", uops)
renderer.render(uops)
return uops
def _assert(self, dtype: DType, a: Tensor):