[change] Disabled round-to even in RequantShift nodes

pulp-platform · Feb 2, 2023 · e8cfc0d · e8cfc0d
1 parent 39816e8
commit e8cfc0d
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/algorithms/pact/pact_ops.py b/algorithms/pact/pact_ops.py
@@ -98,12 +98,9 @@ def forward(ctx, x, mul, add, div, signed, n_levels_out, cmsis_requant):
             # PULP-NN performs multiplication first, then addition and
             # division. Division is with flooring.
             else:
-                try:
-                    y = x * mul + add
-                    y = (y/div).floor()
-                except Exception as e:
-                    print(e)
-                    import IPython; IPython.embed()
+                y = x * mul + add
+                # Avoid round to even behaviour, friggin pytorch
+                y = torch.floor((y / div) + 0.5)
 
             if not signed:
             # if unsigned: clip y to interval (0, n_levels-1)