pytorch/torch/_dynamo/variables
Wanchao Liang c76c84bde4 [dynamo] make ProcessGroupVariable a DistributedVariable (#105593)
This PR move the ProcessGroupVariable from UDO to DistributedVT
so that Distributed VTs are consolidated together

Pull Request resolved: https://github.com/pytorch/pytorch/pull/105593
Approved by: https://github.com/voznesenskym
2023-07-26 06:42:50 +00:00
..
__init__.py Revert "Revert "SetVariable in dynamo (#103205)"" + Fix for improved graph breaks (#105345) 2023-07-17 23:21:30 +00:00
base.py [HigherOrderOp] Fall back on all new side effects in speculate_subgraph (#104077) 2023-06-28 14:20:37 +00:00
builder.py [dynamo] make ProcessGroupVariable a DistributedVariable (#105593) 2023-07-26 06:42:50 +00:00
builtin.py [dynamo] add initial dynamo support for DTensor (#103146) 2023-07-19 16:01:12 +00:00
constant.py Strengthen ConstantVariable invariants (#105796) 2023-07-24 20:41:12 +00:00
ctx_manager.py [dynamo] Simulate torch function enablement state (#105091) 2023-07-13 17:42:20 +00:00
dicts.py Simulate treespec flattening/unflattening (#101896) 2023-06-23 10:53:15 +00:00
distributed.py [dynamo] make ProcessGroupVariable a DistributedVariable (#105593) 2023-07-26 06:42:50 +00:00
functions.py Initial sourceless builder (#104734) 2023-07-24 02:48:32 +00:00
higher_order_ops.py [BE]: Update Ruff to 0.0.280 (#105724) 2023-07-22 23:03:34 +00:00
lists.py [Dynamo] Support list indexed by constant tensor (#105509) 2023-07-20 20:14:04 +00:00
misc.py [BE] Enable ruff's UP rules and autoformat dynamo / functorch and refs (#105432) 2023-07-19 13:48:44 +00:00
nn_module.py Include nn.ParameterDict in dynamo __getitem__ (#99771) 2023-07-11 08:19:04 +00:00
optimizer.py Enable more e2e foreach optimizer compilation tests (#105438) 2023-07-20 02:41:19 +00:00
tensor.py [dynamo][numpy] Add support for builtin len() on numpy ndarray (#105691) 2023-07-21 03:50:40 +00:00
torch.py [dynamo] make ProcessGroupVariable a DistributedVariable (#105593) 2023-07-26 06:42:50 +00:00
user_defined.py [dynamo] make ProcessGroupVariable a DistributedVariable (#105593) 2023-07-26 06:42:50 +00:00