flip Ops.END srcs (#12882)

* flip Ops.END srcs

* backward

* late end split
This commit is contained in:
George Hotz
2025-10-23 12:47:50 +08:00
committed by GitHub
parent 74b4cfe44b
commit e85cee0aad
10 changed files with 44 additions and 43 deletions
+2 -2
View File
@@ -57,8 +57,8 @@ class PythonProgram:
dtp = [dl[v] for v in idp if self.uops[v][0] not in void_ops]
if getenv("TRACE"): print(i, uop, dtype, arg, inp, dtp)
if uop is Ops.END:
loop_ends[idp[0]] = i
i = idp[0]
loop_ends[idp[1]] = i
i = idp[1]
continue
if uop in (Ops.BARRIER, Ops.IF, Ops.ENDIF, Ops.SINK, Ops.NOOP):
# in the python emulator, the warp is always in sync