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

feat: give crt decomposition feedback

Browse Source
This commit is contained in:
Umut
2022-11-21 10:15:12 +01:00
committed by Quentin Bourgerie
parent 1e8c0df381
commit 722e4d2eba
5 changed files with 41 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
compiler/tests/python/test_compilation.py
View File

@@ -374,3 +374,22 @@ def test_compile_invalid(mlir_input):
RuntimeError, match=r"Could not find existing crypto parameters for"
):
engine.compile(mlir_input)
def test_crt_decomposition_feedback():
mlir = """
func.func @main(%arg0: !FHE.eint<16>) -> !FHE.eint<16> {
%tlu = arith.constant dense<60000> : tensor<65536xi64>
%1 = "FHE.apply_lookup_table"(%arg0, %tlu): (!FHE.eint<16>, tensor<65536xi64>) -> (!FHE.eint<16>)
return %1: !FHE.eint<16>
}
"""
engine = JITSupport.new()
compilation_result = engine.compile(mlir, options=CompilationOptions.new("main"))
compilation_feedback = engine.load_compilation_feedback(compilation_result)
assert isinstance(compilation_feedback, CompilationFeedback)
assert compilation_feedback.crt_decompositions_of_outputs == [[7, 8, 9, 11, 13]]