diff --git a/src/finn/custom_op/fpgadataflow/matrixvectoractivation.py b/src/finn/custom_op/fpgadataflow/matrixvectoractivation.py index 204a41e21cd3d1b92b6cebc9c44615b7a83247ca..b125745708ed55d403ef8246161bdd9a0cd0c3db 100644 --- a/src/finn/custom_op/fpgadataflow/matrixvectoractivation.py +++ b/src/finn/custom_op/fpgadataflow/matrixvectoractivation.py @@ -664,6 +664,8 @@ class MatrixVectorActivation(HLSCustomOp): # for no-activation nodes, output dt = acc dt self.set_nodeattr("outputDataType", adt.name) self.set_nodeattr("accDataType", adt.name) + if self.get_nodeattr("noActivation"): + self.set_nodeattr("outputDataType", adt.name) return DataType[self.get_nodeattr("accDataType")] def minimize_weight_bit_width(self, model):