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

fix(gpu): Using tbc for classical 64 bits pbs on H100

Browse Source
This commit is contained in:
Enzo Di Maria
2025-12-17 14:32:25 +01:00
committed by Guillermo Oyarzun
parent 312ce494bf
commit 0a59e86675
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_tbc_classic.cuh
View File

@@ -626,7 +626,7 @@ __host__ bool supports_thread_block_clusters_on_classic_programmable_bootstrap(
uint32_t num_samples, uint32_t glwe_dimension, uint32_t polynomial_size,
uint32_t level_count, uint32_t max_shared_memory) {
if (!cuda_check_support_thread_block_clusters() || num_samples > 128)
if (!cuda_check_support_thread_block_clusters())
return false;
uint64_t full_sm = get_buffer_size_full_sm_programmable_bootstrap_tbc<Torus>(