[dtensor] PART 4: move remaining DTensor ops to core distributed (#88550)

wanchaol · pytorchmergebot · commit 1b88476320a9 · 2022-11-16T08:07:44.000Z
This PR moves the view related DTensor ops to core distributed, tests will be add in follow up PRs part of #88838 Pull Request resolved: #88550 Approved by: https://github.com/fduwjj
diff --git a/torch/distributed/_tensor/__init__.py b/torch/distributed/_tensor/__init__.py
@@ -9,7 +9,7 @@
 
 
 # Import all builtin dist tensor ops
-# import torch.distributed._tensor.ops
+import torch.distributed._tensor.ops
 
 
 def distribute_tensor(
diff --git a/torch/distributed/_tensor/ops/__init__.py b/torch/distributed/_tensor/ops/__init__.py
@@ -4,4 +4,4 @@
 from .tensor_ops import *  # noqa: F403
 from .tp_sharding_ops import *  # noqa: F403
 from .pointwise_ops import *  # noqa: F403
-# from .view_ops import *  # noqa: F403
+from .view_ops import *  # noqa: F403
diff --git a/torch/distributed/_tensor/ops/view_ops.py b/torch/distributed/_tensor/ops/view_ops.py
