From 57311918a40db84521c4d592c5f3c59406a5ec7e Mon Sep 17 00:00:00 2001
From: Tobi-Alonso <tobi.alonso@gmail.com>
Date: Wed, 24 Jun 2020 14:00:08 +0100
Subject: [PATCH] [FPGADataflow] Add check to InferConvInpGen to avoid
inference is input is not integer
---
src/finn/transformation/fpgadataflow/convert_to_hls_layers.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py b/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py
index 06eb5f210..62514ee93 100644
--- a/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py
+++ b/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py
@@ -35,6 +35,7 @@ from finn.transformation.infer_shapes import InferShapes
from finn.transformation.infer_datatypes import InferDataTypes
import finn.core.data_layout as DataLayout
from finn.util.basic import get_by_name
+import warnings
class InferConvInpGen(Transformation):
@@ -52,6 +53,9 @@ class InferConvInpGen(Transformation):
i2c_in_shape = model.get_tensor_shape(i2c_input)
i2c_out_shape = model.get_tensor_shape(i2c_output)
dt = model.get_tensor_datatype(i2c_input)
+ if not dt.is_integer():
+ warnings.warn("Input is not int. Can't infer ConvInpGen")
+ continue
i2c_inst = getCustomOp(n)
stride = i2c_inst.get_nodeattr("stride")
k = i2c_inst.get_nodeattr("kernel_size")
--
GitLab