Tugsbayasgalan Manlaibaatar 75ac6fdcdd Propogate dynamo shape_env to make_fx (#96437) ... Currently, when we use assume_static_by_default flag, dynamo won't produce any symbols for input tensors. But when we pass the dynamo generated graph onto make_fx via torchdynamo.export(aten_graph=True), there is no way to pass this flag. We enable this by directly passing the fake tensors dynamo used to make_fx and call make_fx with "real" mode with fake tensors from dynamo. Note that this is modified version of (https://github.com/pytorch/pytorch/pull/96143) Differential Revision: [D44561753](https://our.internmc.facebook.com/intern/diff/D44561753) Pull Request resolved: https://github.com/pytorch/pytorch/pull/96437 Approved by: https://github.com/jansel, https://github.com/ezyang		2023-04-04 20:37:30 +00:00
..
__init__.py
test_equalize_fx.py
test_model_report_fx.py	[BE] Enable flake8-comprehension rule C417 (#97880)	2023-03-30 14:34:24 +00:00
test_numeric_suite_fx.py
test_quantize_fx.py	[Quant][FX] Add test case for lowering conv_transpose with kwargs (#97311)	2023-03-31 10:39:29 +00:00
test_quantize_pt2e.py	Propogate dynamo shape_env to make_fx (#96437)	2023-04-04 20:37:30 +00:00
test_subgraph_rewriter.py