From 6da2da679081126f0784b10f5a124fee2a639b93 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Thu, 30 Apr 2020 22:40:02 +0100
Subject: [PATCH] [SWG] reflect hlslib restrictions in asserts
---
src/finn/custom_op/fpgadataflow/convolutioninputgenerator.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/src/finn/custom_op/fpgadataflow/convolutioninputgenerator.py b/src/finn/custom_op/fpgadataflow/convolutioninputgenerator.py
index 3c16e8dab..493d4a3f0 100644
--- a/src/finn/custom_op/fpgadataflow/convolutioninputgenerator.py
+++ b/src/finn/custom_op/fpgadataflow/convolutioninputgenerator.py
@@ -95,7 +95,8 @@ class ConvolutionInputGenerator(HLSCustomOp):
simd = self.get_nodeattr("SIMD")
pad = 0
ofm_dim = compute_conv_output_dim(ifm_dim, k, stride, pad)
- assert k * k * ifm_ch % simd == 0, "SIMD must divide sliding window size"
+ assert ifm_ch % simd == 0, "SIMD must divide IFMChannels"
+ assert k % stride == 0, "stride must divide kernel size k"
wf = int((k * k * ifm_ch) // simd)
folded_oshape = (1, ofm_dim, ofm_dim, wf, simd)
return folded_oshape
--
GitLab