pytorch - 💡(How to fix) Fix `torch.nansum` with dtype=torch.int64 raises `NotImplementedError` in eager but succeeds in inductor

import torch

def fn(x):
    return torch.nansum(x, dim=1, dtype=torch.int64)

compiled_fn = torch.compile(fn, backend="inductor", fullgraph=True)
x = torch.tensor([[float("nan"), 2.0], [3.0, float("nan")]], dtype=torch.float32)

try:
    eager = fn(x)
    print("eager returned:", eager)
except Exception as exc:
    eager = type(exc).__name__ + ": " + str(exc)
    print("eager error:", eager)

try:
    compiled = compiled_fn(x)
    print("compiled returned:", compiled)
except Exception as exc:
    compiled = type(exc).__name__ + ": " + str(exc)
    print("compiled error:", compiled)

---

eager error: NotImplementedError: "nansum_cpu" not implemented for 'Long'
compiled returned: tensor([2, 3])