From 926512fd2b86b42f4735e4435b73c95047ebb5ff Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Mon, 30 Mar 2020 14:57:19 +0100
Subject: [PATCH] [StreamingFC] check current mem depth for decoupled
---
src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py | 5 +++++
1 file changed, 5 insertions(+)
diff --git a/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py b/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py
index d63af02dc..f62c65319 100644
--- a/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py
+++ b/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py
@@ -503,6 +503,11 @@ class StreamingFCLayer_Batch(HLSCustomOp):
weight_tensor_unflipped = pack_innermost_dim_as_hex_string(
weight_tensor_unflipped, export_wdt, weight_width
)
+ weight_stream_len = np.prod(weight_tensor_unflipped.shape)
+ assert (
+ weight_stream_len <= 1024
+ ), """Decoupled mem mode needs
+ weight stream length <= 1024 for now"""
weight_pad = np.zeros((1024), int).astype(str)
weight_tensor_unflipped = weight_tensor_unflipped.flatten()
# delete "0x" in the beginning of the hexstring
--
GitLab