Skip to content
Snippets Groups Projects
Commit 6af0d6dc authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[HLSCustomOp] use non-numReps variant for StreamingMaxPool

DWCs in other variant causing issues otherwise
parent 583b2e29
No related branches found
No related tags found
No related merge requests found
......@@ -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)
]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment