Xilinx / finn

Dataflow compiler for QNN inference on FPGAs
https://xilinx.github.io/finn
BSD 3-Clause "New" or "Revised" License
760 stars 243 forks source link

InsertDWC and multiple input/output nodes #1201

Open mdanilow opened 2 months ago

mdanilow commented 2 months ago

A quick fix for the InsertDWC transform, the current implementation assumes all of the outputs of the producer and inputs of the consumer to have the same folded shapes, stream widths and datatypes. It caused problems for models with branched structures, especially when we allow join nodes (stream addition, concat, etc...) to have different datatypes among inputs.