pytorch/test/quantization/ao_migration
Zafar 44c91383d3 [quant][ao_migration] Base package in tests
Adding a base package as an argument to the testing routines.
That will allow us to test other locations that are being migrated.
For example

```
AOMigrationTestCase._test_package_import('my_mackage', base='quantization')
```

would check if `torch.quantization.my_package` and `torch.ao.quantization.my_package` are the same.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/77064

Approved by: https://github.com/jerryzh168
2022-05-20 18:43:37 +00:00
..
__init__.py [quant] AO migration of the quantize.py (resubmission) (#64445) 2021-09-08 04:58:47 -07:00
common.py [quant][ao_migration] Base package in tests 2022-05-20 18:43:37 +00:00
test_ao_migration.py [quant][ao_migration] Base package in tests 2022-05-20 18:43:37 +00:00
test_quantization_fx.py [quant][fx] Move backend_config folder to torch.ao.quantization 2022-04-19 15:38:57 +00:00
test_quantization.py [quant][fx][improvement] Renamed default_affine_fixed_qparams_observer and default_symmetric_fixed_qparams_observer (#76637) 2022-05-04 02:39:20 +00:00