feat(integer): add raw parts API for the KeySwitchingKeyMaterial

zama-ai · Jul 25, 2024 · e7c06ef · e7c06ef
1 parent 7b14fe6
commit e7c06ef
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tfhe/src/integer/key_switching_key/mod.rs b/tfhe/src/integer/key_switching_key/mod.rs
@@ -48,6 +48,17 @@ pub struct KeySwitchingKeyMaterial {
 }
 
 impl KeySwitchingKeyMaterial {
+    pub fn into_raw_parts(self) -> crate::shortint::key_switching_key::KeySwitchingKeyMaterial {
+        let Self { material } = self;
+        material
+    }
+
+    pub fn from_raw_parts(
+        material: crate::shortint::key_switching_key::KeySwitchingKeyMaterial,
+    ) -> Self {
+        Self { material }
+    }
+
     pub fn as_view(&self) -> KeySwitchingKeyMaterialView<'_> {
         KeySwitchingKeyMaterialView {
             material: self.material.as_view(),