chore(gpu): plug in signed gpu tests in the hl api

zama-ai · Feb 24, 2025 · 72c86fd · 72c86fd
1 parent 1588334
commit 72c86fd
Show file tree

Hide file tree

Showing 9 changed files with 771 additions and 694 deletions.
diff --git a/tfhe/src/high_level_api/integers/signed/base.rs b/tfhe/src/high_level_api/integers/signed/base.rs
@@ -186,9 +186,13 @@ where
                 Self::new(ciphertext, cpu_key.tag.clone())
             }
             #[cfg(feature = "gpu")]
-            InternalServerKey::Cuda(_) => {
-                panic!("Cuda devices does not support abs yet")
-            }
+            InternalServerKey::Cuda(cuda_key) => with_thread_local_cuda_streams(|streams| {
+                let result = cuda_key
+                    .key
+                    .key
+                    .abs(&*self.ciphertext.on_gpu(streams), streams);
+                Self::new(result, cuda_key.tag.clone())
+            }),
         })
     }