github/tfhe-rs
1
1
Fork 0
mirror of https://github.com/zama-ai/tfhe-rs.git synced 2026-01-10 07:08:03 -05:00
Code Issues Packages Projects Releases Wiki Activity

chore(gpu): fix multi-bit scalar mul benchmark

Browse Source
This commit is contained in:
Agnes Leroy
2024-04-04 17:59:19 +02:00
committed by Agnès Leroy
parent cdeb647629
commit 9ef3183d2e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
backends/tfhe-cuda-backend/cuda/src/pbs/programmable_bootstrap_multibit.cu
View File

@@ -383,7 +383,7 @@ __host__ uint32_t get_lwe_chunk_size(uint32_t ct_count) {
#if CUDA_ARCH >= 900
// Tesla H100
return (ct_count > 10000) ? 45 : 64;
return (ct_count > 10000) ? 30 : 64;
#elif CUDA_ARCH >= 800
// Tesla A100
return (ct_count > 10000) ? 30 : 45;