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(tests): CompileAndRunWithPrecision(6, 7 bits), more retries

Browse Source
This commit is contained in:
rudy
2021-12-20 10:09:57 +01:00
committed by Ayoub Benaissa
parent d8fee32cea
commit cdffb0ec8e
1 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
compiler/tests/unittest/end_to_end_jit_test.cc
View File

@@ -475,12 +475,11 @@ TEST_P(CompileAndRunWithPrecision, identity_func) {
mlir::zamalang::JitCompilerEngine::Lambda lambda =
checkedJit(mlirProgram.str());
if (precision == 7) {
// Test fails with a probability of 5% for a precision of 7. The
// probability of the test failing 5 times in a row is .05^5,
// which is less than 1:10,000 and comparable to the probability
// of failure for the other values.
static const int max_tries = 3;
if (precision >= 6) {
// This test often fails for this precision, so we need retries.
// Reason: the current encryption parameters are a little short for this precision.
static const int max_tries = 10;
for (uint64_t i = 0; i < sizeOfTLU; i++) {
for (int retry = 0; retry <= max_tries; retry++) {