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

[VVAU] bugfix in weight reps for node-by-node rtlsim

parent b443dbff
No related branches found
No related tags found
No related merge requests found
......@@ -670,7 +670,9 @@ class VectorVectorActivation(HLSCustomOp):
wei = npy_to_rtlsim_input(
"{}/weights.npy".format(code_gen_dir), export_wdt, wnbits
)
num_w_reps = 1
dim_h, dim_w = self.get_nodeattr("Dim")
num_w_reps = dim_h * dim_w
io_dict = {
"inputs": {"in0": inp, "weights": wei * num_w_reps},
"outputs": {"out": []},
......
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