pytorch/torch/ao
albanD 058fb1790f Fix compilation and "import torch" issues for cpython 3.14 (#158184)
Beginning of process for 3.14 bringup.

State of things from this PR:
- Nothing too scary looking from the Dynamo CPython side, nothing we heavily rely on seems to be missing @williamwen42
- The existing check that makes torch.compile() nicely fail is working as expected. So all these empty functions shouldn't cause any weirdness.
- The `__module__` update changes look suspicious, we should investigate what is the reason and impact of that, in particular for our public API checking @jbschlosser
- Leaving the weakref.py thread safety change as a follow up to keep this a bit simpler. I vendored the whole struct in the meantime FYI @ezyang

EDIT: The `__module__` change is even more cursed than I though due to changes to Union and Optional type where the `__module__` field cannot be changed anymore. See https://github.com/python/cpython/issues/132139 for details.
For now, I'm just skipping the `__module__` setting for 3.14 which will trip the public API checks. Will revisit once I have a final answer on the cpython issue.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/158184
Approved by: https://github.com/msaroufim
2025-07-15 05:06:55 +00:00
..
nn remove allow-untyped-defs from torch/ao/nn/intrinsic/quantized/dynamic/modules/linear_relu.py (#157848) 2025-07-12 15:42:12 +00:00
ns [BE][13/16] fix typos in torch/ (torch/ao/) (#156603) 2025-06-29 04:34:04 +00:00
pruning [remove untyped defs] batch 1 (#157011) 2025-06-30 23:54:40 +00:00
quantization Fix compilation and "import torch" issues for cpython 3.14 (#158184) 2025-07-15 05:06:55 +00:00
__init__.py [BE][Easy] improve submodule discovery for torch.ao type annotations (#144680) 2025-01-13 17:16:19 +00:00