From c478189f678e1cc359f2320856494602198002dc Mon Sep 17 00:00:00 2001 From: Thomas Jannaud Date: Thu, 8 May 2025 12:24:16 -0700 Subject: [PATCH] Make constant_folding's _DEFAULT_SKIP_TARGETS public (#10760) Summary: This follows D73513516 and allows customers of constant_prop_pass to use previous defaults This goes with D74350331 and D74280168 Reviewed By: JacobSzwejbka Differential Revision: D74349918 --- exir/passes/constant_prop_pass.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/exir/passes/constant_prop_pass.py b/exir/passes/constant_prop_pass.py index 0049e597f8d..6921bd632f4 100644 --- a/exir/passes/constant_prop_pass.py +++ b/exir/passes/constant_prop_pass.py @@ -29,7 +29,8 @@ # Avoid propagating constants for `exir.ops.edge.aten.full.default`. # Propagating aten.full can significantly increase compiled model size. -_DEFAULT_SKIP_TARGETS = {exir_ops.edge.aten.full.default} +_DEFAULT_SKIP_TARGETS_NO_QUANT = {exir_ops.edge.aten.full.default} +_DEFAULT_SKIP_TARGETS = set(_DEFAULT_SKIP_TARGETS_NO_QUANT) # Do not const prop quantization primitives _QUANT_PRIMITIVES_EDGE = [aten_to_edge(op) for op in _QUANT_PRIMITIVES] @@ -48,6 +49,10 @@ ) +def get_default_skip_targets_no_quant() -> set[EdgeOpOverload]: + return _DEFAULT_SKIP_TARGETS_NO_QUANT + + def is_const( arg, exported_program: ExportedProgram,