From 5615d8d3b89f1b11d90cc3225a2703d7e2f3e8e9 Mon Sep 17 00:00:00 2001
From: mmrahorovic <mmrahorovic@hotmail.com>
Date: Mon, 31 Jul 2023 12:25:17 +0100
Subject: [PATCH] [custom op]: set output datatype MVAU given no activation
function
---
src/finn/custom_op/fpgadataflow/matrixvectoractivation.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/src/finn/custom_op/fpgadataflow/matrixvectoractivation.py b/src/finn/custom_op/fpgadataflow/matrixvectoractivation.py
index 204a41e21..b12574570 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):
--
GitLab