mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

LOOP -> RANGE (#4650)

Browse source
This commit is contained in:
George Hotz 2024-05-19 06:40:20 -07:00 • committed by GitHub
commit 4753283221
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
8 changed files with 23 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tinygrad/runtime/ops_python.py
View file

@ -39,7 +39,7 @@ class PythonProgram:
loop_ends: Dict[int, int] = {}
while i < len(self.uops):
uop, dtype, idp, arg = self.uops[i]
void_ops = {UOps.STORE, UOps.ENDLOOP, UOps.BARRIER, UOps.IF, UOps.ENDIF}
void_ops = {UOps.STORE, UOps.ENDRANGE, UOps.BARRIER, UOps.IF, UOps.ENDIF}
if uop is UOps.DEFINE_ACC: idp.clear()
inp = [ul[v] for v in idp if self.uops[v][0] not in void_ops]
dtp = [dl[v] for v in idp if self.uops[v][0] not in void_ops]
@ -62,7 +62,7 @@ class PythonProgram:
if g: _store(m, o, v)
i += 1
continue
elif uop is UOps.ENDLOOP:
elif uop is UOps.ENDRANGE:
loop_ends[idp[0]] = i
i = idp[0]
continue
@ -90,7 +90,7 @@ class PythonProgram:
ul[i] = [[arg] * warp_size for _ in range(dtype.count)] if dtype.count > 1 else [arg] * warp_size
elif uop is UOps.DEFINE_ACC:
ul[i] = [[arg[0]] * warp_size for _ in range(dtype.count)] if dtype.count > 1 else [arg[0]] * warp_size
elif uop is UOps.LOOP:
elif uop is UOps.RANGE:
if i not in ul: ul[i] = [inp[0][0]] * warp_size
else:
for j in range(len(ul[i])):