Michael Lazos 228760b945 [Dynamo] Use custom backend to reenter metadata tf mode when tracing while/cond (#134732) ... For tracing cond/while in eager, we trace the HOP with the eager backend with metadata torchfunction mode enabled. HOPs disallow the mutation that occurs in this torch function mode, so it is not able to be traced. As a result, we use a custom backend which enters this mode for tracing these HOPs. Thanks to @ydwu4 for the help with implementing this Pull Request resolved: https://github.com/pytorch/pytorch/pull/134732 Approved by: https://github.com/ydwu4		2024-09-14 18:52:22 +00:00
..
__init__.py
common.py	Make debugging backends accept and ignore options kwargs from torch.compile (#132892)	2024-08-09 00:49:45 +00:00
cudagraphs.py	[BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767)	2024-07-31 21:18:11 +00:00
debugging.py	[Dynamo] Use custom backend to reenter metadata tf mode when tracing while/cond (#134732)	2024-09-14 18:52:22 +00:00
distributed.py	Add basic mypy annotations to dynamo (#132415)	2024-08-04 18:43:36 +00:00
inductor.py	[inductor] enable inductor backend for dynamo on Windows. (#133921)	2024-08-20 16:46:19 +00:00
onnxrt.py	[BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767)	2024-07-31 21:18:11 +00:00
registry.py	register all entry_point backends on first attempt (#132546)	2024-08-29 18:59:29 +00:00
tensorrt.py
torchxla.py	[BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767)	2024-07-31 21:18:11 +00:00
tvm.py	[BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767)	2024-07-31 21:18:11 +00:00