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

feat(cuda): implement LWE ciphertext addition on GPU

Browse Source
This commit is contained in:
Agnes Leroy
2022-10-27 11:04:27 +02:00
committed by Agnès Leroy
parent bc66816341
commit d34f53b7ee
8 changed files with 116 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/utils/kernel_dimensions.cuh
View File

@@ -1,6 +1,4 @@
#include "math_functions.h"
__host__ __device__ int nextPow2(int x) {
inline int nextPow2(int x) {
--x;
x |= x >> 1;
x |= x >> 2;
@@ -10,10 +8,9 @@ __host__ __device__ int nextPow2(int x) {
return ++x;
}
__host__ __device__ void getNumBlocksAndThreads(const int n,
const int maxBlockSize,
int &blocks, int &threads) {
inline 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;
}
}