From 13beb17789842de4dbfafe68042ccbf3c72107e9 Mon Sep 17 00:00:00 2001 From: Giuseppe Franco Date: Wed, 25 Oct 2023 17:17:07 +0100 Subject: [PATCH] Fix (ptq): correct bitwidth for layerwise int benchmark --- .../imagenet_classification/ptq/ptq_common.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/brevitas_examples/imagenet_classification/ptq/ptq_common.py b/src/brevitas_examples/imagenet_classification/ptq/ptq_common.py index 20250f7dc..3e355c503 100644 --- a/src/brevitas_examples/imagenet_classification/ptq/ptq_common.py +++ b/src/brevitas_examples/imagenet_classification/ptq/ptq_common.py @@ -199,7 +199,7 @@ def layerwise_bit_width_fn_weight(module): weight_bit_width_dict['weight_bit_width'] = layerwise_bit_width_fn_weight act_bit_width_dict['act_bit_width'] = layerwise_bit_width_fn_act - else: + elif quant_format == 'int' and backend != 'layerwise': weight_bit_width_dict['weight_bit_width'] = weight_bit_width act_bit_width_dict['act_bit_width'] = act_bit_width @@ -210,7 +210,7 @@ def layerwise_bit_width_fn_weight(module): weight_bit_width_dict['weight_exponent_bit_width'] = layerwise_bit_width_fn_weight_exponent act_bit_width_dict['act_mantissa_bit_width'] = layerwise_bit_width_fn_act_mantissa act_bit_width_dict['act_exponent_bit_width'] = layerwise_bit_width_fn_act_exponent - else: + elif quant_format == 'float' and backend != 'layerwise': weight_bit_width_dict['weight_bit_width'] = weight_bit_width act_bit_width_dict['act_bit_width'] = act_bit_width weight_bit_width_dict['weight_mantissa_bit_width'] = weight_mantissa_bit_width