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

fix(python): register hlfhelinalg dialect

Browse Source
This commit is contained in:
youben11
2021-11-03 09:58:23 +01:00
committed by Ayoub Benaissa
parent 5999d402ec
commit 97ee8134ed
10 changed files with 93 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -44,6 +44,15 @@ VALID_INPUTS = [
""",
id="add_eint_int_cst",
),
pytest.param(
"""
func @main(%a0: tensor<4x!HLFHE.eint<2>>, %a1: tensor<4xi3>) -> tensor<4x!HLFHE.eint<2>> {
%1 = "HLFHELinalg.add_eint_int"(%a0, %a1) : (tensor<4x!HLFHE.eint<2>>, tensor<4xi3>) -> tensor<4x!HLFHE.eint<2>>
return %1: tensor<4x!HLFHE.eint<2>>
}
""",
id="add_eint_int_1D",
),
]
INVALID_INPUTS = [
@@ -56,6 +65,16 @@ INVALID_INPUTS = [
""",
id="eint<0>",
),
pytest.param(
"""
func @main(%a0: tensor<2x2x3x4x!HLFHE.eint<2>>, %a1: tensor<2x2x2x4xi3>) -> tensor<2x2x3x4x!HLFHE.eint<2>> {
// expected-error @+1 {{'HLFHELinalg.add_eint_int' op has the dimension #2 of the operand #1 incompatible with other operands, got 2 expect 1 or 3}}
%1 = "HLFHELinalg.add_eint_int"(%a0, %a1) : (tensor<2x2x3x4x!HLFHE.eint<2>>, tensor<2x2x2x4xi3>) -> tensor<2x2x3x4x!HLFHE.eint<2>>
return %1 : tensor<2x2x3x4x!HLFHE.eint<2>>
}
""",
id="incompatible dimensions",
),
]