diff --git a/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py b/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py index ee4990897f5aa5c4c5cff23aa87d854a2ecaae56..75be2bec869959cdd8fc227677f80a5e40c2a56e 100644 --- a/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py +++ b/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py @@ -745,7 +745,7 @@ class StreamingFCLayer_Batch(HLSCustomOp): ) num_w_reps = np.prod(self.get_nodeattr("numInputVectors")) io_dict = { - "inputs": {"in0": inp, "weights": np.tile(wei, num_w_reps)}, + "inputs": {"in0": inp, "weights": wei*num_w_reps}, "outputs": {"out": []}, } self.rtlsim_multi_io(sim, io_dict)