Wanchao Liang 2ee6b97464 [dtensor] move DTensor to public namespace (#133113) ... Moving DTensor to be in the public namespace, to formally add the documentation page that includes all the public APIs. This includes: * many path renames and path import fixes * a dedicated doc page without too much content yet (adding in the next PRs) * To preserve the BC for users still using the `torch.distributed._tensor`, I added a shim script to redirect old path calls to the new module The BC preserving is evidented by the fact that all DTensor tests are still working without changing the public imports. So it's safe to land the changes Pull Request resolved: https://github.com/pytorch/pytorch/pull/133113 Approved by: https://github.com/XilunWu ghstack dependencies: #133305, #133306		2024-08-17 05:09:52 +00:00
..
backends	Make debugging backends accept and ignore options kwargs from torch.compile (#132892)	2024-08-09 00:49:45 +00:00
repro	[BE][Easy] enable ruff rule PIE790: unnecessary pass statement (#133200)	2024-08-15 15:50:19 +00:00
variables	[dtensor] move DTensor to public namespace (#133113)	2024-08-17 05:09:52 +00:00
__init__.py	[Traceable FSDP2][Dynamo] Support reconstructing CUDA event object within Dynamo graph (#133635)	2024-08-16 20:40:46 +00:00
_trace_wrapped_higher_order_op.py	[BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767)	2024-07-31 21:18:11 +00:00
bytecode_analysis.py	[BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767)	2024-07-31 21:18:11 +00:00
bytecode_transformation.py	[3.13, dynamo] decompose fused load/store instructions (#130569)	2024-07-22 18:07:40 +00:00
cache_size.py	[BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767)	2024-07-31 21:18:11 +00:00
callback.py	Flip default value for mypy disallow_untyped_defs [1/11] (#127838)	2024-06-08 18:16:33 +00:00
code_context.py	Add None return type to init (#132335)	2024-08-01 15:26:45 +00:00
codegen.py	Add basic mypy annotations to dynamo (#132415)	2024-08-04 18:43:36 +00:00
compiled_autograd.py	[compiled autograd] move non-hot path logs into default logger (#133541)	2024-08-17 00:46:52 +00:00
comptime.py	Add comptime.sleep (#133362)	2024-08-14 19:32:18 +00:00
config.py	[Traceable FSDP2] Set torch._dynamo.config.skip_fsdp_hooks to True by default (#133531)	2024-08-16 17:18:42 +00:00
convert_frame.py	Consolidate SymDispatchMode into ProxyTensorMode (#132674)	2024-08-08 12:02:54 +00:00
create_parameter_op.py	[BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767)	2024-07-31 21:18:11 +00:00
current_scope_id.py	[BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767)	2024-07-31 21:18:11 +00:00
debug_utils.py	[BE][Easy] enable ruff rule PIE790: unnecessary pass statement (#133200)	2024-08-15 15:50:19 +00:00
decorators.py	[BE][Easy] enable ruff rule PIE790: unnecessary pass statement (#133200)	2024-08-15 15:50:19 +00:00
device_interface.py	Add basic mypy annotations to dynamo (#132415)	2024-08-04 18:43:36 +00:00
distributed.py	[BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767)	2024-07-31 21:18:11 +00:00
eval_frame.py	[export] refactor _process_dynamic_shapes (#133391)	2024-08-15 16:21:21 +00:00
exc.py	[dynamo][user_defined][stable-diffusion] Raise ObservedAttributeError on UserDefinedObject var_getattr (#132806)	2024-08-16 04:30:06 +00:00
external_utils.py	Revert "Deprecate torch._utils.is_compiling() and torch._dynamo.external_utils.is_compiling() (#127690)"	2024-08-07 00:05:20 +00:00
funcname_cache.py	[BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767)	2024-07-31 21:18:11 +00:00
guards.py	[dtensor] move DTensor to public namespace (#133113)	2024-08-17 05:09:52 +00:00
hooks.py	[BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767)	2024-07-31 21:18:11 +00:00
logging.py	[BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767)	2024-07-31 21:18:11 +00:00
mutation_guard.py	[BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767)	2024-07-31 21:18:11 +00:00
output_graph.py	Only thunkify proxies in some situations (#132421)	2024-08-08 12:03:06 +00:00
polyfill.py	[dynamo] Support object creation of classes with custom __new__ (#132977)	2024-08-16 03:09:23 +00:00
profiler.py	Add basic mypy annotations to dynamo (#132415)	2024-08-04 18:43:36 +00:00
replay_record.py	[BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767)	2024-07-31 21:18:11 +00:00
resume_execution.py	[BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767)	2024-07-31 21:18:11 +00:00
side_effects.py	Improve side effects error message (#132223)	2024-07-31 17:29:26 +00:00
source.py	Revert #132806, #132736, #132539, #132487 (#133570)	2024-08-15 20:54:21 +00:00
symbolic_convert.py	[dynamo][user_defined][stable-diffusion] Raise ObservedAttributeError on UserDefinedObject var_getattr (#132806)	2024-08-16 04:30:06 +00:00
tensor_version_op.py	[BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767)	2024-07-31 21:18:11 +00:00
test_case.py	[BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767)	2024-07-31 21:18:11 +00:00
test_minifier_common.py	Flip default value for mypy disallow_untyped_defs [1/11] (#127838)	2024-06-08 18:16:33 +00:00
testing.py	[BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767)	2024-07-31 21:18:11 +00:00
trace_rules.py	[dtensor] move DTensor to public namespace (#133113)	2024-08-17 05:09:52 +00:00
types.py	Populate submodules of torch._C to sys.modules recursively (#132216)	2024-08-08 10:20:25 +00:00
utils.py	[Traceable FSDP2][Dynamo] Support reconstructing CUDA event object within Dynamo graph (#133635)	2024-08-16 20:40:46 +00:00