Simon Fan 25d4d5107e [dynamo] trace local_map with local shapes for AP (#163602) ...

Context is in https://www.internalfb.com/excalidraw/EX519691 and https://docs.google.com/document/d/1qnuXLZk_GYt_PksHTwkn7L2ELRDnYlIRPkHAlXTyuhw/edit?tab=t.0. And the description of the previous PR: https://github.com/pytorch/pytorch/pull/164340.

The previous PR adds the support on the HOP side for eager execution and AOTAutograd. Dynamo is still passing the HOP a subgraph with wrong shapes. This PR fixes that. This is similar to the HOP implementation, however we additionally need to manually keep the TensorVariable metadata in sync.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/163602
Approved by: https://github.com/ydwu4
ghstack dependencies: #164296, #164321, #164419, #164420, #164340

2025-10-10 02:34:27 +00:00

..

__init__.py

[dynamo] Support torch.fx.traceback.annotate (#164678)

2025-10-08 22:41:00 +00:00

base.py

[dynamo][guards] More small guard optimizations (#159345)

2025-07-29 18:36:49 +00:00

builder.py

[export] Preserve nn_module_stack for aliased nn modules (#164311)

2025-10-01 18:04:33 +00:00

builtin.py

Pyrefly suppressions 7/n (#164913)

2025-10-08 07:27:17 +00:00

constant.py

More ruff SIM fixes (#164695)

2025-10-09 03:24:50 +00:00

ctx_manager.py

[dynamo] Support torch.fx.traceback.annotate (#164678)

2025-10-08 22:41:00 +00:00

dicts.py

Realize LazyVariableTracker before raising exception (#163350)

2025-09-19 19:25:17 +00:00

distributed.py

Fix replacement reconstruct (#164937)

2025-10-09 15:31:23 +00:00

functions.py

[dynamo, nested graph breaks] fix nested step graph break related issues (#162737)

2025-10-08 22:02:52 +00:00

higher_order_ops.py

[dynamo] trace local_map with local shapes for AP (#163602)

2025-10-10 02:34:27 +00:00

iter.py

[dynamo, nested graph breaks] move cell codegen before side effects codegen (#160601)

2025-10-08 22:02:52 +00:00

lazy.py

[dynamo] Avoid recompiling over unused objects (#156891)

2025-07-09 20:14:34 +00:00

lists.py

More ruff SIM fixes (#164695)

2025-10-09 03:24:50 +00:00

misc.py

More ruff SIM fixes (#164695)

2025-10-09 03:24:50 +00:00

nn_module.py

More ruff SIM fixes (#164695)

2025-10-09 03:24:50 +00:00

optimizer.py

[Dynamo] Don't guard data ptrs by default with mark_static_address (#162208)

2025-09-12 07:15:10 +00:00

script_object.py

[dynamo] Replace unimplemented with unimplemented_v2 in torch/_dynamo/variables/script_object.py (#159343)

2025-08-01 21:30:41 +00:00

sdpa.py

More ruff SIM fixes (#164695)

2025-10-09 03:24:50 +00:00

tensor.py

[dynamo] trace local_map with local shapes for AP (#163602)

2025-10-10 02:34:27 +00:00

torch_function.py

[dynamo] Be consistent with UserMethodVariable source (#160155)

2025-08-09 04:16:14 +00:00

torch.py

[export][dynamo] Fallback to slowpath for MultiHeadAttention for strict export (#164721)

2025-10-09 03:25:15 +00:00

user_defined.py

NamedTuple: Allow side effects for dynamic attributes (#161645)

2025-09-09 19:42:02 +00:00