github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

intiger div mod fix

Browse Source
This commit is contained in:
V-E-D
2025-05-20 11:46:39 +05:30
parent ceb9d94eab
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)