zama-ai / tfhe-rs

TFHE-rs: A Pure Rust implementation of the TFHE Scheme for Boolean and Integer Arithmetics Over Encrypted Data.
Other
906 stars 143 forks source link

chore(gpu): change the number of threads in smart copy, pack blocks and blocks rotate #1194

Closed agnesLeroy closed 4 months ago

agnesLeroy commented 4 months ago

closes: please link all relevant issues

PR content/description

This PR fixes naming in gpu rotations and increases the number of threads in some kernels (to use the maximum possible number of threads always).

Check-list:

agnesLeroy commented 4 months ago

64 bit multi-bit benchmarks running on h100 -> https://github.com/zama-ai/tfhe-rs/actions/runs/9351572764