From a357a0449a3e7dfaef42438c0ee54fb4e4619993 Mon Sep 17 00:00:00 2001 From: chenyu Date: Mon, 4 May 2026 19:27:36 -0400 Subject: [PATCH] Tensor.div cleanup (#16041) --- tinygrad/tensor.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tinygrad/tensor.py b/tinygrad/tensor.py index c73337c6f2..dd77c1c133 100644 --- a/tinygrad/tensor.py +++ b/tinygrad/tensor.py @@ -1305,8 +1305,7 @@ class Tensor(OpMixin): """ if rounding_mode is None: return super().div(x, reverse) # type: ignore[arg-type] numerator, denominator = self._broadcasted(x, reverse) - if dtypes.is_int(dt:=least_upper_dtype(numerator.dtype, denominator.dtype)): - numerator, denominator = numerator.cast(dt), denominator.cast(dt) + if dtypes.is_int(numerator.dtype): if rounding_mode == "trunc": return numerator.idiv(denominator) if rounding_mode == "floor": truncate_div, truncate_mod = numerator.idiv(denominator), numerator._binop(Ops.MOD, denominator, False)