github/concrete
1
1
Fork 0
mirror of https://github.com/zama-ai/concrete.git synced 2026-02-08 19:44:57 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix: test_quantized_layer

Browse Source
This commit is contained in:
jfrery
2021-11-18 18:23:21 +01:00
committed by jfrery
parent 2fdf22bf4e
commit aa60d8ace6
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/quantization/test_quantized_layers.py
View File

@@ -4,8 +4,9 @@ import pytest
from concrete.quantization import QuantizedArray, QuantizedLinear
# QuantizedLinear unstable with n_bits>23.
N_BITS_LIST = [20, 16, 8, 4]
# QuantizedLinear unstable with n_bits>23
# and hard to test with numpy.isclose with n_bits < 8
N_BITS_LIST = [20, 16, 8]
@pytest.mark.parametrize(