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): reduce integer bench time

Browse Source
This commit is contained in:
Agnes Leroy
2024-07-11 16:26:45 +02:00
committed by Agnès Leroy
parent a62c19b735
commit 0370bf6a3f
2 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tfhe/benches/integer/bench.rs
View File

@@ -1197,7 +1197,7 @@ mod cuda {
let mut bench_group = c.benchmark_group(bench_name);
bench_group
.sample_size(15)
.measurement_time(std::time::Duration::from_secs(60));
.measurement_time(std::time::Duration::from_secs(30));
let mut rng = rand::thread_rng();
let streams = CudaStreams::new_multi_gpu();
@@ -1261,7 +1261,7 @@ mod cuda {
let mut bench_group = c.benchmark_group(bench_name);
bench_group
.sample_size(15)
.measurement_time(std::time::Duration::from_secs(60));
.measurement_time(std::time::Duration::from_secs(30));
let mut rng = rand::thread_rng();
let streams = CudaStreams::new_multi_gpu();
@@ -1330,7 +1330,7 @@ mod cuda {
let mut bench_group = c.benchmark_group(bench_name);
bench_group
.sample_size(15)
.measurement_time(std::time::Duration::from_secs(60));
.measurement_time(std::time::Duration::from_secs(30));
let mut rng = rand::thread_rng();
let stream = CudaStreams::new_multi_gpu();
@@ -1391,7 +1391,7 @@ mod cuda {
let mut bench_group = c.benchmark_group(bench_name);
bench_group
.sample_size(15)
.measurement_time(std::time::Duration::from_secs(60));
.measurement_time(std::time::Duration::from_secs(30));
let mut rng = rand::thread_rng();
let stream = CudaStreams::new_multi_gpu();

10
tfhe/benches/integer/signed_bench.rs
View File

@@ -1224,7 +1224,7 @@ mod cuda {
let mut bench_group = c.benchmark_group(bench_name);
bench_group
.sample_size(15)
.measurement_time(std::time::Duration::from_secs(60));
.measurement_time(std::time::Duration::from_secs(30));
let mut rng = rand::thread_rng();
let stream = CudaStreams::new_multi_gpu();
@@ -1310,7 +1310,7 @@ mod cuda {
let mut bench_group = c.benchmark_group(bench_name);
bench_group
.sample_size(15)
.measurement_time(std::time::Duration::from_secs(60));
.measurement_time(std::time::Duration::from_secs(30));
let mut rng = rand::thread_rng();
let stream = CudaStreams::new_multi_gpu();
@@ -1386,7 +1386,7 @@ mod cuda {
let mut bench_group = c.benchmark_group(bench_name);
bench_group
.sample_size(15)
.measurement_time(std::time::Duration::from_secs(60));
.measurement_time(std::time::Duration::from_secs(30));
let mut rng = rand::thread_rng();
let stream = CudaStreams::new_multi_gpu();
@@ -1476,7 +1476,7 @@ mod cuda {
let mut bench_group = c.benchmark_group(bench_name);
bench_group
.sample_size(15)
.measurement_time(std::time::Duration::from_secs(60));
.measurement_time(std::time::Duration::from_secs(30));
let mut rng = rand::thread_rng();
let stream = CudaStreams::new_multi_gpu();
@@ -1553,7 +1553,7 @@ mod cuda {
let mut bench_group = c.benchmark_group("integer::cuda::signed::if_then_else");
bench_group
.sample_size(15)
.measurement_time(std::time::Duration::from_secs(60));
.measurement_time(std::time::Duration::from_secs(30));
let mut rng = rand::thread_rng();
let stream = CudaStreams::new_multi_gpu();