github/concrete
1
1
Fork 0
mirror of https://github.com/zama-ai/concrete.git synced 2026-02-09 20:25:34 -05:00
Code Issues Packages Projects Releases Wiki Activity

chore(cuda): fix asserts regarding the base log value

Browse Source
This commit is contained in:
Agnes Leroy
2022-11-28 11:56:29 +01:00
committed by Agnès Leroy
parent 739db73d46
commit f04a29aea4
4 changed files with 12 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/bootstrap_amortized.cu
View File

@@ -65,7 +65,7 @@ void cuda_bootstrap_amortized_lwe_ciphertext_vector_32(
uint32_t num_lut_vectors, uint32_t lwe_idx, uint32_t max_shared_memory) {
assert(
("Error (GPU amortized PBS): base log should be <= 16", base_log <= 16));
("Error (GPU amortized PBS): base log should be <= 32", base_log <= 32));
assert(("Error (GPU amortized PBS): glwe_dimension should be equal to 1",
glwe_dimension == 1));
assert(("Error (GPU amortized PBS): polynomial size should be one of 512, "
@@ -123,7 +123,7 @@ void cuda_bootstrap_amortized_lwe_ciphertext_vector_64(
uint32_t num_lut_vectors, uint32_t lwe_idx, uint32_t max_shared_memory) {
assert(
("Error (GPU amortized PBS): base log should be <= 16", base_log <= 16));
("Error (GPU amortized PBS): base log should be <= 64", base_log <= 64));
assert(("Error (GPU amortized PBS): glwe_dimension should be equal to 1",
glwe_dimension == 1));
assert(("Error (GPU amortized PBS): polynomial size should be one of 512, "

8
src/bootstrap_low_latency.cu
View File

@@ -63,8 +63,8 @@ void cuda_bootstrap_low_latency_lwe_ciphertext_vector_32(
uint32_t base_log, uint32_t level_count, uint32_t num_samples,
uint32_t num_lut_vectors, uint32_t lwe_idx, uint32_t max_shared_memory) {
assert(("Error (GPU low latency PBS): base log should be <= 16",
base_log <= 16));
assert(("Error (GPU low latency PBS): base log should be <= 32",
base_log <= 32));
assert(("Error (GPU low latency PBS): glwe_dimension should be equal to 1",
glwe_dimension == 1));
assert(("Error (GPU low latency PBS): polynomial size should be one of 512, "
@@ -116,8 +116,8 @@ void cuda_bootstrap_low_latency_lwe_ciphertext_vector_64(
uint32_t base_log, uint32_t level_count, uint32_t num_samples,
uint32_t num_lut_vectors, uint32_t lwe_idx, uint32_t max_shared_memory) {
assert(("Error (GPU low latency PBS): base log should be <= 16",
base_log <= 16));
assert(("Error (GPU low latency PBS): base log should be <= 64",
base_log <= 64));
assert(("Error (GPU low latency PBS): glwe_dimension should be equal to 1",
glwe_dimension == 1));
assert(("Error (GPU low latency PBS): polynomial size should be one of 512, "

12
src/bootstrap_wop.cu
View File

@@ -6,7 +6,7 @@ void cuda_cmux_tree_32(void *v_stream, void *glwe_array_out, void *ggsw_in,
uint32_t level_count, uint32_t r,
uint32_t max_shared_memory) {
assert(("Error (GPU Cmux tree): base log should be <= 16", base_log <= 16));
assert(("Error (GPU Cmux tree): base log should be <= 32", base_log <= 32));
assert(("Error (GPU Cmux tree): polynomial size should be one of 512, 1024, "
"2048, 4096, 8192",
polynomial_size == 512 || polynomial_size == 1024 ||
@@ -61,7 +61,7 @@ void cuda_cmux_tree_64(void *v_stream, void *glwe_array_out, void *ggsw_in,
uint32_t level_count, uint32_t r,
uint32_t max_shared_memory) {
assert(("Error (GPU Cmux tree): base log should be <= 16", base_log <= 16));
assert(("Error (GPU Cmux tree): base log should be <= 64", base_log <= 64));
assert(("Error (GPU Cmux tree): polynomial size should be one of 512, 1024, "
"2048, 4096, 8192",
polynomial_size == 512 || polynomial_size == 1024 ||
@@ -119,8 +119,8 @@ void cuda_extract_bits_32(
uint32_t lwe_dimension_out, uint32_t glwe_dimension, uint32_t base_log_bsk,
uint32_t level_count_bsk, uint32_t base_log_ksk, uint32_t level_count_ksk,
uint32_t number_of_samples) {
assert(("Error (GPU extract bits): base log should be <= 16",
base_log_bsk <= 16));
assert(("Error (GPU extract bits): base log should be <= 32",
base_log_bsk <= 32));
assert(("Error (GPU extract bits): glwe_dimension should be equal to 1",
glwe_dimension == 1));
assert(("Error (GPU extract bits): lwe_dimension_in should be one of "
@@ -190,8 +190,8 @@ void cuda_extract_bits_64(
uint32_t lwe_dimension_out, uint32_t glwe_dimension, uint32_t base_log_bsk,
uint32_t level_count_bsk, uint32_t base_log_ksk, uint32_t level_count_ksk,
uint32_t number_of_samples) {
assert(("Error (GPU extract bits): base log should be <= 16",
base_log_bsk <= 16));
assert(("Error (GPU extract bits): base log should be <= 64",
base_log_bsk <= 64));
assert(("Error (GPU extract bits): glwe_dimension should be equal to 1",
glwe_dimension == 1));
assert(("Error (GPU extract bits): lwe_dimension_in should be one of "

1
src/negation.cuh
View File

@@ -11,7 +11,6 @@
#include "linear_algebra.h"
#include "utils/kernel_dimensions.cuh"
template <typename T>
__global__ void negation(T *output, T *input, uint32_t num_entries) {