add vmin vmax of SPECIAL

chenyuxyz · Jul 23, 2024 · f3e335a · f3e335a
1 parent a85493b
commit f3e335a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tinygrad/codegen/uops.py b/tinygrad/codegen/uops.py
@@ -101,10 +101,12 @@ def divides(self, v):
   @functools.cached_property
   def vmax(self) -> UOp:
     if self.op is UOps.DEFINE_VAR: return self.src[1]
+    if self.op is UOps.SPECIAL: return self.const(dtypes.bigint, self.arg[1])
     return UOp.const(dtypes.float, math.inf)
   @functools.cached_property
   def vmin(self) -> UOp:
     if self.op is UOps.DEFINE_VAR: return self.src[0]
+    if self.op is UOps.SPECIAL: return self.const(dtypes.bigint, 0)
     return UOp.const(dtypes.float, -math.inf)
 
 class UPat: