From 969886e36506d7f2a9559b6f9492eb7bc315357b Mon Sep 17 00:00:00 2001 From: Giuseppe Franco Date: Mon, 4 Mar 2024 21:06:23 +0000 Subject: [PATCH] fixes --- src/brevitas/export/onnx/standard/qoperator/handler/act.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/brevitas/export/onnx/standard/qoperator/handler/act.py b/src/brevitas/export/onnx/standard/qoperator/handler/act.py index 418c96b0a..7a76bd990 100644 --- a/src/brevitas/export/onnx/standard/qoperator/handler/act.py +++ b/src/brevitas/export/onnx/standard/qoperator/handler/act.py @@ -20,9 +20,9 @@ class StdQOpONNXQuantNLALHandler(StdQOpONNXQuantLayerHandler, ABC): @classmethod def validate(cls, module: QuantNLAL): - if cls.input_quant_supported and module.is_input_quant_enabled: + if cls.input_quant_supported and module.input_quant.is_quant_enabled: assert not module.input_quant.is_quant_enabled, "Narrow range quant not supported." - elif not cls.input_quant_supported and module.is_input_quant_enabled: + elif not cls.input_quant_supported and module.input_quant.is_quant_enabled: raise RuntimeError("Input quant not supported.") if module.act_quant.is_quant_enabled: assert not module.act_quant.is_narrow_range, "Narrow range quant not supported."