From 9703c64121c7066db110d2a17a2f0b167ccd8775 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Mon, 18 May 2020 18:04:35 +0100
Subject: [PATCH] [LabelSel] assert for signedness
LabelSelect is currently broken for signed inputs
will open issue separately
---
src/finn/custom_op/fpgadataflow/labelselect_batch.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/src/finn/custom_op/fpgadataflow/labelselect_batch.py b/src/finn/custom_op/fpgadataflow/labelselect_batch.py
index 264ab51fd..7591f09d8 100644
--- a/src/finn/custom_op/fpgadataflow/labelselect_batch.py
+++ b/src/finn/custom_op/fpgadataflow/labelselect_batch.py
@@ -149,7 +149,9 @@ class LabelSelect_Batch(HLSCustomOp):
def get_input_datatype(self):
"""Returns FINN DataType of input."""
- return DataType[self.get_nodeattr("inputDataType")]
+ ret = DataType[self.get_nodeattr("inputDataType")]
+ assert ret.signed() is False, "LabelSelect is currently broken for signed inputs"
+ return ret
def get_output_datatype(self):
"""Returns FINN DataType of output."""
--
GitLab