From 334e661ec029dbba76d69ac5d43b6ebfaffe53e6 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <yamanu@xilinx.com>
Date: Mon, 11 Oct 2021 14:43:50 +0200
Subject: [PATCH] [Refactor] use get_accumulator_dt_cands as part of DataType
rf.
---
src/finn/custom_op/fpgadataflow/channelwise_op_batch.py | 2 +-
src/finn/transformation/fpgadataflow/convert_to_hls_layers.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/finn/custom_op/fpgadataflow/channelwise_op_batch.py b/src/finn/custom_op/fpgadataflow/channelwise_op_batch.py
index ad91013a2..3cd6a7dfd 100644
--- a/src/finn/custom_op/fpgadataflow/channelwise_op_batch.py
+++ b/src/finn/custom_op/fpgadataflow/channelwise_op_batch.py
@@ -56,7 +56,7 @@ def get_smallest_possible(vals):
for v in vals:
assert int(v) == v, "Error float value"
- for k in DataType.__members__:
+ for k in DataType.get_accumulator_dt_cands():
dt = DataType[k]
if dt in [DataType["BIPOLAR"], DataType["TERNARY"], DataType["FLOAT32"]]:
diff --git a/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py b/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py
index 4ab9c8fe2..8ac3a705b 100644
--- a/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py
+++ b/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py
@@ -1251,7 +1251,7 @@ class InferChannelwiseLinearLayer(Transformation):
for v in vals:
assert int(v) == v, "Error float value"
- for k in DataType.__members__:
+ for k in DataType.get_accumulator_dt_cands():
dt = DataType[k]
if dt in [DataType["BIPOLAR"], DataType["TERNARY"], DataType["FLOAT32"]]:
--
GitLab