From 6af0d6dcc402ee97f29b208add3825100a8e465f Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <yamanu@xilinx.com>
Date: Thu, 12 Aug 2021 10:12:27 +0200
Subject: [PATCH] [HLSCustomOp] use non-numReps variant for StreamingMaxPool
DWCs in other variant causing issues otherwise
---
src/finn/custom_op/fpgadataflow/streamingmaxpool_batch.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/finn/custom_op/fpgadataflow/streamingmaxpool_batch.py b/src/finn/custom_op/fpgadataflow/streamingmaxpool_batch.py
index edbc07300..ab27b2c63 100644
--- a/src/finn/custom_op/fpgadataflow/streamingmaxpool_batch.py
+++ b/src/finn/custom_op/fpgadataflow/streamingmaxpool_batch.py
@@ -212,12 +212,12 @@ class StreamingMaxPool_Batch(HLSCustomOp):
"%s<ImgDim, PoolDim, NumChannels>(in0, out, numReps);" % (op)
]
else:
- op = "StreamingMaxPool_Precision_Batch"
+ op = "StreamingMaxPool_Precision"
dtype = self.get_input_datatype()
dtype_hls = dtype.get_hls_datatype_str()
minval_str = str(int(dtype.min()))
self.code_gen_dict["$DOCOMPUTE$"] = [
- "%s<ImgDim, PoolDim, NumChannels, %s, %s>(in0, out, numReps);"
+ "%s<ImgDim, PoolDim, NumChannels, %s, %s>(in0, out);"
% (op, dtype_hls, minval_str)
]
--
GitLab