geohot/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Actions Activity

no need to DEFINE_VAR arg sort [pr] (#7206)

Browse Source
This commit is contained in:
George Hotz
2024-10-22 12:17:50 +08:00
committed by GitHub
parent 21acfc39d4
commit 3b4587fbf9
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/codegen/linearize.py
+1 -3
View File
@@ -59,9 +59,7 @@ def linearize_uop(sink:UOp, skip_check:bool=not __debug__) -> List[UOp]:
@functools.lru_cache(None)
def tuplize(u:UOp) -> Tuple[int, Any, Optional[DType], Tuple]:
# NOTE: this sort of DEFINE_VAR shouldn't have to be here. only for PTX
if u.op is UOps.DEFINE_VAR: arg = u.arg[0]
elif u.op is UOps.ALU: arg = u.arg.value
if u.op is UOps.ALU: arg = u.arg.value
else: arg = u.arg
return (u.op.value, arg, u.dtype, tuple(tuplize(x) for x in u.src))