From 3b4587fbf9c6df9c6ca4e51c52e945b01e49972e Mon Sep 17 00:00:00 2001
From: George Hotz <72895+geohot@users.noreply.github.com>
Date: Tue, 22 Oct 2024 11:17:50 +0700
Subject: [PATCH] no need to DEFINE_VAR arg sort [pr] (#7206)

---
 tinygrad/codegen/linearize.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/tinygrad/codegen/linearize.py b/tinygrad/codegen/linearize.py
index 566d40b27f..744f0d0da0 100644
--- a/tinygrad/codegen/linearize.py
+++ b/tinygrad/codegen/linearize.py
@@ -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))