From dc21f111de8777755b49db173a9707b576bafbf8 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Fri, 29 Nov 2019 00:50:06 +0000
Subject: [PATCH] [Util] cover the full DataType range, high=max+1
---
src/finn/core/utils.py | 6 ++----
1 file changed, 2 insertions(+), 4 deletions(-)
diff --git a/src/finn/core/utils.py b/src/finn/core/utils.py
index 05e3adf1f..6c7a1c82a 100644
--- a/src/finn/core/utils.py
+++ b/src/finn/core/utils.py
@@ -183,11 +183,9 @@ def gen_finn_dt_tensor(finn_dt, tensor_shape):
tensor_values = 2 * tensor_values - 1
elif finn_dt == DataType.BINARY:
tensor_values = np.random.randint(2, size=tensor_shape)
- elif finn_dt == DataType.TERNARY:
- tensor_values = np.random.randint(-1, high=1, size=tensor_shape)
- elif "INT" in finn_dt.name:
+ elif "INT" in finn_dt.name or finn_dt == DataType.TERNARY:
tensor_values = np.random.randint(
- finn_dt.min(), high=finn_dt.max(), size=tensor_shape
+ finn_dt.min(), high=finn_dt.max() + 1, size=tensor_shape
)
else:
raise ValueError(
--
GitLab