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

feat: mimic the exact numpy behavior for matmul

Browse Source
This commit is contained in:
Umut
2022-02-28 15:06:04 +03:00
parent ed28639c57
commit b71cbc8ecb
8 changed files with 108 additions and 86 deletions
Download Patch File Download Diff File Expand all files Collapse all files

55
tests/numpy/test_compile.py
View File

@@ -1681,6 +1681,61 @@ def test_compile_and_run_constant_dot_correctness(
(3,),
(0, 3),
),
pytest.param(
(5,),
(4, 5, 3),
(0, 5),
),
pytest.param(
(4, 5, 3),
(3,),
(0, 5),
),
pytest.param(
(5,),
(2, 4, 5, 3),
(0, 5),
),
pytest.param(
(2, 4, 5, 3),
(3,),
(0, 5),
),
pytest.param(
(5, 4, 3),
(3, 2),
(0, 5),
),
pytest.param(
(4, 3),
(5, 3, 2),
(0, 5),
),
pytest.param(
(2, 5, 4, 3),
(3, 2),
(0, 5),
),
pytest.param(
(5, 4, 3),
(1, 3, 2),
(0, 5),
),
pytest.param(
(1, 4, 3),
(5, 3, 2),
(0, 5),
),
pytest.param(
(5, 4, 3),
(2, 1, 3, 2),
(0, 5),
),
pytest.param(
(2, 1, 4, 3),
(5, 3, 2),
(0, 5),
),
],
)
def test_compile_and_run_matmul_correctness(