pytorch/torch/distributed/algorithms
Vasiliy Kuznetsov f15ab8a7f2 AO migration: replace torch internal callsites (#94170)
Summary:

Do the following renames:
`torch.quantization` -> `torch.ao.quantization`
`torch.nn.quantized` -> `torch.ao.nn.quantized`
`torch.nn.quantizable` -> `torch.ao.nn.quantizable`
`torch.nn.qat` -> `torch.ao.nn.qat`
`torch.nn.intrinsic` -> `torch.ao.nn.intrinsic`

And then, do
`torch.ao.nn.quantized._reference` -> `torch.ao.nn.quantized.reference` to clean up the aftermath of https://github.com/pytorch/pytorch/pull/84974

Then, manually update `test/test_module_init.py` to fix hanging whitespace due to the replace.

Run this script to do the replacements: https://gist.github.com/vkuzo/7f7afebf8c31b9ba48306223e68a1c82

This is for https://github.com/pytorch/pytorch/issues/81667

Test plan: CI
Pull Request resolved: https://github.com/pytorch/pytorch/pull/94170
Approved by: https://github.com/jerryzh168
2023-02-07 02:32:23 +00:00
..
_checkpoint [AC] Add trailing "." to _CHECKPOINT_PREFIX like FSDP (#87951) 2022-10-28 22:05:29 +00:00
_comm_hooks [FSDP] Clean up FlatParamHandle dtypes, post-backward hook (#90660) 2022-12-13 07:34:59 +00:00
_optimizer_overlap make fsdp folder to be public (#72084) 2022-02-02 15:50:14 +00:00
_quantization Change docstring type callable to Callable for consistency (#82487) 2022-08-01 17:26:09 +00:00
ddp_comm_hooks AO migration: replace torch internal callsites (#94170) 2023-02-07 02:32:23 +00:00
model_averaging Update hierarchical_model_averager.py (#85648) 2022-10-03 06:15:20 +00:00
__init__.py Make _Join, _Joinable, _JoinHook public (#62605) 2021-08-03 12:20:11 -07:00
join.py Integrate xdoctest - Rebased (#82797) 2022-08-12 02:08:01 +00:00