Skip to content

Commit 0e0f8ee

Browse files
committed
chore(integer): add determinism check for sum
1 parent 1025246 commit 0e0f8ee

File tree

1 file changed

+7
-3
lines changed
  • tfhe/src/integer/server_key/radix_parallel/tests_unsigned

1 file changed

+7
-3
lines changed

tfhe/src/integer/server_key/radix_parallel/tests_unsigned/test_sum.rs

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -144,11 +144,12 @@ where
144144
{
145145
let param = param.into();
146146
let nb_tests_smaller = nb_tests_smaller_for_params(param);
147-
let (cks, sks) = KEY_CACHE.get_from_params(param, IntegerKeyKind::Radix);
147+
let (cks, mut sks) = KEY_CACHE.get_from_params(param, IntegerKeyKind::Radix);
148148
let cks = RadixClientKey::from((
149149
cks,
150150
crate::integer::server_key::radix_parallel::tests_unsigned::NB_CTXT,
151151
));
152+
sks.set_deterministic_pbs_execution(true);
152153
let sks = Arc::new(sks);
153154

154155
let mut rng = rand::thread_rng();
@@ -175,10 +176,13 @@ where
175176
.collect::<Vec<_>>();
176177

177178
let ct_res = executor.execute(&ctxts).unwrap();
178-
let ct_res: u64 = cks.decrypt(&ct_res);
179+
let res: u64 = cks.decrypt(&ct_res);
179180
let clear = clears.iter().sum::<u64>() % modulus;
180181

181-
assert_eq!(ct_res, clear);
182+
assert_eq!(res, clear);
183+
184+
let ct_res_2 = executor.execute(&ctxts).unwrap();
185+
assert_eq!(ct_res, ct_res_2, "Failed determinism check");
182186
}
183187
}
184188
}

0 commit comments

Comments
 (0)