diff --git a/src/finn/custom_op/fpgadataflow/streamingmaxpool_batch.py b/src/finn/custom_op/fpgadataflow/streamingmaxpool_batch.py index edbc07300c02c87b47a67297501163766c4cb0dc..ab27b2c63698f1782ff5569b7c9ce4c8e6b61c02 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) ]