diff --git a/tinygrad/codegen/uops.py b/tinygrad/codegen/uops.py index a0690c7881f13..4f644a23941d8 100644 --- 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: