From 715da32c65fe8eb984b31eaca873ecfabc55578b Mon Sep 17 00:00:00 2001
From: Lucian Petrica <lucianp@xilinx.com>
Date: Mon, 29 Jun 2020 16:19:06 +0000
Subject: [PATCH] relaxed check for external weights - mem_mode can be unset
---
src/finn/transformation/fpgadataflow/insert_iodma.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/src/finn/transformation/fpgadataflow/insert_iodma.py b/src/finn/transformation/fpgadataflow/insert_iodma.py
index 2a2e056f0..e4368edea 100644
--- a/src/finn/transformation/fpgadataflow/insert_iodma.py
+++ b/src/finn/transformation/fpgadataflow/insert_iodma.py
@@ -59,6 +59,7 @@ class InsertIODMA(Transformation):
fc_extw_nodes = list(
filter(
lambda x: x.op_type == "StreamingFCLayer_Batch"
+ and get_by_name(x.attribute, "mem_mode") is not None
and get_by_name(x.attribute, "mem_mode").s.decode("UTF-8") == "external"
and model.find_producer(x.input[1]) is None,
all_nodes,
--
GitLab