We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5ad676d commit 277c39dCopy full SHA for 277c39d
exir/passes/constant_prop_pass.py
@@ -29,7 +29,8 @@
29
30
# Avoid propagating constants for `exir.ops.edge.aten.full.default`.
31
# Propagating aten.full can significantly increase compiled model size.
32
-_DEFAULT_SKIP_TARGETS = {exir_ops.edge.aten.full.default}
+_DEFAULT_SKIP_TARGETS_NO_QUANT = {exir_ops.edge.aten.full.default}
33
+_DEFAULT_SKIP_TARGETS = set(_DEFAULT_SKIP_TARGETS_NO_QUANT)
34
35
# Do not const prop quantization primitives
36
_QUANT_PRIMITIVES_EDGE = [aten_to_edge(op) for op in _QUANT_PRIMITIVES]
@@ -48,6 +49,10 @@
48
49
)
50
51
52
+def get_default_skip_targets_no_quant() -> set[EdgeOpOverload]:
53
+ return _DEFAULT_SKIP_TARGETS_NO_QUANT
54
+
55
56
def is_const(
57
arg,
58
exported_program: ExportedProgram,
0 commit comments