From b52ec41fc4376f6691907cc82f840ee85f38eb08 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <yamanu@xilinx.com>
Date: Wed, 21 Aug 2019 14:36:36 +0000
Subject: [PATCH] also check that input is integer for *INT dtypes
---
src/finn/core/tensor.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/finn/core/tensor.py b/src/finn/core/tensor.py
index 274f66d7f..8ad238032 100644
--- a/src/finn/core/tensor.py
+++ b/src/finn/core/tensor.py
@@ -96,7 +96,7 @@ class DataType(Enum):
if "FLOAT" in self.name:
return True
elif "INT" in self.name:
- return (self.min() <= value) and (value <= self.max())
+ return (self.min() <= value) and (value <= self.max()) and float(value).is_integer()
elif self.name == "BINARY":
return value in [0, 1]
elif self.name == "BIPOLAR":
--
GitLab