Skip to content

Commit 388e93c

Browse files
authored
Make lower_ep_to<edge, cadence, executorch> private functions and remove external call sites
Differential Revision: D74089120 Pull Request resolved: #10671
1 parent 58a3d4c commit 388e93c

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

backends/cadence/aot/compiler.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -199,7 +199,7 @@ def export_program(
199199
return expo_program
200200

201201

202-
def lower_ep_to_edge(
202+
def _lower_ep_to_edge(
203203
expo_program: ExportedProgram,
204204
dump_graphs: bool = False,
205205
constant_methods: Optional[dict[str, object]] = None,
@@ -250,7 +250,7 @@ def export_to_edge(
250250
expo_program = export_program(model, inputs)
251251

252252
# Lower the model to edge IR.
253-
edge_prog_manager = lower_ep_to_edge(expo_program, dump_graphs, constant_methods)
253+
edge_prog_manager = _lower_ep_to_edge(expo_program, dump_graphs, constant_methods)
254254

255255
return edge_prog_manager
256256

@@ -272,22 +272,22 @@ def quantize_and_export_to_edge(
272272
dump_graphs=dump_graphs,
273273
)
274274

275-
return lower_ep_to_edge(
275+
return _lower_ep_to_edge(
276276
quantized_model,
277277
dump_graphs=dump_graphs,
278278
constant_methods=constant_methods,
279279
)
280280

281281

282-
def lower_ep_to_cadence(
282+
def _lower_ep_to_cadence(
283283
program: ExportedProgram,
284284
dump_graphs: bool = False,
285285
opt_level: int = 1,
286286
) -> EdgeProgramManager:
287287
"""
288288
Lower an existing ExportedProgram to edge IR and apply frontend optimization passes.
289289
"""
290-
edge_prog_manager = lower_ep_to_edge(program, dump_graphs=dump_graphs)
290+
edge_prog_manager = _lower_ep_to_edge(program, dump_graphs=dump_graphs)
291291
cadence_passes = get_cadence_passes(opt_level)
292292

293293
# Run a couple required passes for quant/dequant ops
@@ -329,7 +329,7 @@ def quantize_and_export_to_cadence(
329329
"""
330330
quantized_model = quantize_pt2(model, inputs)
331331

332-
return lower_ep_to_cadence(
332+
return _lower_ep_to_cadence(
333333
quantized_model,
334334
opt_level=opt_level,
335335
dump_graphs=dump_graphs,

backends/cadence/aot/tests/test_remove_ops_passes.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -511,7 +511,7 @@ def forward(self, x: torch.Tensor):
511511
inputs = (x,)
512512

513513
graph_module = (
514-
compiler.export_to_edge(
514+
export_to_edge(
515515
model,
516516
inputs,
517517
)

0 commit comments

Comments
 (0)