From 5351af299c70247ae5176a2e731ac2452eaf86e6 Mon Sep 17 00:00:00 2001
From: mmrahorovic <mmrahorovic@hotmail.com>
Date: Sun, 20 Feb 2022 22:43:00 +0000
Subject: [PATCH] [transform]: InferStreamingMaxPool support for ceil mode
---
src/finn/transformation/fpgadataflow/convert_to_hls_layers.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py b/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py
index eb9912b48..652136c82 100644
--- a/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py
+++ b/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py
@@ -363,6 +363,7 @@ class InferStreamingMaxPool(Transformation):
ifm_dim_h = mp_in_shape[1]
ifm_dim_w = mp_in_shape[2]
pe = 1
+ ceil_mode = mp_inst.get_nodeattr("ceil_mode")
is_1d = (ifm_dim_h == 1 and k_h == 1) or (ifm_dim_w == 1 and k_w == 1)
is_divisable = ifm_dim_h % k_h == 0 or ifm_dim_w % k_w == 0
if is_1d or is_divisable:
@@ -378,6 +379,7 @@ class InferStreamingMaxPool(Transformation):
ImgDim=(ifm_dim_h, ifm_dim_w),
dataType=dt.name,
PE=pe,
+ CeilMode=ceil_mode,
name="StreamingMaxPool_Batch_" + n.name,
)
graph.node.insert(node_ind, new_node)
--
GitLab