Remove to_server (#854)

Summary: Pull Request resolved: facebookincubator/AITemplate#854 Reviewed By: ydwu4 Differential Revision: D47810737 fbshipit-source-id: 3128cfd206b9e3b9de05de0c09dab74a5e05f5e0
pytorch · Jul 27, 2023 · b1a389c · b1a389c
1 parent d55bac7
commit b1a389c
Show file tree

Hide file tree

Showing 4 changed files with 1 addition and 25 deletions.
diff --git a/exir/__init__.py b/exir/__init__.py
@@ -6,7 +6,6 @@
     CaptureConfig,
     EdgeCompileConfig,
     ExecutorchBackendConfig,
-    ServerCompileConfig,
 )
 from executorch.exir.emit import emit_program, EmitterOutput
 from executorch.exir.program import (
@@ -44,7 +43,6 @@
     "MultiMethodExecutorchProgram",
     "CaptureConfig",
     "EdgeCompileConfig",
-    "ServerCompileConfig",
     "ExecutorchBackendConfig",
     "Value",
     "serialize_to_flatbuffer",

diff --git a/exir/capture/__init__.py b/exir/capture/__init__.py
@@ -5,7 +5,6 @@
     CaptureConfig,
     EdgeCompileConfig,
     ExecutorchBackendConfig,
-    ServerCompileConfig,
 )
 from executorch.exir.capture._unlift import unlift_exported_program_lifted_states
 
@@ -14,7 +13,6 @@
     "capture_multiple",
     "CaptureConfig",
     "EdgeCompileConfig",
-    "ServerCompileConfig",
     "ExecutorchBackendConfig",
     "unlift_exported_program_lifted_states",
 ]
diff --git a/exir/capture/_config.py b/exir/capture/_config.py
@@ -30,12 +30,6 @@ class EdgeCompileConfig:
     _use_edge_ops: bool = False
 
 
-@compatibility(is_backward_compatible=False)
-@dataclass
-class ServerCompileConfig:
-    passes: List[PassType] = field(default_factory=list)
-
-
 @compatibility(is_backward_compatible=False)
 @dataclass
 class ExecutorchBackendConfig:

diff --git a/exir/program/_program.py b/exir/program/_program.py
@@ -4,11 +4,7 @@
 
 import torch
 import torch._export
-from executorch.exir.capture._config import (
-    EdgeCompileConfig,
-    ExecutorchBackendConfig,
-    ServerCompileConfig,
-)
+from executorch.exir.capture._config import EdgeCompileConfig, ExecutorchBackendConfig
 from executorch.exir.emit import emit_program, EmitterOutput
 from executorch.exir.error import ExportError
 from executorch.exir.pass_manager import PassManager, PassType
@@ -66,16 +62,6 @@ def to_edge(
     def dump(self) -> None:
         print(self.exported_program.graph_module.graph)
 
-    def _to_server(
-        self, config: Optional[ServerCompileConfig] = None
-    ) -> torch.nn.Module:
-        config = config or ServerCompileConfig()
-        res = PassManager(config.passes)(self.exported_program.graph_module)
-        assert res is not None
-        # TODO ServerDialectGraphModule
-        # return graph_module now.
-        return res.graph_module
-
     def to_executorch(
         self,
         config: Optional[ExecutorchBackendConfig] = None,