pytorch/torch/_dynamo
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
..
backends [dynamo] Maintainable code - Move decorators in a separate file (#105070) 2023-07-13 07:41:19 +00:00
repro [dynamo][inductor] Provide public API to get compiler options/configs (#105026) 2023-07-18 06:12:06 +00:00
variables [dynamo] make ProcessGroupVariable a DistributedVariable (#105593) 2023-07-26 06:42:50 +00:00
__init__.py Compiled autograd (#103822) 2023-07-24 21:12:05 +00:00
allowed_functions.py [dynamo] add initial dynamo support for DTensor (#103146) 2023-07-19 16:01:12 +00:00
bytecode_analysis.py consider CALL_FINALLY non-jumping in stacksize_analysis (#103621) 2023-06-29 20:23:20 +00:00
bytecode_transformation.py [dynamo] fine-grained bytecode-source attribution in python 3.11 (#104676) 2023-07-20 17:18:52 +00:00
codegen.py [dynamo] Support wrapping + returning tensor subclasses (#104802) 2023-07-09 22:16:10 +00:00
compiled_autograd.py Compiled autograd (#103822) 2023-07-24 21:12:05 +00:00
comptime.py Pretty dataclass dynamo explain (#102869) 2023-06-07 22:38:57 +00:00
config_utils.py [dynamo][inductor] Provide public API to get compiler options/configs (#105026) 2023-07-18 06:12:06 +00:00
config.py Revert "Correct dynamo logging docs (#105658)" 2023-07-21 22:38:28 +00:00
convert_frame.py Supress eager fallback suggestions when exporting (#105767) 2023-07-22 19:17:08 +00:00
current_scope_id.py [HigherOrderOp] Fall back on all new side effects in speculate_subgraph (#104077) 2023-06-28 14:20:37 +00:00
debug_utils.py [BE] Enable ruff's UP rules and autoformat dynamo / functorch and refs (#105432) 2023-07-19 13:48:44 +00:00
decorators.py [dynamo] Reland Move decorators into decorators.py (#105273) 2023-07-16 05:33:54 +00:00
eval_frame.py Tweak dynamic=False behavior (#105715) 2023-07-24 16:56:41 +00:00
exc.py Supress eager fallback suggestions when exporting (#105767) 2023-07-22 19:17:08 +00:00
external_utils.py Compiled autograd (#103822) 2023-07-24 21:12:05 +00:00
guards.py [BE]: Update Ruff to 0.0.280 (#105724) 2023-07-22 23:03:34 +00:00
hooks.py
logging.py Fix flake8 lint errors - part 2 - manual fixes (#99799) 2023-04-24 06:03:26 +00:00
mutation_guard.py [Dynamo] Fix constructing lazy submodule inside of lazy module's initialize_parameters (#100047) 2023-04-26 23:36:31 +00:00
output_graph.py Add telemetry for number of nodes being compiled (#105741) 2023-07-22 01:56:02 +00:00
profiler.py Don't test dynamic_shapes in profiler (#103516) 2023-06-13 17:47:25 +00:00
replay_record.py [inductor] Refactor RNG operators (#100064) 2023-05-20 03:43:33 +00:00
resume_execution.py [dynamo] fine-grained bytecode-source attribution in python 3.11 (#104676) 2023-07-20 17:18:52 +00:00
side_effects.py [HigherOrderOp] Fall back on all new side effects in speculate_subgraph (#104077) 2023-06-28 14:20:37 +00:00
skipfiles.py Revert "Disable torchrec/sparse from top-level Dynamo tracing (#105733)" 2023-07-26 03:44:47 +00:00
source.py [dynamo] Simulate torch function enablement state (#105091) 2023-07-13 17:42:20 +00:00
symbolic_convert.py [dynamo] fine-grained bytecode-source attribution in python 3.11 (#104676) 2023-07-20 17:18:52 +00:00
test_case.py Enable default workflow PyT 2.0 UTs on ROCm stack (#100981) 2023-05-15 23:45:04 +00:00
test_minifier_common.py [BE] Enable ruff's UP rules and autoformat dynamo / functorch and refs (#105432) 2023-07-19 13:48:44 +00:00
testing.py Enable automatic_dynamic_shapes by default (#103623) 2023-07-05 00:25:02 +00:00
types.py Test and fix guard fail message in CompileProfiler (#97055) 2023-03-22 02:17:57 +00:00
utils.py [BE]: Update Ruff to 0.0.280 (#105724) 2023-07-22 23:03:34 +00:00