pytorch - 💡(How to fix) Fix torch.compile crashes with floating point exception for integer modulo by zero [1 comments, 2 participants]

🐛 Describe the bug

torch.compile with the Inductor backend crashes the Python process when compiling an integer modulo operation with zero divisors.

Eager execution raises a RuntimeError, but the compiled version causes a floating point exception and core dump.

import torch

def fn(a, b):
    return a % b

a = torch.tensor([15, -8, 22, 4])
b = torch.tensor([4, 0, 0, 1])

torch._dynamo.reset()

try:
    print("eager:", fn(a, b).tolist())
except Exception as e:
    print("eager caught:", type(e).__name__)

torch._dynamo.reset()

print("compile:", flush=True, end=" ")
c = torch.compile(fn, backend="inductor", fullgraph=True)(a, b)
print(c.tolist())

Error logs

eager caught: RuntimeError compile: Floating point exception (core dumped)

Versions

PyTorch version: 2.11.0+cu130 Is debug build: False CUDA used to build PyTorch: 13.0 ROCM used to build PyTorch: N/A

OS: Ubuntu 24.04.2 LTS (x86_64) GCC version: (Ubuntu 13.3.0-6ubuntu2~24.04.1) 13.3.0 Clang version: Could not collect CMake version: Could not collect Libc version: glibc-2.39

Python version: 3.10.20 (main, Mar 11 2026, 17:46:40) [GCC 14.3.0] (64-bit runtime) Python platform: Linux-5.15.167.4-microsoft-standard-WSL2-x86_64-with-glibc2.39 Is CUDA available: False CUDA runtime version: Could not collect CUDA_MODULE_LOADING set to: N/A GPU models and configuration: GPU 0: NVIDIA GeForce RTX 3080 Laptop GPU Nvidia driver version: 545.92 cuDNN version: Could not collect Is XPU available: False HIP runtime version: N/A MIOpen runtime version: N/A Is XNNPACK available: True Caching allocator config: N/A

Versions of relevant libraries: [pip3] numpy==2.2.6 [pip3] onnx==1.21.0 [pip3] onnx2torch==1.5.15 [pip3] onnxruntime==1.23.2 [pip3] torch==2.11.0 [pip3] torchvision==0.26.0 [pip3] triton==3.6.0

cc @chauhang @penguinwu @voznesenskym @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @ipiszy @kadeng @muchulee8 @amjames @aakhundov @coconutruben @jataylo