mawei/tinygrad
1
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-11 15:35:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'master' into move_gates_to_load_store

Browse Source
This commit is contained in:
George Hotz
2026-05-04 17:05:21 -07:00
committed by GitHub
parent 09ec34437d a357a0449a
commit d810bd2b41
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tinygrad/tensor.py
View File

@@ -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)