frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Support nanj in inductor (#144064)

Browse Source 
Fixes https://github.com/pytorch/pytorch/issues/144029
Pull Request resolved: https://github.com/pytorch/pytorch/pull/144064
Approved by: https://github.com/amjames, https://github.com/eellison
This commit is contained in:
Isuru Fernando
2025-01-06 04:55:22 +00:00
committed by PyTorch MergeBot
parent 417354d953
commit 8633845090
3 changed files with 15 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
torch/_inductor/codegen/wrapper.py
View File

@ -759,6 +759,7 @@ class PythonWrapperCodegen(CodeGen):
import os
import tempfile
from math import inf, nan
from cmath import nanj
from torch._inductor.hooks import run_intermediate_hooks
from torch._inductor.utils import maybe_profile
from torch._inductor.codegen.memory_planning import _align as align