github/concrete
1
1
Fork 0
mirror of https://github.com/zama-ai/concrete.git synced 2026-02-10 04:35:03 -05:00
Code Issues Packages Projects Releases Wiki Activity

feat(cuda): implement negation of an LWE ciphertext vector

Browse Source
This commit is contained in:
Agnes Leroy
2022-10-26 16:08:13 +02:00
committed by Agnès Leroy
parent 841c1e6952
commit bc66816341
5 changed files with 100 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/utils/kernel_dimensions.cuh
View File

@@ -1,4 +1,6 @@
int nextPow2(int x) {
#include "math_functions.h"
__host__ __device__ int nextPow2(int x) {
--x;
x |= x >> 1;
x |= x >> 2;
@@ -8,8 +10,10 @@ int nextPow2(int x) {
return ++x;
}
void getNumBlocksAndThreads(const int n, const int maxBlockSize, int &blocks,
int &threads) {
threads = (n < maxBlockSize * 2) ? nextPow2((n + 1) / 2) : maxBlockSize;
__host__ __device__ void getNumBlocksAndThreads(const int n,
const int maxBlockSize,
int &blocks, int &threads) {
threads =
(n < maxBlockSize * 2) ? max(128, nextPow2((n + 1) / 2)) : maxBlockSize;
blocks = (n + threads - 1) / threads;
}