Skip to content

Commit a375b6d

Browse files
fix
1 parent 9dd77de commit a375b6d

File tree

1 file changed

+0
-3
lines changed
  • optimum/fx/parallelization/backend

1 file changed

+0
-3
lines changed

optimum/fx/parallelization/backend/base.py

-3
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,6 @@ def create_parallel_cross_entropy(
9999
else:
100100
return sharded_cross_entropy_wrapper_fn(process_group=parallel_ctx.tp_group)
101101

102-
@abstractmethod
103102
def pre_process(self, graph_module: GraphModule, ctx: "ParallelExecutionCtx", config: "Config") -> GraphModule:
104103
"""
105104
Mark tie information right before we run passes because dynamo tracing will alter the parameter name while our
@@ -115,15 +114,13 @@ def pre_process(self, graph_module: GraphModule, ctx: "ParallelExecutionCtx", co
115114
param_meta.tied_to = parameter_mp[key]
116115
return graph_module
117116

118-
@abstractmethod
119117
def post_process(self, graph_module: GraphModule, ctx: "ParallelExecutionCtx", config: "Config") -> nn.Module:
120118
"""
121119
This method is called after the parallelization passes have been applied. It is used to perform any backend-specific
122120
post-processing on the graph module.
123121
"""
124122
return graph_module
125123

126-
@abstractmethod
127124
def init_parallelization_pass_pipeline(self) -> PassPipeline:
128125
"""
129126
Ensemble a pass pipeline which contains the following passes:

0 commit comments

Comments
 (0)