fix: added zero_point in [0,nlvl] assert in asymmetric_linear_quanitzation_params

Signed-off-by: Brandon Groth <[email protected]>

Author: BrandonGroth

fms_mo/quant_refactor/linear_utils.py

@@ -296,6 +296,11 @@ def asymmetric_linear_quantization_params(
             zero_point = zero_point.round()
         if signed:
             zero_point += 2 ** (num_bits - 1)
+
+        # Ensure zp in [0, n_levels]
+        zp_bounds = torch.all((zero_point > 0) & (zero_point < n_levels))
+        assert zp_bounds, "Asymmetric zero points should be in [0, 2**bits-1]"
+
         return n_levels, scale, zero_point