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

fix(cuda): Fix an error on an assert.

Browse Source
This commit is contained in:
Pedro Alves
2022-12-20 10:33:35 -03:00
committed by Pedro Alves
parent 29284b4260
commit e82a8d4e81
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/bit_extraction.cu
View File

@@ -29,7 +29,7 @@ void cuda_extract_bits_32(
"equal to the "
"number of streaming multiprocessors on the device divided by 8 * "
"level_count_bsk",
number_of_samples <= number_of_sm * 4. / 2. / level_count_bsk));
number_of_samples <= number_of_sm / 4. / 2. / level_count_bsk));
switch (lwe_dimension_in) {
case 512:
@@ -126,7 +126,7 @@ void cuda_extract_bits_64(
"equal to the "
"number of streaming multiprocessors on the device divided by 8 * "
"level_count_bsk",
number_of_samples <= number_of_sm * 4. / 2. / level_count_bsk));
number_of_samples <= number_of_sm / 4. / 2. / level_count_bsk));
switch (lwe_dimension_in) {
case 512: