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

[StreamingFC] bring back IPI codegen for mem_mode=const -- ouchie

parent a7b2224c
No related branches found
No related tags found
No related merge requests found
......@@ -745,7 +745,7 @@ class StreamingFCLayer_Batch(HLSCustomOp):
)
num_w_reps = np.prod(self.get_nodeattr("numInputVectors"))
io_dict = {
"inputs": {"in0": inp, "weights": wei*num_w_reps},
"inputs": {"in0": inp, "weights": wei * num_w_reps},
"outputs": {"out": []},
}
self.rtlsim_multi_io(sim, io_dict)
......@@ -1126,6 +1126,11 @@ class StreamingFCLayer_Batch(HLSCustomOp):
% (node_name, dout_name, node_name, node_name, dout_name)
)
cmd.append("save_bd_design")
elif mem_mode == "const":
# base class impl sufficient for const mode
return super().code_generation_ipi()
else:
raise Exception("Unrecognized mem_mode for StreamingFCLayer")
return cmd
def get_verilog_top_module_intf_names(self):
......
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