pytorch/torch/testing
Vasiliy Kuznetsov 37622db76a ns for fx - stubs of the three APIs (compare weights, activations, activations with shadow) (#51669)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/51669

Adds the basic functionality for the three Numeric Suite core APIs to work on FX models:
1. comparing weights
2. comparing activations, with same input fed to both models
3. comparing activations, with nodes of A shadowing nodes of B

Note: there are a lot of TODOs in the code, and some/most of the APIs and implementation details may change as we iterate.  This is just the first PR.

Test Plan:
We have unit test coverage for all of the APIs, for now this is with toy models:

```
python test/test_quantization.py TestFXNumericSuiteCoreAPIs
```

Imported from OSS

Reviewed By: raghuramank100

Differential Revision: D26403094

fbshipit-source-id: 9752331d4ae0105346d3da309b13c895b593b450
2021-02-12 17:52:21 -08:00
..
_internal ns for fx - stubs of the three APIs (compare weights, activations, activations with shadow) (#51669) 2021-02-12 17:52:21 -08:00
__init__.py Warn about floor_divide performing incorrect rounding (#50281) (#50281) 2021-02-10 03:13:34 -08:00
check_kernel_launches.py Fix check_kernel_launches.py for macros and provide extended context (#49365) 2020-12-14 22:09:33 -08:00