James Wu 4649aeaebf Make AOTAutogradCache support remote FXGraphCache (#136173) ... Summary: After the previous refactor, we can now call load_with_key directly from AOTAutogradCache to use the remote FXGraphCache. This does *not* implement a remote AOTAutogradCache. It just allows AOTAutogradCache to work with remote FXGraphCache. Test Plan: (Meta only tests) Reviewed By: aorenste Differential Revision: D62384944 Pull Request resolved: https://github.com/pytorch/pytorch/pull/136173 Approved by: https://github.com/oulgen		2024-09-23 17:24:27 +00:00
..
_aot_autograd	Make AOTAutogradCache support remote FXGraphCache (#136173)	2024-09-23 17:24:27 +00:00
__init__.py	Revert "Replace _dynamo.config with an object instead of module (#96455)"	2023-06-05 07:06:14 +00:00
aot_autograd.py	Make AOTAutogradCache support remote FXGraphCache (#136173)	2024-09-23 17:24:27 +00:00
apis.py	Revert "Deprecate torch._utils.is_compiling() and torch._dynamo.external_utils.is_compiling() (#127690)"	2024-08-07 00:05:20 +00:00
autograd_function.py	[Dynamo][autograd.Function] Support mark_non_differentiable (#134087)	2024-08-28 08:12:37 +00:00
batch_norm_replacement.py	Revert "[BE] typing for decorators - _library/custom_ops (#131578)"	2024-07-28 03:29:32 +00:00
benchmark_utils.py	[BE]: Apply ruff FURB 118. (#124743)	2024-04-26 14:34:52 +00:00
compile_utils.py	Fix unintentional deduplication of returned tensors (#134726)	2024-09-04 23:42:56 +00:00
compilers.py	[BE] UFMT directory torch/_functorch (#123723)	2024-04-12 08:04:51 +00:00
config.py	[BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765)	2024-07-31 10:42:50 +00:00
deprecated.py	Flip default value for mypy disallow_untyped_defs [2/11] (#127839)	2024-06-08 18:23:08 +00:00
eager_transforms.py	Fix linearize(grad(...)) call (#133364)	2024-08-15 17:55:36 +00:00
functional_call.py	Allow optional positional arguments for torch.func.functional_call (#134643)	2024-09-12 15:22:06 +00:00
fx_minifier.py	[BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765)	2024-07-31 10:42:50 +00:00
make_functional.py	[BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765)	2024-07-31 10:42:50 +00:00
partitioners.py	[Traceable FSDP2][Partitioner] Must save AC output if output has a backward hook (#135727)	2024-09-14 08:45:58 +00:00
pyfunctorch.py	[BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765)	2024-07-31 10:42:50 +00:00
python_key.py	[BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765)	2024-07-31 10:42:50 +00:00
pytree_hacks.py	[BE] wrap deprecated function/class with typing_extensions.deprecated (#127689)	2024-06-02 12:30:43 +00:00
top_operators_github_usage.py	[BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765)	2024-07-31 10:42:50 +00:00
utils.py	[BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765)	2024-07-31 10:42:50 +00:00
vmap.py	[BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765)	2024-07-31 10:42:50 +00:00