pytorch/torch/_dynamo/variables
Aaron Orenstein 57d8278ab9 pickler for GraphModule (#141659)
Pickling GraphModule needs some special handling for wrapping things that normally can't be pickled - but async compile needs to pass them across a wire so we need to be able to serialize it - add some helpers to enable that.

Differential Revision: [D68921318](https://our.internmc.facebook.com/intern/diff/D68921318)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/141659
Approved by: https://github.com/jamesjwu
2025-01-31 05:34:28 +00:00
..
__init__.py [dynamo][dicts] Support construction of types.MappingProxyType (#145994) 2025-01-31 00:47:31 +00:00
base.py Turn on mypy for _dynamo/variables/builtin.py (#145552) 2025-01-30 22:21:32 +00:00
builder.py Fix dynamo use of list[int] in graph break (#145554) 2025-01-30 22:21:40 +00:00
builtin.py [dynamo][functions] Support id on function (#145987) 2025-01-31 00:47:23 +00:00
constant.py Remove incorrect BuiltinVariable.call_hasattr() (#145551) 2025-01-30 22:21:19 +00:00
ctx_manager.py Revert "[dynamo] Use polyfill to implement comparison operators (#144485)" 2025-01-29 21:30:42 +00:00
dicts.py [dynamo][dicts] Support construction of types.MappingProxyType (#145994) 2025-01-31 00:47:31 +00:00
distributed.py PEP585 update - torch/_dynamo (#145105) 2025-01-18 20:47:11 +00:00
functions.py Remove incorrect BuiltinVariable.call_hasattr() (#145551) 2025-01-30 22:21:19 +00:00
higher_order_ops.py [hop] fix unbacked_bindings meta for while_loop (#143559) 2025-01-30 21:33:09 +00:00
iter.py PEP585 update - torch/_dynamo (#145105) 2025-01-18 20:47:11 +00:00
lazy.py PEP585 update - torch/_dynamo (#145105) 2025-01-18 20:47:11 +00:00
lists.py Remove incorrect BuiltinVariable.call_hasattr() (#145551) 2025-01-30 22:21:19 +00:00
misc.py pickler for GraphModule (#141659) 2025-01-31 05:34:28 +00:00
nn_module.py Remove incorrect BuiltinVariable.call_hasattr() (#145551) 2025-01-30 22:21:19 +00:00
optimizer.py PEP585 update - torch/_dynamo (#145105) 2025-01-18 20:47:11 +00:00
script_object.py PEP585 update - torch/_dynamo (#145105) 2025-01-18 20:47:11 +00:00
sdpa.py [dynamo] Simplify creation of VariableTrackers (#135714) 2024-10-18 09:36:46 +00:00
tensor.py Remove incorrect BuiltinVariable.call_hasattr() (#145551) 2025-01-30 22:21:19 +00:00
torch_function.py PEP585 update - torch/_dynamo (#145105) 2025-01-18 20:47:11 +00:00
torch.py [dynamo/export] call local_scalar_dense when full() value is scalar tensor (#144999) 2025-01-31 02:45:43 +00:00
user_defined.py [dynamo][dicts] Support construction of types.MappingProxyType (#145994) 2025-01-31 00:47:31 +00:00