range has only one src now [pr] (#10100)

* range has only one op now

* fix z3 checker

* ci fix

* needs shell

* try pip ensure update

* that ensurepip is useless

* upgrade pip before cache

* windows happy?
This commit is contained in:
George Hotz
2025-04-29 10:31:05 -04:00
committed by GitHub
parent 427471550a
commit c3ff308abb
14 changed files with 34 additions and 32 deletions
+2 -2
View File
@@ -91,11 +91,11 @@ class PythonProgram:
elif uop is Ops.CAST and isinstance(dtype, PtrDType):
ul[i] = inp[0]
elif uop is Ops.RANGE:
if i not in ul: ul[i] = [inp[0][0]] * warp_size
if i not in ul: ul[i] = [0] * warp_size
else:
for j in range(len(ul[i])):
ul[i][j] += 1
if ul[i][0] == inp[1][0]:
if ul[i][0] == inp[0][0]:
del ul[i]
i = loop_ends[i] + 1
continue