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

fix(TestLib): add missing test using a LUT/PBS.

Browse Source
This commit is contained in:
Antoniu Pop
2022-02-28 10:12:43 +00:00
committed by Antoniu Pop
parent 8ee456315e
commit f2bb077108
1 changed files with 28 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
compiler/tests/TestLib/testlib_unit_test.cpp
View File

@@ -25,6 +25,14 @@ using concretelang::clientlib::tensor1_out;
using concretelang::clientlib::tensor2_out;
using concretelang::clientlib::tensor3_out;
std::vector<uint8_t>
values_3bits() {
return {0, 1, 2, 5, 7};
}
std::vector<uint8_t>
values_6bits() {
return {0, 1, 2, 13, 22, 59, 62, 63};
}
std::vector<uint8_t>
values_7bits() {
return {0, 1, 2, 63, 64, 65, 125, 126};
@@ -364,3 +372,23 @@ func @extract(%arg0: tensor<3x!FHE.eint<7>>, %arg1: tensor<3x!FHE.eint<7>>) -> !
fileContent(THIS_TEST_DIRECTORY + "/call_2t_1s_with_header-client.h.generated"),
fileContent(OUT_DIRECTORY + "/call_2t_1s_with_header-client.h"));
}
TEST(CompiledModule, call_2s_1s_lookup_table) {
std::string source = R"(
func @main(%arg0: !FHE.eint<6>, %arg1: !FHE.eint<3>) -> !FHE.eint<6> {
%tlu_7 = arith.constant dense<[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63]> : tensor<64xi64>
%tlu_3 = arith.constant dense<[0, 1, 2, 3, 4, 5, 6, 7]> : tensor<8xi64>
%a = "FHE.apply_lookup_table"(%arg0, %tlu_7): (!FHE.eint<6>, tensor<64xi64>) -> (!FHE.eint<6>)
%b = "FHE.apply_lookup_table"(%arg1, %tlu_3): (!FHE.eint<3>, tensor<8xi64>) -> (!FHE.eint<6>)
%a_plus_b = "FHE.add_eint"(%a, %b): (!FHE.eint<6>, !FHE.eint<6>) -> (!FHE.eint<6>)
return %a_plus_b: !FHE.eint<6>
}
)";
std::string outputLib = outputLibFromThis(this->test_info_);
auto compiled = compile(outputLib, source);
auto lambda = load<TestTypedLambda<scalar_out, scalar_in, scalar_in>>(outputLib);
for(auto a: values_6bits()) for(auto b: values_3bits()) {
auto res = lambda.call(a, b);
ASSERT_EQ_OUTCOME(res, a + b);
}
}