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

feat(compiler): support woppbs in simulation

Browse Source
This commit is contained in:
youben11
2023-09-25 10:42:01 +01:00
committed by Ayoub Benaissa
parent 0caa659244
commit e4835bd002
4 changed files with 130 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
compilers/concrete-compiler/compiler/tests/python/test_simulation.py
View File

@@ -170,6 +170,22 @@ end_to_end_fixture = [
).reshape((4, 4)),
id="matul_chain_with_crt",
),
pytest.param(
"""
func.func @main(%arg0: !FHE.eint<14>, %arg1: tensor<16384xi64>) -> !FHE.eint<14> {
%cst = arith.constant 15 : i15
%v = "FHE.add_eint_int"(%arg0, %cst): (!FHE.eint<14>, i15) -> (!FHE.eint<14>)
%1 = "FHE.apply_lookup_table"(%v, %arg1): (!FHE.eint<14>, tensor<16384xi64>) -> (!FHE.eint<14>)
return %1: !FHE.eint<14>
}
""",
(
81,
np.array(range(16384), dtype=np.uint64),
),
96,
id="add_lut_crt",
),
]
end_to_end_parallel_fixture = [