Ease MPS backend and partitioner imports. (pytorch#3714)

Summary:
Pull Request resolved: pytorch#3714

.

Reviewed By: kirklandsign

Differential Revision: D57709407

fbshipit-source-id: 3e680cd046d3e74b74c4ff1960442f6868661a60

backends/apple/mps/__init__.py

@@ -0,0 +1,10 @@
+#
+#  Copyright (c) 2023 Apple Inc. All rights reserved.
+#  Provided subject to the LICENSE file in the top level directory.
+#
+
+from .mps_preprocess import MPSBackend
+
+__all__ = [
+    MPSBackend,
+]

backends/apple/mps/partition/__init__.py

@@ -0,0 +1,10 @@
+#
+#  Copyright (c) 2023 Apple Inc. All rights reserved.
+#  Provided subject to the LICENSE file in the top level directory.
+#
+
+from .mps_partitioner import MPSPartitioner
+
+__all__ = [
+    MPSPartitioner,
+]

backends/apple/mps/partition/mps_partitioner.py

@@ -7,7 +7,7 @@
 from typing import Any, cast, Dict, List, Union
 
 import torch
-from executorch.backends.apple.mps.mps_preprocess import MPSBackend
+from executorch.backends.apple.mps import MPSBackend
 from executorch.backends.apple.mps.operators.node_visitor import get_node_visitors
 from executorch.backends.transforms import get_shape
 from executorch.exir.backend.backend_details import CompileSpec

backends/apple/mps/test/test_mps_utils.py

@@ -10,8 +10,8 @@
 
 import executorch.exir as exir
 import torch
-from executorch.backends.apple.mps.mps_preprocess import MPSBackend
-from executorch.backends.apple.mps.partition.mps_partitioner import MPSPartitioner
+from executorch.backends.apple.mps import MPSBackend
+from executorch.backends.apple.mps.partition import MPSPartitioner
 from executorch.exir import (
     EdgeCompileConfig,
     EdgeProgramManager,

examples/apple/mps/scripts/mps_example.py

@@ -12,8 +12,8 @@
 import torch
 from examples.apple.mps.scripts.bench_utils import bench_torch, compare_outputs
 from executorch import exir
-from executorch.backends.apple.mps.mps_preprocess import MPSBackend
-from executorch.backends.apple.mps.partition.mps_partitioner import MPSPartitioner
+from executorch.backends.apple.mps import MPSBackend
+from executorch.backends.apple.mps.partition import MPSPartitioner
 
 from executorch.exir import (
     EdgeCompileConfig,