From e82a8d4e81e21cda5c29dce8f0efa31a1211ca78 Mon Sep 17 00:00:00 2001 From: Pedro Alves Date: Tue, 20 Dec 2022 10:33:35 -0300 Subject: [PATCH] fix(cuda): Fix an error on an assert. --- src/bit_extraction.cu | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/bit_extraction.cu b/src/bit_extraction.cu index f7963d642..e02a254dc 100644 --- a/src/bit_extraction.cu +++ b/src/bit_extraction.cu @@ -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: