8000 [pt2] [Precompile] Store parameters in BundledAOTAutogradCacheEntry · Issue #155433 · pytorch/pytorch · GitHub
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content
[pt2] [Precompile] Store parameters in BundledAOTAutogradCacheEntry #155433
New issue
New issue
Open
Open
[pt2] [Precompile] Store parameters in BundledAOTAutogradCacheEntry#155433
Assignees
jamesjwu
Labels
actionablein progressoncall: pt2triagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module
@jamesjwu

Description

@jamesjwu
jamesjwu
opened on Jun 9, 2025

🚀 The feature, motivation and pitch

AOTDispatch has an extra wrapper here which adds flattened GraphModule parameters to the runtime arguments passed into the AOTAutograd callable.

pytorch/torch/_functorch/aot_autograd.py

Line 1229 in 2908c10

full_args.extend(params_flat)

These need to be saved by BundledAOTAutogradCacheEntry or elsewhere for precompile.

Alternatives

No response

Additional context

No response

cc @chauhang @penguinwu

Metadata

Metadata

Assignees

Labels

actionablein progressoncall: pt2triagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions
    0