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

feat(execution): run on both int and tensor args

Browse Source
This commit is contained in:
youben11
2021-09-09 16:02:17 +01:00
committed by Ayoub Benaissa
parent c37ac41c1a
commit 967fda07a0
4 changed files with 138 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -15,6 +15,18 @@ from zamalang import CompilerEngine
(5, 7),
12,
),
(
"""
func @main(%arg0: tensor<4x!HLFHE.eint<7>>, %arg1: tensor<4xi8>) -> !HLFHE.eint<7>
{
%ret = "HLFHE.dot_eint_int"(%arg0, %arg1) :
(tensor<4x!HLFHE.eint<7>>, tensor<4xi8>) -> !HLFHE.eint<7>
return %ret : !HLFHE.eint<7>
}
""",
([1, 2, 3, 4], [4, 3, 2, 1]),
20,
),
],
)
def test_compile_and_run(mlir_input, args, expected_result):