pytorch/torch/_dynamo/variables
PyTorch MergeBot d9ac80f80c Revert "Defer setting capturable in optimizer variable (#123497)"
This reverts commit 76b290344f.

Reverted https://github.com/pytorch/pytorch/pull/123497 on behalf of https://github.com/PaliC due to seems to have broken distributed/fsdp/test_fsdp_hybrid_shard.py as per 26bf05ccac ([comment](https://github.com/pytorch/pytorch/pull/123496#issuecomment-2043251234))
2024-04-08 17:06:05 +00:00
..
__init__.py [dynamo] Support custom __setattr__ on UserDefinedObjectVariable (#123318) 2024-04-07 21:06:52 +00:00
base.py [dynamo] Support __next__ on UserDefinedObjectVariable (#122565) 2024-03-31 19:00:03 +00:00
builder.py Swap to ID guard for optimizer Variable (#123496) 2024-04-08 05:03:34 +00:00
builtin.py [dynamo] Support custom __setattr__ on UserDefinedObjectVariable (#123318) 2024-04-07 21:06:52 +00:00
constant.py [dynamo] Remove VariableTracker.parents_tracker (#122058) 2024-03-19 04:23:24 +00:00
ctx_manager.py Teach dynamo about torch.func.jvp (#119926) 2024-03-22 20:25:47 +00:00
dicts.py [dynamo, 3.12] remove LOAD_METHOD, update LOAD_ATTR (#122356) 2024-03-27 20:39:39 +00:00
distributed.py Don't create world pg variable out of thin air when rewriting c10d collectives (#122561) 2024-03-26 20:12:08 +00:00
functions.py [dynamo] Support custom __setattr__ on UserDefinedObjectVariable (#123318) 2024-04-07 21:06:52 +00:00
higher_order_ops.py Use graph.find_nodes in dynamo (#122257) 2024-04-07 18:51:18 +00:00
iter.py [dynamo] Support __next__ on UserDefinedObjectVariable (#122565) 2024-03-31 19:00:03 +00:00
lazy.py [dynamo] Replace VariableTracker.apply with visit/realize_all (#122218) 2024-03-20 07:53:18 +00:00
lists.py [dynamo] Support __next__ on UserDefinedObjectVariable (#122565) 2024-03-31 19:00:03 +00:00
misc.py [dynamo] Support custom __setattr__ on UserDefinedObjectVariable (#123318) 2024-04-07 21:06:52 +00:00
nn_module.py [dynamo] Support custom __setattr__ on UserDefinedObjectVariable (#123318) 2024-04-07 21:06:52 +00:00
optimizer.py Revert "Defer setting capturable in optimizer variable (#123497)" 2024-04-08 17:06:05 +00:00
sdpa.py [dynamo] Refactor reconstruct() not to return anything (#120150) 2024-02-17 17:13:41 +00:00
tensor.py dynamo: support placement kwargs for DTensor.to_local() (#119947) 2024-03-22 14:42:27 +00:00
torch_function.py [dynamo] Optimize SourcelessBuilder (#122063) 2024-03-19 04:23:30 +00:00
torch.py Teach dynamo about torch.func.jvp (#119926) 2024-03-22 20:25:47 +00:00
user_defined.py [dynamo] Support custom __setattr__ on UserDefinedObjectVariable (#123318) 2024-04-07 21:06:52 +00:00