[RTL MVAU] Allow for 4bit compute with dsp48 for versal devices

Xilinx · Aug 22, 2024 · 380d2ac · 380d2ac
1 parent 188bf17
commit 380d2ac
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/finn/custom_op/fpgadataflow/rtl/matrixvectoractivation_rtl.py b/src/finn/custom_op/fpgadataflow/rtl/matrixvectoractivation_rtl.py
@@ -208,7 +208,10 @@ def _resolve_impl_style(self, dsp_block):
         weight_width = self.get_input_datatype(1).bitwidth()
 
         if dsp_block == "DSP58":
-            return "mvu_vvu_8sx9_dsp58"
+            if act_width <= 4 and weight_width <= 4:
+                return "mvu_4sx4u_dsp48e2"
+            else:
+                return "mvu_vvu_8sx9_dsp58"
         else:
             if act_width <= 4 and weight_width <= 4:
                 if dsp_block == "DSP48E1":