[quant_tensor] Fixed NaN is zero point calculation of __truediv__

Xilinx · Nov 23, 2023 · 993c869 · 993c869
1 parent bd46f89
commit 993c869
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/brevitas/quant_tensor/__init__.py b/src/brevitas/quant_tensor/__init__.py
@@ -412,7 +412,7 @@ def __truediv__(self, other):
             output_signed = self.signed or other.signed
             output_training = self.training or other.training
             if self.is_zero_zero_point(self) and self.is_zero_zero_point(other):
-                output_zero_point = self.zero_point / other.zero_point
+                output_zero_point = self.zero_point * other.zero_point  # Output zero_point is a new, zero-valued tensor
             else:
                 output_zero_point = None  # TODO non-zero zero point
             output = QuantTensor(