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

fix(bench): HpuDevice should no be shared across thread

Browse Source 
Last rebase introduce rayon parralelisation inside erc20. This is not beneficial for all hardware, thus duplicate the whitepaper function to have sequential/parrallel version
This commit is contained in:
Baptiste Roux
2025-05-14 14:27:27 +02:00
parent ba0060815f
commit ce9cbe4fb5
1 changed files with 28 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
tfhe-benchmark/benches/high_level_api/erc20.rs
View File

@@ -26,6 +26,28 @@ pub fn transfer_whitepaper<FheType>(
to_amount: &FheType,
amount: &FheType,
) -> (FheType, FheType)
where
FheType: Add<Output = FheType> + for<'a> FheOrd<&'a FheType>,
FheBool: IfThenElse<FheType>,
for<'a> &'a FheType: Add<Output = FheType> + Sub<Output = FheType>,
{
let has_enough_funds = (from_amount).ge(amount);
let mut new_to_amount = to_amount + amount;
new_to_amount = has_enough_funds.if_then_else(&new_to_amount, to_amount);
let mut new_from_amount = from_amount - amount;
new_from_amount = has_enough_funds.if_then_else(&new_from_amount, from_amount);
(new_from_amount, new_to_amount)
}
/// Parallel variant of [`transfer_whitepaper`].
pub fn par_transfer_whitepaper<FheType>(
from_amount: &FheType,
to_amount: &FheType,
amount: &FheType,
) -> (FheType, FheType)
where
FheType: Add<Output = FheType> + for<'a> FheOrd<&'a FheType> + Send + Sync,
FheBool: IfThenElse<FheType>,
@@ -493,7 +515,7 @@ fn main() {
&cks,
"FheUint64",
"transfer::whitepaper",
transfer_whitepaper::<FheUint64>,
par_transfer_whitepaper::<FheUint64>,
);
print_transfer_pbs_counts(&cks, "FheUint64", "no_cmux", transfer_no_cmux::<FheUint64>);
print_transfer_pbs_counts(
@@ -514,7 +536,7 @@ fn main() {
bench_name,
"FheUint64",
"transfer::whitepaper",
transfer_whitepaper::<FheUint64>,
par_transfer_whitepaper::<FheUint64>,
);
bench_transfer_latency(
&mut group,
@@ -553,7 +575,7 @@ fn main() {
bench_name,
"FheUint64",
"transfer::whitepaper",
transfer_whitepaper::<FheUint64>,
par_transfer_whitepaper::<FheUint64>,
);
bench_transfer_throughput(
&mut group,
@@ -607,7 +629,7 @@ fn main() {
&cks,
"FheUint64",
"transfer::whitepaper",
transfer_whitepaper::<FheUint64>,
par_transfer_whitepaper::<FheUint64>,
);
print_transfer_pbs_counts(&cks, "FheUint64", "no_cmux", transfer_no_cmux::<FheUint64>);
print_transfer_pbs_counts(
@@ -628,7 +650,7 @@ fn main() {
bench_name,
"FheUint64",
"transfer::whitepaper",
transfer_whitepaper::<FheUint64>,
par_transfer_whitepaper::<FheUint64>,
);
bench_transfer_latency(
&mut group,
@@ -667,7 +689,7 @@ fn main() {
bench_name,
"FheUint64",
"transfer::whitepaper",
transfer_whitepaper::<FheUint64>,
par_transfer_whitepaper::<FheUint64>,
);
cuda_bench_transfer_throughput(
&mut group,