Simon Fan 21697feff2 [hop] run local_map with interpreter to preserve fx_traceback annotations (#165336) ...

We have an issue when using fx_traceback.annotate and HOPs that trace joint graphs. HOPs have bodies that have already been traced by Dynamo, and after Animesh's PR, does have the annotations. But when we lower that Dynamo HOP body to aten in either pre-dispatch or post-dispatch, we need to propagate the annotations to the aten nodes.

AOTAutograd does this indirectly by piggybacking off the `PropagateUnbackedSymInts` fx.Interpreter. I'm not sure if all HOPs should be using it to trace their joints or not. This PR adds an interpreter to local_map's implementation.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/165336
Approved by: https://github.com/yushangdi

2025-10-16 02:53:17 +00:00

..

__init__.py

[dynamo][hop] Introduce Local Map HOP (#161458)

2025-09-17 09:32:38 +00:00

_invoke_quant.py

[BaseHOP] change hop(subgraph, operands) to hop(subgraph, *operands) (#146730)

2025-02-20 02:30:36 +00:00

aoti_call_delegate.py

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

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

associative_scan.py

Pyrefly suppressions 6/n (#164877)

2025-10-08 02:30:57 +00:00

auto_functionalize.py

More ruff SIM fixes (#164695)

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

base_hop.py

Pyrefly suppressions 4/n (#164615)

2025-10-06 16:14:36 +00:00

cond.py

Pyrefly suppressions 4/n (#164615)

2025-10-06 16:14:36 +00:00

effects.py

Pyrefly suppressions 4/n (#164615)

2025-10-06 16:14:36 +00:00

executorch_call_delegate.py

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

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

flat_apply.py

[5/N] Apply ruff UP035 rule (#164423)

2025-10-02 07:31:11 +00:00

flex_attention.py

fix flex attention eager bwd: more rounding (#164317)

2025-10-09 15:40:49 +00:00

foreach_map.py

[5/N] Apply ruff UP035 rule (#164423)

2025-10-02 07:31:11 +00:00

hints_wrap.py

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

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

invoke_subgraph.py

Pyrefly suppressions 4/n (#164615)

2025-10-06 16:14:36 +00:00

local_map.py

[hop] run local_map with interpreter to preserve fx_traceback annotations (#165336)

2025-10-16 02:53:17 +00:00

map.py

Pyrefly suppressions 4/n (#164615)

2025-10-06 16:14:36 +00:00

out_dtype.py

Enable _int_mm on Intel GPU (#157769)

2025-08-02 05:16:01 +00:00

partitioner.py

Enable ruff rule E721 (#165162)

2025-10-13 01:48:55 +00:00

run_const_graph.py

remove allow-untyped-defs from torch/_higher_order_ops/run_const_graph.py (#157847)

2025-07-12 15:42:12 +00:00

scan.py

Pyrefly suppressions 4/n (#164615)

2025-10-06 16:14:36 +00:00

schema.py

[2/N] Use "is" in python type comparison (#165142)

2025-10-10 15:36:44 +00:00

strict_mode.py

[5/N] Apply ruff UP035 rule (#164423)

2025-10-02 07:31:11 +00:00

torchbind.py

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

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

triton_kernel_wrap.py

Add pyrefly suppressions (#164748)

2025-10-07 17:31:18 +00:00

utils.py

Pyrefly suppressions 4/n (#164615)

2025-10-06 16:14:36 +00:00

while_loop.py

Pyrefly suppressions 6/n (#164877)

2025-10-08 02:30:57 +00:00

wrap.py

Fix pre-dispatch AC HOP calling convention (#165145)

2025-10-12 02:28:21 +00:00