mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

intiger div mod fix

Browse source
This commit is contained in:
V-E-D 2025-05-20 11:46:39 +05:30
commit d5d2f201bf
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tinygrad/tensor.py
View file

@ -3471,7 +3471,11 @@ class Tensor(SimpleMathTrait):
```
"""
numerator, denominator = self._broadcasted(x, reverse)
d = numerator.cast(least_upper_float(numerator.dtype)) * denominator.cast(least_upper_float(denominator.dtype)).reciprocal()
if dtypes.is_int(numerator.dtype) or dtypes.is_int(denominator.dtype):
d = numerator.cast(dtypes.float64) * denominator.cast(dtypes.float64).reciprocal()
d = d.cast(dtypes.float32)
else:
d = numerator * denominator.reciprocal()
output_dtype = numerator.dtype if dtypes.is_int(numerator.dtype) else d.dtype
if rounding_mode == "trunc": return d.trunc().cast(output_dtype)
if rounding_mode == "floor": return d.floor().cast(output_dtype)