pytorch/test/quantization/serialized
Vasiliy Kuznetsov 1d9a6862cd fx quant: add a BC test for loading old torch.package models (#65538)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/65538

Adds a test which verifies that `prepare_fx` and `convert_fx` work
on models created by `torch.package` in the past.  In detail:

1. (one time) create a model and save it with torch.package. Also save input,
expected output, and names of quantization related get_attrs added by
our passes.
2. (every time) load the model from (1), and verify that expected output
matches current output, and that get_attr targets did not change.

Test Plan:
```
python test/test_quantization.py TestSerialization.test_linear_relu_package_quantization_transforms
```

Imported from OSS

Reviewed By: supriyar

Differential Revision: D31512939

fbshipit-source-id: 718ad5fb66e09b6b31796ebe0dc698186e9a659f
2021-10-11 08:23:38 -07:00
..
TestSerialization.test_conv2d.expected.pt
TestSerialization.test_conv2d.input.pt
TestSerialization.test_conv2d.scripted.pt
TestSerialization.test_conv2d.state_dict.pt
TestSerialization.test_conv2d.traced.pt
TestSerialization.test_conv2d_graph.expected.pt
TestSerialization.test_conv2d_graph.input.pt
TestSerialization.test_conv2d_graph.scripted.pt
TestSerialization.test_conv2d_graph.traced.pt
TestSerialization.test_conv2d_graph_v2.expected.pt
TestSerialization.test_conv2d_graph_v2.input.pt
TestSerialization.test_conv2d_graph_v2.scripted.pt
TestSerialization.test_conv2d_graph_v2.traced.pt
TestSerialization.test_conv2d_graph_v3.expected.pt
TestSerialization.test_conv2d_graph_v3.input.pt
TestSerialization.test_conv2d_graph_v3.scripted.pt
TestSerialization.test_conv2d_graph_v3.traced.pt
TestSerialization.test_conv2d_nobias.expected.pt
TestSerialization.test_conv2d_nobias.input.pt
TestSerialization.test_conv2d_nobias.scripted.pt
TestSerialization.test_conv2d_nobias.state_dict.pt
TestSerialization.test_conv2d_nobias.traced.pt
TestSerialization.test_conv2d_nobias_graph.expected.pt
TestSerialization.test_conv2d_nobias_graph.input.pt
TestSerialization.test_conv2d_nobias_graph.scripted.pt
TestSerialization.test_conv2d_nobias_graph.traced.pt
TestSerialization.test_conv2d_nobias_graph_v2.expected.pt
TestSerialization.test_conv2d_nobias_graph_v2.input.pt
TestSerialization.test_conv2d_nobias_graph_v2.scripted.pt
TestSerialization.test_conv2d_nobias_graph_v2.traced.pt
TestSerialization.test_conv2d_nobias_graph_v3.expected.pt
TestSerialization.test_conv2d_nobias_graph_v3.input.pt
TestSerialization.test_conv2d_nobias_graph_v3.scripted.pt
TestSerialization.test_conv2d_nobias_graph_v3.traced.pt
TestSerialization.test_conv2d_relu.expected.pt
TestSerialization.test_conv2d_relu.input.pt
TestSerialization.test_conv2d_relu.scripted.pt
TestSerialization.test_conv2d_relu.state_dict.pt
TestSerialization.test_conv2d_relu.traced.pt
TestSerialization.test_conv3d.expected.pt
TestSerialization.test_conv3d.input.pt
TestSerialization.test_conv3d.scripted.pt
TestSerialization.test_conv3d.state_dict.pt
TestSerialization.test_conv3d.traced.pt
TestSerialization.test_conv3d_relu.expected.pt
TestSerialization.test_conv3d_relu.input.pt
TestSerialization.test_conv3d_relu.scripted.pt
TestSerialization.test_conv3d_relu.state_dict.pt
TestSerialization.test_conv3d_relu.traced.pt
TestSerialization.test_default_qat_qconfig.expected.pt [quant] Make version 1 the default for get_default_qat_qconfig (#63043) 2021-08-11 22:06:44 -07:00
TestSerialization.test_default_qat_qconfig.input.pt [quant] Make version 1 the default for get_default_qat_qconfig (#63043) 2021-08-11 22:06:44 -07:00
TestSerialization.test_default_qat_qconfig.state_dict.pt [quant] Make version 1 the default for get_default_qat_qconfig (#63043) 2021-08-11 22:06:44 -07:00
TestSerialization.test_linear.expected.pt
TestSerialization.test_linear.input.pt
TestSerialization.test_linear.scripted.pt
TestSerialization.test_linear.state_dict.pt
TestSerialization.test_linear.traced.pt
TestSerialization.test_linear_dynamic_float16.expected.pt
TestSerialization.test_linear_dynamic_float16.input.pt
TestSerialization.test_linear_dynamic_float16.scripted.pt
TestSerialization.test_linear_dynamic_float16.state_dict.pt
TestSerialization.test_linear_dynamic_float16.traced.pt
TestSerialization.test_linear_dynamic_qint8.expected.pt
TestSerialization.test_linear_dynamic_qint8.input.pt
TestSerialization.test_linear_dynamic_qint8.scripted.pt
TestSerialization.test_linear_dynamic_qint8.state_dict.pt
TestSerialization.test_linear_dynamic_qint8.traced.pt
TestSerialization.test_linear_relu.expected.pt
TestSerialization.test_linear_relu.input.pt
TestSerialization.test_linear_relu.scripted.pt
TestSerialization.test_linear_relu.state_dict.pt
TestSerialization.test_linear_relu.traced.pt
TestSerialization.test_linear_relu_package_quantization_transforms.expected.pt fx quant: add a BC test for loading old torch.package models (#65538) 2021-10-11 08:23:38 -07:00
TestSerialization.test_linear_relu_package_quantization_transforms.get_attr_targets.pt fx quant: add a BC test for loading old torch.package models (#65538) 2021-10-11 08:23:38 -07:00
TestSerialization.test_linear_relu_package_quantization_transforms.input.pt fx quant: add a BC test for loading old torch.package models (#65538) 2021-10-11 08:23:38 -07:00
TestSerialization.test_linear_relu_package_quantization_transforms.package.pt fx quant: add a BC test for loading old torch.package models (#65538) 2021-10-11 08:23:38 -07:00
TestSerialization.test_lstm.expected.pt
TestSerialization.test_lstm.input.pt
TestSerialization.test_lstm.scripted.pt
TestSerialization.test_lstm.state_dict.pt
TestSerialization.test_lstm.traced.pt
TestSerialization.test_per_channel_observer.expected.pt
TestSerialization.test_per_channel_observer.input.pt
TestSerialization.test_per_channel_observer.state_dict.pt
TestSerialization.test_per_tensor_observer.expected.pt
TestSerialization.test_per_tensor_observer.input.pt
TestSerialization.test_per_tensor_observer.state_dict.pt