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 3c21e3a commit 9164e51Copy full SHA for 9164e51
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]
@@ -47,6 +48,9 @@
47
48
torch.layout,
49
)
50
51
+def get_default_skip_targets_no_quant() -> set[EdgeOpOverload]:
52
+ return _DEFAULT_SKIP_TARGETS_NO_QUANT
53
+
54
55
def is_const(
56
arg,
0 commit comments