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

tests(LUT): use relative error

Browse Source
This commit is contained in:
youben11
2021-09-03 08:55:50 +01:00
committed by Quentin Bourgerie
parent 1e07733257
commit c4dd639ea7
2 changed files with 21 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
compiler/tests/python/test_compiler_engine.py
View File

@@ -15,6 +15,17 @@ from zamalang import CompilerEngine
(5, 7),
12,
),
],
)
def test_compile_and_run(mlir_input, args, expected_result):
engine = CompilerEngine()
engine.compile_fhe(mlir_input)
assert engine.run(*args) == expected_result
@pytest.mark.parametrize(
"mlir_input, args, expected_result, tab_size",
[
(
"""
func @main(%arg0: !HLFHE.eint<7>) -> !HLFHE.eint<7> {
@@ -26,10 +37,11 @@ from zamalang import CompilerEngine
""",
(5,),
5,
128,
),
],
)
def test_compile_and_run(mlir_input, args, expected_result):
def test_compile_and_run_tlu(mlir_input, args, expected_result, tab_size):
engine = CompilerEngine()
engine.compile_fhe(mlir_input)
assert engine.run(*args) == expected_result
assert abs(engine.run(*args) - expected_result) / tab_size < 0.1