From 84c7aee134652dea5b3ab79f005d5b73822527a2 Mon Sep 17 00:00:00 2001 From: Giuseppe Franco Date: Fri, 21 Apr 2023 16:07:00 +0100 Subject: [PATCH] Fix (graph_quant): add check for unsigned activation --- src/brevitas/graph/quantize_impl.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/brevitas/graph/quantize_impl.py b/src/brevitas/graph/quantize_impl.py index 6ee408db2..a57f4c226 100644 --- a/src/brevitas/graph/quantize_impl.py +++ b/src/brevitas/graph/quantize_impl.py @@ -57,7 +57,8 @@ def are_inputs_unsigned(model, node, is_unsigned_list, quant_act_map, unsigned_a for inp_node in node.all_input_nodes: if inp_node.op == 'call_module': inp_module = get_module(model, inp_node.target) - if isinstance(inp_module, tuple(quant_act_map.keys())): + if isinstance(inp_module, tuple(quant_act_map.keys())) and isinstance( + inp_module, unsigned_act_tuple): is_unsigned_list.append(True) elif isinstance(inp_module, tuple(SIGN_PRESERVING_MODULES)): are_inputs_unsigned(