James Wu e466dab164 [WIP] Automatically load and save dynamo entries via caching_precompile (#155913) ...

This PR adds a new config option, `caching_precompile`, and a `DynamoCache`, which loads and saves Dynamo Cache entries automatically. It also hooks up DynamoCache to PrecompileContext, so that we can save multiple cache entries.

When this configuration is turned on, we:
- Automatically create and initialize a CompilePackage on every torch.compile
- Automatically use BundledAutogradcache
- Automatically save the CompilePackage entry to DynamoCache after every compile

You can also use PrecompileContext.serialize() to manually serialize a full object.

I've added unit tests to exhibit this behavior.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/155913
Approved by: https://github.com/zhxchen17

2025-07-07 11:56:30 +00:00

..

__init__.py

[BE][PYFMT] migrate PYFMT for torch/[a-c]*/ to ruff format (#144554)

2025-07-03 18:56:07 +00:00

_cache.py

[WIP] Automatically load and save dynamo entries via caching_precompile (#155913)

2025-07-07 11:56:30 +00:00

config.py

Combine sticky pgo key with job id (#154863)

2025-06-03 07:58:38 +00:00