Lucas Kabela ada44e5ba7 [Dynamo][Better Engineering] Add typing to bytecode analysis and transform (#158293) ...

As part of better engineering week, we would like to improve out type support to improve dev experience in dynamo

This PR adds strict typing support to a critical tracing point for dynamo, `bytecode_transformation.py` and by extension, `bytecode_analysis.py`

Running
```
mypy torch/_dynamo/bytecode_transformation.py torch/_dynamo/bytecode_analysis.py --linecount-report /tmp/coverage_log
```

| -------- | Lines Unannotated | Lines Total | % lines covered | Funcs Unannotated | Funcs Total | % funcs covered |
| -------- | ------- | -------- | ------- | ------- | ------- | ------- |
| Main  |  1422 | 1920 | 74.06% | 73 | 93 | 78.49% |
| This PR | 1968 | 1968 | 100.00% | 93 | 93 | 100.00% |
| Delta    | +546 | +48 | +25.94% | 20 | 0 | +21.51% |

Pull Request resolved: https://github.com/pytorch/pytorch/pull/158293
Approved by: https://github.com/StrongerXi, https://github.com/Skylion007

2025-07-16 21:50:55 +00:00

..

backends

[ONNX] Remove legacy Dort (#158258)

2025-07-15 19:14:06 +00:00

polyfills

[dict] Implement dict.__eq__ and dict.__ne__ (#154942)

2025-07-10 22:50:39 +00:00

repro

[dynamo] Add FakeProcessGroup support for fx_graph_runnable with distributed collectives (#157162)

2025-07-10 20:30:27 +00:00

variables

[hop][dynamo] track run-ahead sym variables in side effects (#158273)

2025-07-15 23:48:20 +00:00

__init__.py

[Log] add a hook for recompile user context (#157961)

2025-07-11 03:41:33 +00:00

_trace_wrapped_higher_order_op.py

[BE][4/16] fix typos in torch/ (torch/_dynamo/) (#156314)

2025-06-23 02:57:19 +00:00

bytecode_analysis.py

[Dynamo][Better Engineering] Add typing to bytecode analysis and transform (#158293)

2025-07-16 21:50:55 +00:00

bytecode_transformation.py

[Dynamo][Better Engineering] Add typing to bytecode analysis and transform (#158293)

2025-07-16 21:50:55 +00:00

cache_size.py

[BE][4/16] fix typos in torch/ (torch/_dynamo/) (#156314)

2025-06-23 02:57:19 +00:00

callback.py

increment pending_callbacks_counter before initation the pt2 compile callbacks (#157185)

2025-06-30 01:23:59 +00:00

code_context.py

Document dynamo (#146736)

2025-02-13 00:02:21 +00:00

codegen.py

[dynamo] trace through torch.get_device_module (#157980)

2025-07-12 06:25:46 +00:00

compiled_autograd.py

[ca] suggest to disable compiled autograd for trace-time NotImplementedErrors (#156509)

2025-06-21 18:33:46 +00:00

comptime.py

[BE][PYFMT] migrate PYFMT for torch._dynamo to ruff format (#144549)

2025-02-28 03:03:53 +00:00

config.py

Revert "[dynamo][fsdp] Consistent behavior of int attributes (#157262)"

2025-07-10 23:11:18 +00:00

convert_frame.py

[precompile] Increment frame and add compile ids when loading packages (#158028)

2025-07-15 00:53:52 +00:00

create_parameter_op.py

PEP585 update - torch/_dynamo (#145105)

2025-01-18 20:47:11 +00:00

current_scope_id.py

Document dynamo (#146736)

2025-02-13 00:02:21 +00:00

debug_utils.py

Migrate from lru_cache to cache (#155613)

2025-06-11 19:44:18 +00:00

decorators.py

Fix einops x torch.compile interaction (#157600)

2025-07-07 23:04:02 +00:00

device_interface.py

Move functions from torch._functorch.aot_autograd that are not frontend functions to frontend_utils (#158251)

2025-07-16 03:59:41 +00:00

distributed.py

Implement guard collectives (optimized version) (#156562)

2025-06-24 04:59:49 +00:00

eval_frame.py

Make guard collective logging less chatty (#157995)

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

exc.py

[dynamo] removing string literals for weblink generation (#157820)

2025-07-08 23:08:06 +00:00

external_utils.py

[dynamo] handle fullgraph toggle using nested torch.compile (#155166)

2025-06-26 21:40:38 +00:00

funcname_cache.py

gracefully handle tokenize.TokenError in funcname parser. Adds support for non-Python source (#148737)

2025-03-11 19:49:28 +00:00

graph_break_hints.py

[dynamo] unimplemented -> unimplemented_v2 in variables/builder.py (#151044)

2025-04-11 09:07:01 +00:00

graph_break_registry.json

feat(dynamo): raise UnsupportedError for ndarray.astype(object) (#157810)

2025-07-14 01:22:49 +00:00

graph_deduplication.py

[Dynamo] Fix typing in graph_deduplication.py (#152572)

2025-05-13 12:17:59 +00:00

graph_region_tracker.py

[BE][4/16] fix typos in torch/ (torch/_dynamo/) (#156314)

2025-06-23 02:57:19 +00:00

graph_utils.py

[Hierarchical Compile] Take into account mutation deps in cycle detection (#152506)

2025-05-13 12:17:59 +00:00

guards.py

[dynamo] trace through torch.get_device_module (#157980)

2025-07-12 06:25:46 +00:00

hooks.py

Replace frame_traced_fn hook with get_traced_code() util (#155249)

2025-06-10 22:40:58 +00:00

logging.py

Migrate from lru_cache to cache (#155613)

2025-06-11 19:44:18 +00:00

metrics_context.py

[logging] Fix duration logging for dynamo_compile (#151749)

2025-04-22 03:29:13 +00:00

mutation_guard.py

Add functionality for installing free variables (#151134)

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

output_graph.py

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

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

package.py

[precompile] Increment frame and add compile ids when loading packages (#158028)

2025-07-15 00:53:52 +00:00

pgo.py

[PGO] frame-specific whitelist logging (#155959)

2025-06-21 06:15:51 +00:00

precompile_context.py

Automatically load and save dynamo entries via caching_precompile (#155913)

2025-07-07 23:57:17 +00:00

profiler.py

Document dynamo (#146736)

2025-02-13 00:02:21 +00:00

replay_record.py

Document dynamo (#146736)

2025-02-13 00:02:21 +00:00

resume_execution.py

[Dynamo][Better Engineering] Add typing to bytecode analysis and transform (#158293)

2025-07-16 21:50:55 +00:00

side_effects.py

[hop][dynamo] track run-ahead sym variables in side effects (#158273)

2025-07-15 23:48:20 +00:00

source.py

[dynamo] trace through torch.get_device_module (#157980)

2025-07-12 06:25:46 +00:00

symbolic_convert.py

[Dynamo][Better Engineering] Add typing to bytecode analysis and transform (#158293)

2025-07-16 21:50:55 +00:00

tensor_version_op.py

Document dynamo (#146736)

2025-02-13 00:02:21 +00:00

test_case.py

Fixes for CPython int/float tests (#155978)

2025-07-02 15:04:00 +00:00

test_dont_skip_tracing_functions.py

[dynamo] context manager/decorator for dynamo config patching during tracing (#150586)

2025-04-23 09:12:13 +00:00

test_minifier_common.py

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

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

testing.py

[precompile] Add CompilePackage to serialize dynamo states. (#155118)

2025-06-13 13:54:10 +00:00

trace_rules.py

[dynamo] trace through torch.get_device_module (#157980)

2025-07-12 06:25:46 +00:00

types.py

Implement guard collectives (optimized version) (#156562)

2025-06-24 04:59:49 +00:00

utils.py

[Log] add a hook for recompile user context (#157961)

2025-07-11 03:41:33 +00:00