From 1c5b86df4178c78179895658905668b47e47ab22 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Mon, 17 May 2021 11:47:35 +0100
Subject: [PATCH] [Conv] check for dilation=1 differently
---
src/finn/custom_op/fpgadataflow/convolutioninputgenerator.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/src/finn/custom_op/fpgadataflow/convolutioninputgenerator.py b/src/finn/custom_op/fpgadataflow/convolutioninputgenerator.py
index dc583cf90..6e77cd3da 100644
--- a/src/finn/custom_op/fpgadataflow/convolutioninputgenerator.py
+++ b/src/finn/custom_op/fpgadataflow/convolutioninputgenerator.py
@@ -68,7 +68,7 @@ class ConvolutionInputGenerator(HLSCustomOp):
"SIMD": ("i", True, 0),
"Stride": ("ints", True, [1, 1]), # [H, W] = [Y, X]
# note: only dilation=1 supported for now
- "Dilation": ("ints", True, [1, 1], {[1, 1]}), # [H, W] = [Y, X]
+ "Dilation": ("ints", True, [1, 1]), # [H, W] = [Y, X]
# FINN DataTypes for inputs, weights, outputs
"inputDataType": ("s", True, ""),
"outputDataType": ("s", True, ""),
@@ -97,6 +97,8 @@ class ConvolutionInputGenerator(HLSCustomOp):
if name in props_to_check:
is_square = ret[0] == ret[1]
assert is_square, "Only square %s supported" % name
+ if name == "Dilation":
+ assert ret[0] == ret[1] == 1, "Only dilation=1 supported"
return ret
def get_normal_input_shape(self):
--
GitLab