Fix (core/float): add default for float_scaling_impl

Xilinx · Jun 18, 2024 · 181ef80 · 181ef80
1 parent 6bdb1f8
commit 181ef80
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/brevitas/core/quant/float.py b/src/brevitas/core/quant/float.py
@@ -52,6 +52,9 @@ def __init__(
         if scaling_impl is None:
             scaling_impl = ConstScaling(1., device=device, dtype=dtype)
 
+        if float_scaling_impl is None:
+            float_scaling_impl = ConstScaling(1., device=device, dtype=dtype)
+
         # Zero-point is currently hardcoded to 0
         self.zero_point_impl = StatelessBuffer(torch.tensor(0., device=device, dtype=dtype))
         self.float_scaling_impl = float_scaling_impl