Ryan Guo f14247d5aa [dynamo] Accurately identify mutated cells captured by multiple functions (#138632) ...

This patch changes `mutated_closure_cell_contents: Set[str]` to
`mutated_closure_cell_ids: Set[int]` so that Dynamo can more accurately
identify closure cells across different instances of
`UserFunctionVariable`. This prevents Dynamo from mistakenly treat a
cell as immutable, despite it'll be mutated when referenced as closure
cell from another function.

More context in
https://github.com/pytorch/pytorch/issues/138112#issuecomment-2420580779.

Fixes #138112.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/138632
Approved by: https://github.com/jansel
ghstack dependencies: #138639

2024-10-26 02:17:07 +00:00

..

__init__.py

Add host-side Triton TMA support to Dynamo (#137677)

2024-10-16 02:18:48 +00:00

base.py

[dynamo] Simplify creation of VariableTrackers (#135714)

2024-10-18 09:36:46 +00:00

builder.py

Fix lru_cache where config is used (#134235)

2024-10-24 10:43:34 +00:00

builtin.py

[dynamo] Replace __str__ with __repr__ in some places (#136316)

2024-10-21 19:50:38 +00:00

constant.py

[dynamo] Replace __str__ with __repr__ in some places (#136316)

2024-10-21 19:50:38 +00:00

ctx_manager.py

[Dynamo] Trace enter/exit of TorchFunctionModes (#135422) (#137114)

2024-10-09 02:29:40 +00:00

dicts.py

Avoid calling realize() on LazyVariableTracker on reconstruct (#138495)

2024-10-25 19:01:15 +00:00

distributed.py

[compiled autograd] tls access helpers (#138061)

2024-10-22 08:03:52 +00:00

functions.py

[dynamo] Accurately identify mutated cells captured by multiple functions (#138632)

2024-10-26 02:17:07 +00:00

higher_order_ops.py

[Dynamo] Improve args in higher_order_ops [1/N] (#138799)

2024-10-25 13:55:41 +00:00

iter.py

[dynamo] Simplify creation of VariableTrackers (#135714)

2024-10-18 09:36:46 +00:00

lazy.py

[dynamo] Replace __str__ with __repr__ in some places (#136316)

2024-10-21 19:50:38 +00:00

lists.py

[dynamo] Simplify creation of VariableTrackers (#135714)

2024-10-18 09:36:46 +00:00

misc.py

[compiled autograd] tls access helpers (#138061)

2024-10-22 08:03:52 +00:00

nn_module.py

[dynamo][modules] Support overridden __call__ on nn modules (#138619)

2024-10-24 03:49:26 +00:00

optimizer.py

[dynamo] Simplify creation of VariableTrackers (#135714)

2024-10-18 09:36:46 +00:00

script_object.py

Add basic mypy annotations to dynamo (#132415)

2024-08-04 18:43:36 +00:00

sdpa.py

[dynamo] Simplify creation of VariableTrackers (#135714)

2024-10-18 09:36:46 +00:00

tensor.py

Have as_tensor always return a float64 tensor in dynamo (#138598)

2024-10-24 20:50:28 +00:00

torch_function.py

[dynamo] Simplify creation of VariableTrackers (#135714)

2024-10-18 09:36:46 +00:00

torch.py

[dynamo] Simplify creation of VariableTrackers (#135714)

2024-10-18 09:36:46 +00:00

user_defined.py

[dynamo] Replace __str__ with __repr__ in some places (#136316)

2024-10-21 19:50:38 +00:00