OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
56afad4eb3
pytorch/torch/_dynamo
History
zhxchen17 56afad4eb3 [precompile] Pickle and check closure variable properly. (#166351) 
Summary:

Previously we didn't correctly handle closure tuple when there's content in it. Adding additional code for serializing the tuple and merge it with guard manager local scope.

Test Plan:

pytest test/dynamo/test_aot_compile.py

Reviewers:

Subscribers:

Tasks:

Tags:

Fixes #ISSUE_NUMBER

Pull Request resolved: https://github.com/pytorch/pytorch/pull/166351
Approved by: https://github.com/Lucaskabela
2025-10-29 00:28:21 +00:00
..
backends Fix error suppression syntax in onnx, jit, _dynamo (#166249) 2025-10-27 02:01:54 +00:00
polyfills Fix error suppression syntax in onnx, jit, _dynamo (#166249) 2025-10-27 02:01:54 +00:00
repro Fix error suppression syntax in onnx, jit, _dynamo (#166249) 2025-10-27 02:01:54 +00:00
variables [dynamo] Move more FUNCTION_MATCH to CLOSURE_MATCH (#166444) 2025-10-28 23:19:42 +00:00
__init__.py Fix error suppression syntax in onnx, jit, _dynamo (#166249) 2025-10-27 02:01:54 +00:00
_trace_wrapped_higher_order_op.py Fix error suppression syntax in onnx, jit, _dynamo (#166249) 2025-10-27 02:01:54 +00:00
aot_compile_types.py Implement an AOT precompile mode for standalone_compile (#165843) 2025-10-21 15:02:45 +00:00
aot_compile.py [precompile] Pickle and check closure variable properly. (#166351) 2025-10-29 00:28:21 +00:00
bytecode_analysis.py Fix error suppression syntax in onnx, jit, _dynamo (#166249) 2025-10-27 02:01:54 +00:00
bytecode_transformation.py Fix error suppression syntax in onnx, jit, _dynamo (#166249) 2025-10-27 02:01:54 +00:00
cache_size.py [Dynamo][Better Engineering] Add typing for comptime, cache, and convert_frame (#158379) 2025-07-18 02:11:57 +00:00
callback.py [10/N] Apply ruff UP035 rule (#165709) 2025-10-25 00:20:13 +00:00
code_context.py
codegen.py [10/N] Apply ruff UP035 rule (#165709) 2025-10-25 00:20:13 +00:00
compiled_autograd.py [10/N] Apply ruff UP035 rule (#165709) 2025-10-25 00:20:13 +00:00
comptime.py [10/N] Apply ruff UP035 rule (#165709) 2025-10-25 00:20:13 +00:00
config.py [dynamo] Revert C++-fying of symbolic shape guards (#166427) 2025-10-28 21:57:31 +00:00
convert_frame.py [export] Update dynamo_graph_capture_for_export to return GraphModule. (#166091) 2025-10-28 04:23:28 +00:00
create_parameter_op.py Fix error suppression syntax in onnx, jit, _dynamo (#166249) 2025-10-27 02:01:54 +00:00
current_scope_id.py
debug_utils.py Fix error suppression syntax in onnx, jit, _dynamo (#166249) 2025-10-27 02:01:54 +00:00
decorators.py Fix error suppression syntax in onnx, jit, _dynamo (#166249) 2025-10-27 02:01:54 +00:00
device_interface.py Fix error suppression syntax in onnx, jit, _dynamo (#166249) 2025-10-27 02:01:54 +00:00
distributed.py
eval_frame.py Fixes torch.compile(nn.ModuleList()) changes bool() behavior (#159208) 2025-10-28 19:21:24 +00:00
exc.py Fix error suppression syntax in onnx, jit, _dynamo (#166249) 2025-10-27 02:01:54 +00:00
external_utils.py Fix error suppression syntax in onnx, jit, _dynamo (#166249) 2025-10-27 02:01:54 +00:00
funcname_cache.py
functional_export.py [export] Update dynamo_graph_capture_for_export to return GraphModule. (#166091) 2025-10-28 04:23:28 +00:00
graph_break_hints.py
graph_break_registry.json [dynamo] Clean up assert in dynamo [3/N] (#165903) 2025-10-28 22:29:35 +00:00
graph_bytecode_inputs.py [user-cuda-streams] Pass streams/events to the graph via lookup table (#162899) 2025-10-14 05:43:19 +00:00
graph_deduplication.py Fix error suppression syntax in onnx, jit, _dynamo (#166249) 2025-10-27 02:01:54 +00:00
graph_region_tracker.py Fix error suppression syntax in onnx, jit, _dynamo (#166249) 2025-10-27 02:01:54 +00:00
graph_utils.py [Intel GPU][pre_compile] Add XPU toolkit version and hardware info in compiled model check. (#162951) 2025-09-18 00:04:22 +00:00
guards.py [dynamo][easy] Use CONSTANT_MATCH for __code__ guard (#166445) 2025-10-28 23:19:42 +00:00
hooks.py [10/N] Apply ruff UP035 rule (#165709) 2025-10-25 00:20:13 +00:00
logging.py [10/N] Apply ruff UP035 rule (#165709) 2025-10-25 00:20:13 +00:00
metrics_context.py [10/N] Apply ruff UP035 rule (#165709) 2025-10-25 00:20:13 +00:00
mutation_guard.py Revert "[dynamo][executorch] Do not trace into exeuctorch LoweredBackendModule (#165126)" 2025-10-10 19:21:41 +00:00
output_graph.py [export] Update dynamo_graph_capture_for_export to return GraphModule. (#166091) 2025-10-28 04:23:28 +00:00
package.py Fix error suppression syntax in onnx, jit, _dynamo (#166249) 2025-10-27 02:01:54 +00:00
pgo.py Fix error suppression syntax in onnx, jit, _dynamo (#166249) 2025-10-27 02:01:54 +00:00
precompile_context.py [10/N] Apply ruff UP035 rule (#165709) 2025-10-25 00:20:13 +00:00
profiler.py Fix error suppression syntax in onnx, jit, _dynamo (#166249) 2025-10-27 02:01:54 +00:00
replay_record.py [Dynamo][Better Engineering] Type coverage for torch/_dynamo/utils.py (#159580) 2025-08-04 21:51:53 +00:00
resume_execution.py [dynamo] fix keyerror in resume_execution (again) (#166040) 2025-10-28 07:04:29 +00:00
side_effects.py Fix error suppression syntax in onnx, jit, _dynamo (#166249) 2025-10-27 02:01:54 +00:00
source.py [10/N] Apply ruff UP035 rule (#165709) 2025-10-25 00:20:13 +00:00
symbolic_convert.py [dynamo] fix keyerror in resume_execution (again) (#166040) 2025-10-28 07:04:29 +00:00
tensor_version_op.py [Dynamo][Better Engineering] Type devices, resume_execution and testing utils (#158593) 2025-07-18 18:22:06 +00:00
test_case.py [dynamo, nested graph breaks] add TestCaseWithNestedGraphBreaks subclass (#166013) 2025-10-28 03:03:37 +00:00
test_dont_skip_tracing_functions.py
test_minifier_common.py Pyrefly suppressions 7/n (#164913) 2025-10-08 07:27:17 +00:00
testing.py Fix error suppression syntax in onnx, jit, _dynamo (#166249) 2025-10-27 02:01:54 +00:00
trace_rules.py [10/N] Apply ruff UP035 rule (#165709) 2025-10-25 00:20:13 +00:00
types.py [10/N] Apply ruff UP035 rule (#165709) 2025-10-25 00:20:13 +00:00
utils.py [dynamo] Fix recompilation error message to point to new programming model docs (#165260) 2025-10-27 19:31:11 +00:00