feat(hl): add serialize/versionize for hl KSK

zama-ai · Jul 17, 2024 · 8d6bf42 · 8d6bf42
1 parent 35201b0
commit 8d6bf42
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/tfhe/src/high_level_api/backward_compatibility/keys.rs b/tfhe/src/high_level_api/backward_compatibility/keys.rs
@@ -151,3 +151,8 @@ pub(in crate::high_level_api) enum IntegerCompactPublicKeyVersions {
 pub(in crate::high_level_api) enum IntegerCompressedCompactPublicKeyVersions {
     V0(IntegerCompressedCompactPublicKey),
 }
+
+#[derive(VersionsDispatch)]
+pub enum KeySwitchingKeyVersions {
+    V0(KeySwitchingKey),
+}
diff --git a/tfhe/src/high_level_api/keys/key_switching_key.rs b/tfhe/src/high_level_api/keys/key_switching_key.rs
@@ -1,3 +1,6 @@
+use tfhe_versionable::Versionize;
+
+use crate::backward_compatibility::keys::KeySwitchingKeyVersions;
 use crate::high_level_api::integers::{FheIntId, FheUintId};
 use crate::integer::BooleanBlock;
 use crate::prelude::FheKeyswitch;
@@ -16,6 +19,8 @@ impl Display for IncompatibleParameters {
 
 impl std::error::Error for IncompatibleParameters {}
 
+#[derive(serde::Deserialize, serde::Serialize, Versionize)]
+#[versionize(KeySwitchingKeyVersions)]
 pub struct KeySwitchingKey {
     key: crate::integer::key_switching_key::KeySwitchingKey,
 }