pytorch/torch/_dynamo/variables
Yidi Wu 282f39a4bc [vmap][dynamo] use create_proxy instead of create_node in vmap increate nesting ctx manager (#165675)
create_node won't do the auto closure lifting, this cause problems when the context manager is used in a hop region. Switch to create_proxy instead.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/165675
Approved by: https://github.com/zou3519, https://github.com/guilhermeleobas
2025-10-22 09:46:00 +00:00
..
__init__.py [user-streams] Move stream code to streams module (#163027) 2025-10-14 05:43:19 +00:00
base.py [dynamo] Clean up assert in dynamo [1/N] (#165430) 2025-10-19 21:00:05 +00:00
builder.py [Bugfix][Dynamo] Fix Sparse tensors by graph break in Dynamo (#164873) 2025-10-16 15:06:20 +00:00
builtin.py [dynamo] Clean up assert in dynamo [2/N] (#165745) 2025-10-22 07:12:37 +00:00
constant.py [dynamo] Clean up assert in dynamo [2/N] (#165745) 2025-10-22 07:12:37 +00:00
ctx_manager.py [vmap][dynamo] use create_proxy instead of create_node in vmap increate nesting ctx manager (#165675) 2025-10-22 09:46:00 +00:00
dicts.py [dynamo] Clean up assert in dynamo [2/N] (#165745) 2025-10-22 07:12:37 +00:00
distributed.py Fixing get_local_rank() variable missing when compiled (#165432) 2025-10-16 18:20:34 +00:00
functions.py [dynamo] Clean up assert in dynamo [2/N] (#165745) 2025-10-22 07:12:37 +00:00
higher_order_ops.py Enable PLW0127 in ruff (#165851) 2025-10-21 03:30:57 +00:00
iter.py [dynamo, nested graph breaks] move cell codegen before side effects codegen (#160601) 2025-10-08 22:02:52 +00:00
lazy.py [Dynamo][Logging] Add sources/types to LazyVariableTracker logging (#165402) 2025-10-15 23:23:09 +00:00
lists.py [dynamo] Clean up assert in dynamo [2/N] (#165745) 2025-10-22 07:12:37 +00:00
misc.py [dynamo] Clean up assert in dynamo [2/N] (#165745) 2025-10-22 07:12:37 +00:00
nn_module.py [dynamo] Clean up assert in dynamo [2/N] (#165745) 2025-10-22 07:12:37 +00:00
optimizer.py [Dynamo] Don't guard data ptrs by default with mark_static_address (#162208) 2025-09-12 07:15:10 +00:00
script_object.py [dynamo] Replace unimplemented with unimplemented_v2 in torch/_dynamo/variables/script_object.py (#159343) 2025-08-01 21:30:41 +00:00
sdpa.py More ruff SIM fixes (#164695) 2025-10-09 03:24:50 +00:00
streams.py [RFC] Add pyrefly to lintrunner (#165179) 2025-10-16 20:07:09 +00:00
tensor.py [dynamo] Clean up assert in dynamo [2/N] (#165745) 2025-10-22 07:12:37 +00:00
torch_function.py [dynamo][remaining] Replace UserFunctionVariable with VariableTracker build (#165896) 2025-10-22 02:13:00 +00:00
torch.py [dynamo][remaining] Replace UserFunctionVariable with VariableTracker build (#165896) 2025-10-22 02:13:00 +00:00
user_defined.py [dynamo] Clean up assert in dynamo [2/N] (#165745) 2025-10-22 07:12:37 +00:00