Animesh Jain 801e282f39 [dynamo] Support torch.fx.traceback.annotate (#164678) ...

Builds on top of https://github.com/pytorch/pytorch/pull/163673 and https://github.com/pytorch/pytorch/pull/164174. This will be used in the followup PRs to apply regional inductor compilation.

The existing implementation let Dynamo trace into the `torch.fx.traceback.annotate`, but thats not what we want. We want Dynamo to essentially run the torch.fx.traceback.annotate function in eager, so that every Fx node created in Dynamo Fx graph has the custom meta node.

This does not work with graph breaks yet. But we can solve that problem, if needed, in a separate PR.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/164678
Approved by: https://github.com/SherlockNoMad, https://github.com/jansel, https://github.com/xmfan

2025-10-07 14:54:26 +00:00

..

cpython/3_13

Fix range.__getitem__() (#161804)

2025-09-04 02:33:03 +00:00

mock_modules

…

__init__.py

…

_test_nested_graph_breaks_helper.py

[dynamo, nested graph breaks] add nested graph break tests (#144516)

2025-08-27 03:00:56 +00:00

imports_non_circular_repro.py

[Bugfix] Fix circular import between export and dynamo from tensor fn map (#158931)

2025-07-24 22:24:57 +00:00

mock_store_global_crossfile_inline.py

[Reland] Fix inlining module-scoped store global (#132439)

2024-08-02 09:13:52 +00:00

test_activation_checkpointing.py

[dynamo][sac] Support functools partial context_fn for sac (#164308)

2025-10-01 02:47:55 +00:00

test_after_aot.py

[BE][Easy][11/19] enforce style for empty lines in import segments in test/dy*/ (#129762)

2024-07-27 17:43:53 +00:00

test_aot_autograd_cache.py

Reapply "Make functionalization ViewMeta serializable with pickle. (#143712)" (#163769)

2025-09-25 10:27:37 +00:00

test_aot_autograd.py

Support propagating custom meta field to backward graph nodes (#164174)

2025-10-04 05:03:32 +00:00

test_aot_compile.py

[precompile] Add option to disable guard check on aot-compiled function. (#163432)

2025-09-30 16:10:15 +00:00

test_autograd_function.py

Enable XPU for test_autograd_function.py (#160309)

2025-08-13 06:38:34 +00:00

test_backends.py

[Code Clean] Remove deadcodes about Python3.9 [2/N] (#163627)

2025-09-24 07:30:50 +00:00

test_backward_higher_order_ops.py

Revert "Support setting grad_dtype on leaf tensors (#162815)"

2025-10-03 23:14:28 +00:00

test_base_hop.py

Fix requires_cuda to requires_cuda_and_triton (#160222)

2025-08-10 07:05:52 +00:00

test_base_output.py

[BE][Easy][11/19] enforce style for empty lines in import segments in test/dy*/ (#129762)

2024-07-27 17:43:53 +00:00

test_buffers_override.py

Typo fixes for "overridden" in comments and function names (#155944)

2025-06-14 03:37:38 +00:00

test_bytecode_utils.py

[Code Clean] Remove deadcodes about Python3.9 [8/N] (#163728)

2025-09-25 05:12:46 +00:00

test_callback.py

Use dataclass features in two classes (#164221)

2025-10-01 03:20:39 +00:00

test_compile.py

dynamo: Handle objects in graph that do not support weakref (#163168)

2025-09-22 22:11:09 +00:00

test_compiler_bisector.py

Use torch.testing.test_close instead of torch.testing.test_allclose (#164539)

2025-10-03 14:39:10 +00:00

test_comptime.py

[aotd] Support saved tensors hooks in aot_autograd (#150032)

2025-05-22 14:09:38 +00:00

test_config.py

Rename cache limit to recompile limit in configs (#143709)

2024-12-22 10:03:57 +00:00

test_ctx_manager.py

[dynamo] Fix graph break on calling functions decorated with special context manager (#160703)

2025-08-18 20:33:45 +00:00

test_cudagraphs_expandable_segments.py

Revert "Use absolute path path.resolve() -> path.absolute() (#129409)"

2025-01-04 14:17:20 +00:00

test_cudagraphs.py

Fix unused Python variables in test/[a-d]* (#134665)

2024-12-13 22:13:12 +00:00

test_debug_utils.py

Fix requires_cuda to requires_cuda_and_triton (#160222)

2025-08-10 07:05:52 +00:00

test_decorators.py

[dynamo] change error_on_graph_break/fullgraph semantics (#161747)

2025-09-04 17:10:17 +00:00

test_deque_reconstruct.py

[dynamo] allow global import from collections import deque in user code (#148676)

2025-03-10 13:14:05 +00:00

test_deviceguard.py

[BE][Easy][11/19] enforce style for empty lines in import segments in test/dy*/ (#129762)

2024-07-27 17:43:53 +00:00

test_dicts.py

[dynamo] Correctly track mutation class source for MutableMappingVariable (#161568)

2025-08-27 21:47:17 +00:00

test_dynamic_shapes.py

fix dynamic float when dynamic=True (#149564)

2025-03-22 05:58:59 +00:00

test_einops.py

Add einops x torch.compile testing in PyTorch CI (#157416)

2025-07-03 17:36:39 +00:00

test_error_messages.py

[dynamo, 3.14] fix context managers (#163109)

2025-09-30 17:42:03 +00:00

test_exc.py

[dynamo] update pytorch-labs -> meta-pytorch in graph break URLs (#159975)

2025-08-06 23:57:31 +00:00

test_exceptions.py

[dynamo] Fix handling of kwargs in exception constructor (#163390)

2025-09-24 22:44:14 +00:00

test_export_mutations.py

[Dynamo] Replace torch._dynamo.optimize() with torch.compile() [6/N] (#140688)

2024-11-18 04:09:09 +00:00

test_export.py

remove more no longer needed torch._check_is_size calls 1 (#164630)

2025-10-04 22:06:04 +00:00

test_fake_distributed.py

[dynamo][device_mesh] Support mesh_dim_names (#164200)

2025-09-30 07:16:28 +00:00

test_flat_apply.py

[export] Add meta[val] to getattr nodes (#154934)

2025-06-13 05:48:21 +00:00

test_frame_init.py

Use dataclass features in two classes (#164221)

2025-10-01 03:20:39 +00:00

test_functions.py

[Code Clean] Remove deadcodes about Python3.9 [8/N] (#163728)

2025-09-25 05:12:46 +00:00

test_fx_annotate.py

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

2025-10-07 14:54:26 +00:00

test_fx_graph_runnable.py

Make user defined Triton kernels serializable for fx_graph_runnable (#160002)

2025-08-11 20:54:33 +00:00

test_fx_passes_pre_grad.py

[Codemod][AddExplicitStrictExportArg] caffe2/test (#143688)

2024-12-27 07:58:44 +00:00

test_generator.py

[generator] Raise StopIteration(value) with value from the return stmt (#157152)

2025-08-15 01:42:40 +00:00

test_global.py

[Dynamo] Replace torch._dynamo.optimize() with torch.compile() [2/N] (#140238)

2024-11-13 05:13:39 +00:00

test_graph_deduplication.py

[Dynamo][Hierarchical Compile] Flatten tuple inputs for regions (#158812)

2025-08-18 18:03:11 +00:00

test_graph_region_tracker.py

[ROCm] Remove HIPBLASLT_ALLOW_TF32 from codebase (#162998)

2025-09-18 13:53:48 +00:00

test_guard_manager.py

Revert "[dynamo][guards] Do not construct entire framelocals dict for LAMBDA_GUARD (#162525)"

2025-09-19 06:15:28 +00:00

test_guard_serialization.py

[precompile] Fix guard serialization loading bugs. (#164490)

2025-10-03 19:20:07 +00:00

test_higher_order_ops.py

graph break on tolist if capture_scalar_outputs is false (#163807)

2025-09-28 04:02:52 +00:00

test_hooks.py

[BE][5/6] fix typos in test/ (test/dynamo/) (#157639)

2025-07-06 06:34:25 +00:00

test_inline_and_install.py

[dynamo][export] Make the source_stack and fqn info same between dynamo and export (#164085)

2025-09-29 04:35:13 +00:00

test_input_attr_tracking.py

[BE]: Update ruff to 0.11.8 (#153249)

2025-05-12 18:30:52 +00:00

test_install_free_tensors.py

Add functionality for installing free variables (#151134)

2025-04-24 17:57:54 +00:00

test_interop.py

[dynamo] Properly handle torch.script.jit under @staticmethod (#153984)

2025-05-21 19:45:06 +00:00

test_list.py

[list] Add list.__delitem__ (#156339)

2025-07-07 14:51:32 +00:00

test_logging.py

Adjust ...mark_unbacked() -> ...decorators.mark_unbacked() in logs. (#164131)

2025-09-29 17:44:00 +00:00

test_metrics_context.py

[BE][5/6] fix typos in test/ (test/dynamo/) (#157639)

2025-07-06 06:34:25 +00:00

test_minifier.py

Port two dynamo test cases for Intel GPU (#156056)

2025-06-19 12:49:04 +00:00

test_misc.py

remove check_is_size from test_misc.py (#164667)

2025-10-07 07:33:50 +00:00

test_model_output.py

Port three dynamo test to Intel GPU (#156575)

2025-06-27 05:56:22 +00:00

test_modes.py

[dynamo] Fix TorchFunctionMode handling with get_rng_state (#163412)

2025-09-23 15:37:45 +00:00

test_modules.py

[dynamo] Trace nn.Module __delattr__ (#159969)

2025-08-06 23:43:19 +00:00

test_nested_graph_breaks.py

[dynamo, nested graph breaks] support nested graph breaks that cause skipped frames (#160470)

2025-09-03 02:47:07 +00:00

test_nops.py

[BE][Easy][11/19] enforce style for empty lines in import segments in test/dy*/ (#129762)

2024-07-27 17:43:53 +00:00

test_optimizers.py

[BE][PYFMT] migrate PYFMT for test/[a-h]*/ to ruff format (#144555)

2025-06-24 04:53:54 +00:00

test_package.py

Support partial _DynamoCacheEntries when not all backends available (#163521)

2025-10-03 16:14:32 +00:00

test_pgo.py

[PGO] ignore extra PGO key if warm/cold cache present (#163810)

2025-09-25 07:16:05 +00:00

test_pre_dispatch.py

Replace clone.detach with detach.clone (#140264)

2024-11-13 07:01:02 +00:00

test_precompile_context.py

Simplify PrecompileContext to no longer be a CacheArtifactManager (#162886)

2025-09-20 01:24:37 +00:00

test_profiler.py

[dynamo][export] Do not graph break on torch.autograd._profiler_enabled for export (#164418)

2025-10-02 09:00:00 +00:00

test_python_autograd.py

Revert "Enable all SIM rules except disabled ones (#164645)"

2025-10-05 19:32:21 +00:00

test_python_dispatcher.py

port 4 dynamo test files to Intel GPU (#157779)

2025-07-11 10:11:49 +00:00

test_recompile_ux.py

port 4 dynamo test files to Intel GPU (#157779)

2025-07-11 10:11:49 +00:00

test_recompiles.py

Add logging for when inbuilt_inline_nn_modules will help with ID_MATCH guard triggered recompiles (#160592)

2025-08-15 17:09:39 +00:00

test_reconstruct.py

[Linter] Expanding the scope of detecting device-bias code. (#159949)

2025-08-09 09:41:16 +00:00

test_reorder_logs.py

[dynamo] update pytorch-labs -> meta-pytorch in graph break URLs (#159975)

2025-08-06 23:57:31 +00:00

test_repros.py

[dynamo] Inline nn module calls using __call__ methods (#164817)

2025-10-07 08:57:20 +00:00

test_resume.py

Fix unused Python variables in test/[a-d]* (#134665)

2024-12-13 22:13:12 +00:00

test_sdpa.py

SDP Backend function fix (#161169)

2025-09-19 20:19:59 +00:00

test_sets.py

Capture TypeError in CONTAINS_OP (#161069)

2025-09-04 04:49:09 +00:00

test_skip_guard_eval_unsafe.py

[dynamo][guards] Make class members go through obj.__class__.__dict__ (#159534)

2025-08-04 05:12:44 +00:00

test_skip_non_tensor.py

Deprecate torch._utils.is_compiling() (#127690)

2024-12-08 22:55:36 +00:00

test_sources.py

[Codemod][AddExplicitStrictExportArg] caffe2/test (#143688)

2024-12-27 07:58:44 +00:00

test_structured_trace.py

Add local file path to inductor_output_code trace metadata (#160920)

2025-09-18 18:39:46 +00:00

test_subclasses.py

Revert "Enable all SIM rules except disabled ones (#164645)"

2025-10-05 19:32:21 +00:00

test_subgraphs.py

[BE][5/6] fix typos in test/ (test/dynamo/) (#157639)

2025-07-06 06:34:25 +00:00

test_torchrec.py

PEP585 update - test (#145176)

2025-01-22 04:48:28 +00:00

test_trace_rules.py

[BE][5/6] fix typos in test/ (test/dynamo/) (#157639)

2025-07-06 06:34:25 +00:00

test_unittest.py

Add infra to run CPython tests under Dynamo (#150787)

2025-05-07 04:03:14 +00:00

test_unspec.py

[fx][pass] Support converting a float32 tensor to a scalar in FX trace. (#158216)

2025-08-09 15:13:13 +00:00

test_utils.py

Log exception_stack_trace to dynamo_compile (#161096)

2025-08-22 03:29:15 +00:00

test_verify_correctness.py

[Dynamo] Replace torch._dynamo.optimize() with torch.compile() [2/N] (#140238)

2024-11-13 05:13:39 +00:00

test_view.py

Better fix for __index__ SymInt issue (#157201)

2025-07-01 07:06:46 +00:00

utils.py

[Dynamo] Introduce hook receiving list of traced code objects (#153622)

2025-05-28 15:40:09 +00:00