Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
elad-c committed Jan 21, 2025
1 parent b811e54 commit 4a503be
Showing 1 changed file with 2 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -139,8 +139,9 @@ def _get_valid_candidates_indices(node_candidates: List[CandidateNodeQuantizatio
activation_num_bits = current_candidate.activation_quantization_cfg.activation_n_bits

# Filter candidates that have higher bit-width for both weights and activations (except for the current index).
# TODO: activation bits comparison: should be >= if ACTIVATION or TOTAL ru is used. else should be ==.
return [i for i, c in enumerate(node_candidates) if
c.activation_quantization_cfg.activation_n_bits >= activation_num_bits
c.activation_quantization_cfg.activation_n_bits == activation_num_bits
and c.weights_quantization_cfg.get_attr_config(kernel_attr).weights_n_bits >= weights_num_bits
and not (c.activation_quantization_cfg.activation_n_bits == activation_num_bits
and c.weights_quantization_cfg.get_attr_config(kernel_attr).weights_n_bits == weights_num_bits)]

0 comments on commit 4a503be

Please sign in to comment.