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
..
ao_migration	Enable UFMT on all of test/quantization/ao_migration &bc (#123994)	2024-04-13 06:36:10 +00:00
bc	Fix failures when default is flipped for weights_only (#127627)	2024-08-16 00:22:43 +00:00
core	Change wrapped_linear_prepack and wrapped_quantized_linear_prepacked to private by adding _ as prefix (#135401)	2024-09-08 04:16:24 +00:00
eager	Add None return type to init -- tests (#132352)	2024-08-01 15:44:51 +00:00
fx	Fix failures when default is flipped for weights_only (#127627)	2024-08-16 00:22:43 +00:00
jit	Add None return type to init -- tests (#132352)	2024-08-01 15:44:51 +00:00
pt2e	[Dynamo] Trace torch function modes entered outside of torch.compile (#133137)	2024-09-14 18:52:22 +00:00
serialized
__init__.py