fix(hlapi): bind missing cuda bitnot

zama-ai · Feb 14, 2024 · cb42636 · cb42636
1 parent 767e7c4
commit cb42636
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/tfhe/src/high_level_api/booleans/base.rs b/tfhe/src/high_level_api/booleans/base.rs
@@ -1246,7 +1246,7 @@ impl std::ops::Not for &FheBool {
             }
             #[cfg(feature = "gpu")]
             InternalServerKey::Cuda(cuda_key) => with_thread_local_cuda_stream(|stream| {
-                let inner = cuda_key
+                let mut inner = cuda_key
                     .key
                     .scalar_bitxor(&self.ciphertext.on_gpu(), 1, stream);
                 inner.info.blocks[0].degree = crate::shortint::ciphertext::Degree::new(1);

diff --git a/tfhe/src/high_level_api/integers/unsigned/ops.rs b/tfhe/src/high_level_api/integers/unsigned/ops.rs
@@ -1979,9 +1979,10 @@ where
                 FheUint::new(ciphertext)
             }
             #[cfg(feature = "gpu")]
-            InternalServerKey::Cuda(_) => {
-                panic!("Not '!' is not yet supported by Cuda devices")
-            }
+            InternalServerKey::Cuda(cuda_key) => with_thread_local_cuda_stream(|stream| {
+                let inner_result = cuda_key.key.bitnot(&self.ciphertext.on_gpu(), stream);
+                FheUint::new(inner_result)
+            }),
         })
     }
 }