Tugsbayasgalan (Tugsuu) Manlaibaatar 194fcfcfbd Add support for param mutation under inference mode (#159661) ...

Summary:
In HF model rwkv, we have parameter mutation under inference mode which should be safe. This PR does multiple things to make sure it works:
1. We execute global autograd mutation while tracing so that we can actually trace through parameter inplace mutation
2. Add support for parameter mutation under inference mode in AOTAutograd
3. Add support for parameter mutation under inference mode in export.

Test Plan:
test

Rollback Plan:

Differential Revision: D79460136

Pull Request resolved: https://github.com/pytorch/pytorch/pull/159661
Approved by: https://github.com/ydwu4

2025-08-14 03:34:04 +00:00

..

db

PEP585 update - torch/_export (#145138)

2025-01-19 18:48:35 +00:00

pass_infra

pyfmt lint more export files (#155783)

2025-06-12 23:04:11 +00:00

passes

[export] Fix nn_module_stack of assert_tensor_metadata nodes (#159625)

2025-08-02 02:52:42 +00:00

serde

Add support for param mutation under inference mode (#159661)

2025-08-14 03:34:04 +00:00

__init__.py

[aoti] package weights to disk and dedup (#155241)

2025-06-19 17:17:17 +00:00

converter.py

[BE][2/16] fix typos in torch/ (torch/_*/) (#156312)

2025-07-12 05:47:06 +00:00

error.py

Fix global flake8 issues (#124771)

2024-04-26 15:35:53 +00:00

non_strict_utils.py

[Dynamo][Better Engineering] Add typing annotations to guard and source (#158397) (#159491)

2025-07-30 22:57:50 +00:00

pass_base.py

[BE][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format (#144551)

2025-06-25 06:16:06 +00:00

tools.py

PEP585 update - torch/_export (#145138)

2025-01-19 18:48:35 +00:00

utils.py

[export] Fix nn_module_stack of assert_tensor_metadata nodes (#159625)

2025-08-02 02:52:42 +00:00

verifier.py

Add support for param mutation under inference mode (#159661)

2025-08-14 03:34:04 +00:00

wrappers.py

[BE][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format (#144551)

2025-06-25 06:16:06 +00:00