github/tfhe-rs
1
1
Fork 0
mirror of https://github.com/zama-ai/tfhe-rs.git synced 2026-01-10 07:08:03 -05:00
Code Issues Packages Projects Releases Wiki Activity

chore(gpu): print inputs to more determinism tests for debugging purposes

Browse Source
This commit is contained in:
Agnes Leroy
2024-10-14 10:30:20 +02:00
committed by Agnès Leroy
parent f2651ed558
commit d794f4dd0f
5 changed files with 42 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tfhe/src/integer/server_key/radix_parallel/tests_cases_unsigned.rs
View File

@@ -1327,6 +1327,7 @@ where
res = executor.execute((&res, &ctxt_2));
assert!(res.block_carries_are_empty());
assert_eq!(res, tmp);
assert_eq!(res, tmp, "Failed determinism check, \n\n\n msg0: {clear1}, msg1: {clear2}, \n\n\nctxt0: {ctxt_1:?}, \n\n\nctxt1: {ctxt_2:?}\n\n\n");
clear = (clear * clear2) % modulus;
}
let dec: u64 = cks.decrypt(&res);

20
tfhe/src/integer/server_key/radix_parallel/tests_signed/test_ilog2.rs
View File

@@ -145,7 +145,10 @@ pub(crate) fn signed_default_count_consecutive_bits_test<P, T>(
let ct_res = executor.execute(&ctxt);
let tmp = executor.execute(&ctxt);
assert!(ct_res.block_carries_are_empty());
assert_eq!(ct_res, tmp);
assert_eq!(
ct_res, tmp,
"Failed determinism check, \n\n\n msg: {clear}, \n\n\nctxt: {ctxt:?}\n\n\n"
);
let decrypted_result: u32 = cks.decrypt(&ct_res);
let expected_result = compute_expected_clear(clear);
@@ -298,7 +301,10 @@ where
let ct_res = executor.execute(&ctxt);
let tmp = executor.execute(&ctxt);
assert!(ct_res.block_carries_are_empty());
assert_eq!(ct_res, tmp);
assert_eq!(
ct_res, tmp,
"Failed determinism check, \n\n\n msg: {clear}, \n\n\nctxt: {ctxt:?}\n\n\n"
);
let decrypted_result: u32 = cks.decrypt(&ct_res);
let expected_result = clear.ilog2();
@@ -353,7 +359,10 @@ where
let ct_res = executor.execute(&ctxt);
let tmp = executor.execute(&ctxt);
assert!(ct_res.block_carries_are_empty());
assert_eq!(ct_res, tmp);
assert_eq!(
ct_res, tmp,
"Failed determinism check, \n\n\n msg: {clear}, \n\n\nctxt: {ctxt:?}\n\n\n"
);
let decrypted_result: u32 = cks.decrypt(&ct_res);
let expected_result = clear.ilog2();
@@ -429,7 +438,10 @@ where
let (ct_res, is_ok) = executor.execute(&ctxt);
let (tmp, tmp_is_ok) = executor.execute(&ctxt);
assert!(ct_res.block_carries_are_empty());
assert_eq!(ct_res, tmp);
assert_eq!(
ct_res, tmp,
"Failed determinism check, \n\n\n msg: {clear}, \n\n\nctxt: {ctxt:?}\n\n\n"
);
assert_eq!(is_ok, tmp_is_ok);
let decrypted_result: u32 = cks.decrypt(&ct_res);

1
tfhe/src/integer/server_key/radix_parallel/tests_signed/test_mul.rs
View File

@@ -253,6 +253,7 @@ where
let tmp_ct = executor.execute((&ctxt_0, &ctxt_1));
assert!(ct_res.block_carries_are_empty());
assert_eq!(ct_res, tmp_ct);
assert_eq!(ct_res, tmp_ct, "Failed determinism check, \n\n\n msg0: {clear_0}, msg1: {clear_1}, \n\n\nctxt0: {ctxt_0:?}, \n\n\nctxt1: {ctxt_1:?}\n\n\n");
clear = signed_mul_under_modulus(clear_0, clear_1, modulus);

25
tfhe/src/integer/server_key/radix_parallel/tests_unsigned/test_ilog2.rs
View File

@@ -139,7 +139,10 @@ pub(crate) fn default_count_consecutive_bits_test<P, T>(
let ct_res = executor.execute(&ctxt);
let tmp = executor.execute(&ctxt);
assert!(ct_res.block_carries_are_empty());
assert_eq!(ct_res, tmp);
assert_eq!(
ct_res, tmp,
"Failed determinism check, \n\n\n msg: {clear}, \n\n\nctxt: {ctxt:?}\n\n\n"
);
let decrypted_result: u32 = cks.decrypt(&ct_res);
let expected_result = compute_expected_clear(clear);
@@ -184,7 +187,10 @@ pub(crate) fn default_count_consecutive_bits_test<P, T>(
let ct_res = executor.execute(&ctxt);
let tmp = executor.execute(&ctxt);
assert!(ct_res.block_carries_are_empty());
assert_eq!(ct_res, tmp);
assert_eq!(
ct_res, tmp,
"Failed determinism check, \n\n\n msg: {clear}, \n\n\nctxt: {ctxt:?}\n\n\n"
);
let decrypted_result: u32 = cks.decrypt(&ct_res);
let expected_result = compute_expected_clear(clear);
@@ -285,7 +291,10 @@ where
let ct_res = executor.execute(&ctxt);
let tmp = executor.execute(&ctxt);
assert!(ct_res.block_carries_are_empty());
assert_eq!(ct_res, tmp);
assert_eq!(
ct_res, tmp,
"Failed determinism check, \n\n\n msg: {clear}, \n\n\nctxt: {ctxt:?}\n\n\n"
);
let decrypted_result: u32 = cks.decrypt(&ct_res);
let expected_result = clear.ilog2();
@@ -340,7 +349,10 @@ where
let ct_res = executor.execute(&ctxt);
let tmp = executor.execute(&ctxt);
assert!(ct_res.block_carries_are_empty());
assert_eq!(ct_res, tmp);
assert_eq!(
ct_res, tmp,
"Failed determinism check, \n\n\n msg: {clear}, \n\n\nctxt: {ctxt:?}\n\n\n"
);
let decrypted_result: u32 = cks.decrypt(&ct_res);
let expected_result = clear.ilog2();
@@ -413,7 +425,10 @@ where
let (ct_res, is_ok) = executor.execute(&ctxt);
let (tmp, tmp_is_ok) = executor.execute(&ctxt);
assert!(ct_res.block_carries_are_empty());
assert_eq!(ct_res, tmp);
assert_eq!(
ct_res, tmp,
"Failed determinism check, \n\n\n msg: {clear}, \n\n\nctxt: {ctxt:?}\n\n\n"
);
assert_eq!(is_ok, tmp_is_ok);
let decrypted_result: u32 = cks.decrypt(&ct_res);

5
tfhe/src/integer/server_key/radix_parallel/tests_unsigned/test_scalar_div_mod.rs
View File

@@ -78,7 +78,8 @@ where
assert!(q.block_carries_are_empty());
assert!(r.block_carries_are_empty());
assert_eq!(q, q2);
assert_eq!(r, r2);
assert_eq!(q, q2, "Failed determinism check, \n\n\n msg: {clear}, scalar: {scalar}, \n\n\nctxt: {ct:?}\n\n\n");
assert_eq!(r, r2, "Failed determinism check, \n\n\n msg: {clear}, scalar: {scalar}, \n\n\nctxt: {ct:?}\n\n\n");
let q_res: u64 = cks.decrypt(&q);
let r_res: u64 = cks.decrypt(&r);
@@ -95,7 +96,9 @@ where
assert!(q.block_carries_are_empty());
assert!(r.block_carries_are_empty());
assert_eq!(q, q2);
assert_eq!(q, q2, "Failed determinism check, \n\n\n msg: {clear}, scalar: {scalar}, \n\n\nctxt: {ct:?}\n\n\n");
assert_eq!(r, r2);
assert_eq!(r, r2, "Failed determinism check, \n\n\n msg: {clear}, scalar: {scalar}, \n\n\nctxt: {ct:?}\n\n\n");
let q_res: u64 = cks.decrypt(&q);
let r_res: u64 = cks.decrypt(&r);