pytorch - 💡(How to fix) Fix `torch.remainder` returns incorrect results with float16 tensor and Python scalar on Inductor CPU [1 participants]

import torch

def f(x):
    return torch.remainder(x, 3.14)

x = torch.tensor(
    [-4.0, -1.5, -0.0, 0.0, 0.1, 1.0, 3.140625, 8.0],
    dtype=torch.float16,
)

eager = f(x)
compiled = torch.compile(f, backend="inductor")(x)

print("eager =", eager)
print("compiled =", compiled)

---

eager = tensor([2.2812, 1.6406, -0.0000, 0.0000, 0.1000, 1.0000, 0.0000, 1.7188],
       dtype=torch.float16)
compiled = tensor([2.2793e+00, 1.6396e+00, 0.0000e+00, 0.0000e+00, 9.9976e-02, 1.0000e+00,
        6.2466e-04, 1.7197e+00], dtype=torch.float16)