We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d141bcd commit d66340cCopy full SHA for d66340c
fms_mo/quant_refactor/linear_utils.py
@@ -63,7 +63,7 @@ def qint_bounds(
63
[torch.IntTensor, torch.IntTensor, torch.dtype]: qint_bounds and torch.qint_dtype
64
"""
65
# Set quantization bounds and datatype based on zero_point
66
- if symmetric and zero_point == 0:
+ if symmetric and torch.sum(zero_point) == 0:
67
qlevel_symmetric = torch.tensor(1) if qlevel_lowering else torch.tensor(0)
68
qlevel_min, qlevel_max = (
69
-(2 ** (num_bits - 1)) + qlevel_symmetric,
0 commit comments