pytorch/test/export
Tugsbayasgalan (Tugsuu) Manlaibaatar a44c072c89 Make InternalModel and Resnet work with rexportable flow (#106676)
Summary: Internal model and Resnet uses "re-export" flow now. Also did some refactoring to make the code little cleaner

Some changes for OSS:
1. Correctly use the "cached" fake tensors so that static symbols are still resolved to static
2. Change logic in PassBase to allocate static shapes for parameters
3. Add "is_torch_exported" tag to every node to make it survive during various graph transformations.
4. Added experimental wrapper API for quantization team to get pre_dispatch=True graph. Note that it doesn't actually do that right now. But we plan to switch soon.

Test Plan: CI

Differential Revision: D47890878

Pull Request resolved: https://github.com/pytorch/pytorch/pull/106676
Approved by: https://github.com/jerryzh168
2023-08-09 20:10:48 +00:00
..
test_db.py [export] add kwargs support for export. (#105337) 2023-07-20 19:53:08 +00:00
test_export.py Make InternalModel and Resnet work with rexportable flow (#106676) 2023-08-09 20:10:48 +00:00
test_funtionalized_assertions.py [RFC] Create functional aten assertion ops (#103751) 2023-06-23 06:20:42 +00:00
test_pass_infra.py [export] cleanup pass base. [1/n] (#106480) 2023-08-03 19:48:05 +00:00
test_passes.py [export] Remove experimental runtime assertion configs from export API. (#105043) 2023-07-26 16:21:29 +00:00
test_serialize.py [export] Remove experimental runtime assertion configs from export API. (#105043) 2023-07-26 16:21:29 +00:00
test_upgrade.py [export] Remove experimental runtime assertion configs from export API. (#105043) 2023-07-26 16:21:29 +00:00
test_verifier.py Back out "Reland "Make adding buffers more like adding parameters (#104069)" (#106224)" (#106743) 2023-08-08 15:27:34 +00:00