github/concrete
1
1
Fork 0
mirror of https://github.com/zama-ai/concrete.git synced 2026-02-09 03:55:04 -05:00
Code Issues Packages Projects Releases Wiki Activity

chore: fix test for matmul correctness

Browse Source 
- bit-width could get too high, rename input_range to be clearer on bounds
- fix bounds that seemed to want to use a set number of bits but being off
by one (e.g. 4 -> 3 as 0-3 is the range for a 2 bits integer)

closes #1352
This commit is contained in:
Arthur Meyre
2022-01-10 09:50:49 +01:00
parent d0f08bc378
commit d8a64f25b8
1 changed files with 16 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
tests/numpy/test_compile.py
View File

@@ -1329,42 +1329,42 @@ def test_compile_and_run_constant_dot_correctness(
@pytest.mark.parametrize(
"lhs_shape,rhs_shape,input_range",
"lhs_shape,rhs_shape,input_range_inclusive_bound",
[
pytest.param(
(3, 2),
(2, 3),
(0, 4),
(0, 3),
),
pytest.param(
(1, 2),
(2, 1),
(0, 4),
(0, 3),
),
pytest.param(
(3, 3),
(3, 3),
(0, 4),
(0, 3),
),
pytest.param(
(2, 1),
(1, 2),
(0, 8),
(0, 7),
),
pytest.param(
(2,),
(2,),
(0, 8),
(0, 7),
),
pytest.param(
(5, 5),
(5,),
(0, 4),
(0, 3),
),
pytest.param(
(5,),
(5, 5),
(0, 4),
(0, 3),
),
pytest.param(
(3, 2),
@@ -1374,21 +1374,25 @@ def test_compile_and_run_constant_dot_correctness(
pytest.param(
(5,),
(5, 3),
(0, 4),
(0, 3),
),
pytest.param(
(5, 3),
(3,),
(0, 4),
(0, 3),
),
],
)
def test_compile_and_run_matmul_correctness(
lhs_shape, rhs_shape, input_range, default_compilation_configuration, check_array_equality
lhs_shape,
rhs_shape,
input_range_inclusive_bound,
default_compilation_configuration,
check_array_equality,
):
"""Test correctness of results when running a compiled function"""
low, high = input_range
low, high = input_range_inclusive_bound
check_mod = low < 0 or high < 0