Skip to content

Commit ce1869f

Browse files
committed
Fix PR comments
1 parent 54b7306 commit ce1869f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

model_compression_toolkit/core/common/mixed_precision/resource_utilization_tools/resource_utilization_calculator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -682,7 +682,7 @@ def _get_target_activation_nodes(self,
682682
if target_criterion == TargetInclusionCriterion.QConfigurable:
683683
nodes = [n for n in nodes if n.has_configurable_activation()]
684684
elif target_criterion == TargetInclusionCriterion.AnyQuantized:
685-
nodes = [n for n in nodes if n.is_activation_quantization_enabled()]
685+
nodes = [n for n in nodes if n.is_activation_quantization_enabled() or n.is_quantization_preserving()]
686686
elif target_criterion == TargetInclusionCriterion.QNonConfigurable:
687687
nodes = [n for n in nodes if n.is_activation_quantization_enabled() and not n.has_configurable_activation()]
688688
elif target_criterion != TargetInclusionCriterion.Any: # pragma: no cover

0 commit comments

Comments
 (0)