chore(hlapi): add gpu stuff to FheInteger trait

Author: tmontaigu

tfhe/src/high_level_api/integers/mod.rs

@@ -58,6 +58,9 @@ pub(super) mod unsigned;
 pub trait IntegerId: FheId + 'static {
     type InnerCpu: crate::integer::IntegerRadixCiphertext;
 
+    #[cfg(feature = "gpu")]
+    type InnerGpu: crate::integer::gpu::ciphertext::CudaIntegerRadixCiphertext;
+    #[cfg(not(feature = "gpu"))]
     type InnerGpu;
 
     type InnerHpu;
@@ -89,4 +92,23 @@ pub trait FheIntegerType: Tagged + private::Sealed {
         tag: Tag,
         re_randomization_metadata: ReRandomizationMetadata,
     ) -> Self;
+
+    #[cfg(feature = "gpu")]
+    fn on_gpu(
+        &self,
+        streams: &crate::core_crypto::gpu::CudaStreams,
+    ) -> MaybeCloned<'_, <Self::Id as IntegerId>::InnerGpu>;
+
+    #[cfg(feature = "gpu")]
+    fn into_gpu(
+        self,
+        streams: &crate::core_crypto::gpu::CudaStreams,
+    ) -> <Self::Id as IntegerId>::InnerGpu;
+
+    #[cfg(feature = "gpu")]
+    fn from_gpu(
+        inner: <Self::Id as IntegerId>::InnerGpu,
+        tag: Tag,
+        re_randomization_metadata: ReRandomizationMetadata,
+    ) -> Self;
 }

tfhe/src/high_level_api/integers/signed/base.rs

@@ -137,6 +137,31 @@ where
     ) -> Self {
         Self::new(inner, tag, re_randomization_metadata)
     }
+
+    #[cfg(feature = "gpu")]
+    fn on_gpu(
+        &self,
+        streams: &crate::core_crypto::gpu::CudaStreams,
+    ) -> MaybeCloned<'_, <Self::Id as IntegerId>::InnerGpu> {
+        self.ciphertext.on_gpu(streams)
+    }
+
+    #[cfg(feature = "gpu")]
+    fn into_gpu(
+        self,
+        streams: &crate::core_crypto::gpu::CudaStreams,
+    ) -> <Self::Id as IntegerId>::InnerGpu {
+        self.ciphertext.into_gpu(streams)
+    }
+
+    #[cfg(feature = "gpu")]
+    fn from_gpu(
+        inner: <Self::Id as IntegerId>::InnerGpu,
+        tag: Tag,
+        re_randomization_metadata: ReRandomizationMetadata,
+    ) -> Self {
+        Self::new(inner, tag, re_randomization_metadata)
+    }
 }
 
 impl<Id> FheInt<Id>

tfhe/src/high_level_api/integers/unsigned/base.rs

@@ -174,6 +174,31 @@ where
     ) -> Self {
         Self::new(inner, tag, re_randomization_metadata)
     }
+
+    #[cfg(feature = "gpu")]
+    fn on_gpu(
+        &self,
+        streams: &crate::core_crypto::gpu::CudaStreams,
+    ) -> MaybeCloned<'_, <Self::Id as IntegerId>::InnerGpu> {
+        self.ciphertext.on_gpu(streams)
+    }
+
+    #[cfg(feature = "gpu")]
+    fn into_gpu(
+        self,
+        streams: &crate::core_crypto::gpu::CudaStreams,
+    ) -> <Self::Id as IntegerId>::InnerGpu {
+        self.ciphertext.into_gpu(streams)
+    }
+
+    #[cfg(feature = "gpu")]
+    fn from_gpu(
+        inner: <Self::Id as IntegerId>::InnerGpu,
+        tag: Tag,
+        re_randomization_metadata: ReRandomizationMetadata,
+    ) -> Self {
+        Self::new(inner, tag, re_randomization_metadata)
+    }
 }
 
 impl<Id> Tagged for FheUint<Id>