Skip to content

Commit dcfe91b

Browse files
ThomasJannaudfacebook-github-bot
authored andcommitted
Make constant_folding's _DEFAULT_SKIP_TARGETS public
Summary: This follows D73513516 and allows customers of constant_prop_pass to use previous defaults Differential Revision: D74349918
1 parent b875a7a commit dcfe91b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

exir/passes/constant_prop_pass.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,8 @@
2929

3030
# Avoid propagating constants for `exir.ops.edge.aten.full.default`.
3131
# Propagating aten.full can significantly increase compiled model size.
32-
_DEFAULT_SKIP_TARGETS = {exir_ops.edge.aten.full.default}
32+
DEFAULT_SKIP_TARGETS_NO_QUANT = {exir_ops.edge.aten.full.default}
33+
_DEFAULT_SKIP_TARGETS = set(DEFAULT_SKIP_TARGETS_NO_QUANT)
3334

3435
# Do not const prop quantization primitives
3536
_QUANT_PRIMITIVES_EDGE = [aten_to_edge(op) for op in _QUANT_PRIMITIVES]

0 commit comments

Comments
 (0)