Michael Lazos 5c5c33ac32 [Dynamo] Trace torch function modes entered outside of torch.compile (#133137) ... This PR adds initial tracing for torch function modes. Details: In essence, this adds tracing into the torch function of modes entered outside of the torch.compile call. This does not yet support tracing enter/exit of a torch function mode/ tracing set_default_device properly using the new mode infra (this will be a very good stress test for modes). I am adding more PRs to this stack to support these. The overall plan is to support tracing enter/exit and handling graph breaks like we do other torch.* context managers. Previously landed: https://github.com/pytorch/pytorch/pull/133135 https://github.com/pytorch/pytorch/pull/133136 https://github.com/pytorch/pytorch/pull/133134 https://github.com/pytorch/pytorch/pull/133133 https://github.com/pytorch/pytorch/pull/133132 https://github.com/pytorch/pytorch/pull/133131 https://github.com/pytorch/pytorch/pull/133729 https://github.com/pytorch/pytorch/pull/133130 Pull Request resolved: https://github.com/pytorch/pytorch/pull/133137 Approved by: https://github.com/jansel, https://github.com/zou3519 ghstack dependencies: #134732		2024-09-14 18:52:22 +00:00
..
test_duplicate_dq.py	Replace capture_pre_autograd_graph with export_for_training in torch tests (#135623)	2024-09-11 19:23:08 +00:00
test_graph_utils.py
test_metadata_porting.py	[Dynamo] Trace torch function modes entered outside of torch.compile (#133137)	2024-09-14 18:52:22 +00:00
test_numeric_debugger.py	[Dynamo] Trace torch function modes entered outside of torch.compile (#133137)	2024-09-14 18:52:22 +00:00
test_quantize_pt2e.py	Fix attr check for quantization spec (#135736)	2024-09-13 23:01:22 +00:00
test_quantize_pt2e_qat.py	"Remove BLOCK_LIST" (#135729)	2024-09-12 01:22:06 +00:00
test_representation.py
test_x86inductor_quantizer.py	Replace capture_pre_autograd_graph with export_for_training in torch tests (#135623)	2024-09-11 19:23:08 +00:00
test_xnnpack_quantizer.py	[export][training ir migration] quantized_decomposed.quantize_per_tensor decomposition (#134525)	2024-09-06 07:06:06 +00:00