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

add assert

Browse Source
This commit is contained in:
Mayeul@Zama
2023-10-10 18:15:26 +02:00
parent 655f7e6214
commit 5201ff1c51
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tfhe/src/shortint/parameters/mod.rs
View File

@@ -501,11 +501,17 @@ impl ShortintParameterSet {
}
}
pub const fn encryption_key_choice(&self) -> EncryptionKeyChoice {
pub fn encryption_key_choice(&self) -> EncryptionKeyChoice {
match self.inner {
ShortintParameterSetInner::PBSOnly(params) => params.encryption_key_choice(),
ShortintParameterSetInner::WopbsOnly(params) => params.encryption_key_choice,
ShortintParameterSetInner::PBSAndWopbs(params, _) => params.encryption_key_choice(),
ShortintParameterSetInner::PBSAndWopbs(pbs_params, wop_pbs_params) => {
assert_eq!(
pbs_params.encryption_key_choice(),
wop_pbs_params.encryption_key_choice
);
pbs_params.encryption_key_choice()
}
}
}