Revert "name from uops [pr] (#9151)" (#9154)

This reverts commit 28897be9a2.
2025-02-18 16:06:44 +08:00
parent 28897be9a2
commit caee42e8a6
4 changed files with 5 additions and 6 deletions
@@ -697,5 +697,5 @@ class Kernel:
    mem_bytes = sum(max(x.src[0].dtype.itemsize * x.st_arg.real_size() for x in group)
      for _, group in itertools.groupby([x for x in self.ast.toposort if x.op in GroupOp.Buffer and x.src[0].op is Ops.DEFINE_GLOBAL],
                        key=lambda x: (x.op, x.src[0].arg)))
-    return ProgramSpec(src, self.opts.device, self.ast, self.uops, mem_estimate=mem_bytes,
+    return ProgramSpec(self.uops[0].arg, src, self.opts.device, self.ast, self.uops, mem_estimate=mem_bytes,
                       global_size=[1,1,1] if self.opts.has_local else None, local_size=[1,1,1] if self.opts.has_local else None)